导出数据库的所有数据成YML文件

最新推荐文章于 2024-08-27 17:54:16 发布

最新推荐文章于 2024-08-27 17:54:16 发布 · 670 阅读

文章标签：

42 篇文章

订阅专栏

本文介绍了一段Ruby代码，该代码用于从现有数据库中创建YAML测试数据集。默认使用开发环境数据库，可通过设置RAILS_ENV变量进行覆盖。代码支持导出所有表数据及特定引用表数据。

ruby 代码

namespace :db do
namespace :fixtures do
desc 'Create YAML test fixtures from data in an existing database.
Defaults to development database. Set RAILS_ENV to override.'
task :dump_all => :environment do
sql = "SELECT * FROM %s"
skip_tables = ["schema_info"]
ActiveRecord::Base.establish_connection(:development)
(ActiveRecord::Base.connection.tables - skip_tables).each do |table_name|
i = "000"
File.open("#{RAILS_ROOT}/test/fixtures/#{table_name}.yml", 'w') do |file|
data = ActiveRecord::Base.connection.select_all(sql % table_name)
file.write data.inject({}) { |hash, record|
hash["#{table_name}_#{i.succ!}"] = record
hash
}.to_yaml
end
end
end
end
namespace :fixtures do
desc 'Create YAML test fixtures for references. Defaults to development database.
Set RAILS_ENV to override.'
task :dump_references => :environment do
sql = "SELECT * FROM %s"
dump_tables = ["areas","countries"]
ActiveRecord::Base.establish_connection(:development)
dump_tables.each do |table_name|
i = "000"
file_name = "#{RAILS_ROOT}/test/fixtures/#{table_name}.yml"
p "Fixture save for table #{table_name} to #{file_name}"
File.open(file_name, 'w') do |file|
data = ActiveRecord::Base.connection.select_all(sql % table_name)
file.write data.inject({}) { |hash, record|
hash["#{table_name}_#{i.succ!}"] = record
hash
}.to_yaml
end
end
end
end
end