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

最新推荐文章于 2023-03-04 20:29:41 发布

原创最新推荐文章于 2023-03-04 20:29:41 发布 · 143 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Rails #ActiveRecord #Ruby

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

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