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

ruby 代码

1. namespace :db do  
2.   namespace :fixtures do  
3.   
4.     desc 'Create YAML test fixtures from data in an existing database.   
5. Defaults to development database. Set RAILS_ENV to override.'   
6.     task :dump_all => :environment do  
7.       sql = "SELECT * FROM %s"  
8.       skip_tables = ["schema_info"]   
9.       ActiveRecord::Base.establish_connection(:development)   
10.       (ActiveRecord::Base.connection.tables - skip_tables).each do |table_name|   
11.         i = "000"  
12.         File.open("#{RAILS_ROOT}/test/fixtures/#{table_name}.yml", 'w') do |file|   
13.           data = ActiveRecord::Base.connection.select_all(sql % table_name)   
14.           file.write data.inject({}) { |hash, record|   
15.             hash["#{table_name}_#{i.succ!}"] = record   
16.             hash   
17.           }.to_yaml   
18.         end  
19.       end  
20.     end  
21.   end  
22.      
23.   namespace :fixtures do  
24.     desc 'Create YAML test fixtures for references. Defaults to development database.    
25.     Set RAILS_ENV to override.'   
26.     task :dump_references => :environment do  
27.       sql = "SELECT * FROM %s"  
28.       dump_tables = ["areas","countries"]   
29.       ActiveRecord::Base.establish_connection(:development)   
30.       dump_tables.each do |table_name|   
31.         i = "000"  
32.         file_name = "#{RAILS_ROOT}/test/fixtures/#{table_name}.yml"  
33.         p "Fixture save for table #{table_name} to #{file_name}"  
34.         File.open(file_name, 'w') do |file|   
35.           data = ActiveRecord::Base.connection.select_all(sql % table_name)   
36.           file.write data.inject({}) { |hash, record|   
37.             hash["#{table_name}_#{i.succ!}"] = record   
38.             hash   
39.           }.to_yaml   
40.         end  
41.       end  
42.     end  
43.   end  
44. end