将现在的数据库记录生成fixture数据文件

iteye_3589

于 2009-10-25 12:01:11 发布

阅读量131

点赞数

CC 4.0 BY-SA版权

文章标签： Rails ActiveRecord Ruby

本文链接：https://blog.youkuaiyun.com/iteye_3589/article/details/81787137

本文介绍了一个Rails任务，用于从当前的开发数据库中选取特定表的数据，并将其转换为YAML格式的固定测试数据文件。该过程包括连接到数据库、选择数据并写入文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

desc 'generate yml data file from current development database'

task :generate_fixture => :environment do
  ActiveRecord::Base.establish_connection

  table_name = STDIN.gets.chomp
  raise "fixtures file of #{table_name} alreay exist!" if File.exist?("#{RAILS_ROOT}/spec/fixtures/#{table_name}.yml")
  i= "000"
  File.open("#{RAILS_ROOT}/spec/fixtures/#{table_name}.yml", 'w') do |file|
    data = ActiveRecord::Base.connection.select_all("SELECT * FROM #{table_name} where rownum < 5 order by id desc ")
    file.write data.inject({}) { |hash, record|
      hash["#{table_name}_#{i.succ!}"] = record
      hash
    }.to_yaml
  end
end