set svn:eol-style native at lib/tasks/extract_fixtures.rake

git-svn-id: svn+ssh://rubyforge.org/var/svn/redmine/trunk@7744 e93f8b46-1217-0410-a6f0-8f06a7374b81
This commit is contained in:
Toshi MARUYAMA 2011-11-06 17:04:14 +00:00
parent e8b8406ecf
commit c4c8c3e3e2
1 changed files with 22 additions and 22 deletions

View File

@ -1,22 +1,22 @@
desc 'Create YAML test fixtures from data in an existing database. desc 'Create YAML test fixtures from data in an existing database.
Defaults to development database. Set RAILS_ENV to override.' Defaults to development database. Set RAILS_ENV to override.'
task :extract_fixtures => :environment do task :extract_fixtures => :environment do
sql = "SELECT * FROM %s" sql = "SELECT * FROM %s"
skip_tables = ["schema_info"] skip_tables = ["schema_info"]
ActiveRecord::Base.establish_connection ActiveRecord::Base.establish_connection
(ActiveRecord::Base.connection.tables - skip_tables).each do |table_name| (ActiveRecord::Base.connection.tables - skip_tables).each do |table_name|
i = "000" i = "000"
File.open("#{Rails.root}/#{table_name}.yml", 'w' ) do |file| File.open("#{Rails.root}/#{table_name}.yml", 'w' ) do |file|
data = ActiveRecord::Base.connection.select_all(sql % table_name) data = ActiveRecord::Base.connection.select_all(sql % table_name)
file.write data.inject({}) { |hash, record| file.write data.inject({}) { |hash, record|
# cast extracted values # cast extracted values
ActiveRecord::Base.connection.columns(table_name).each { |col| ActiveRecord::Base.connection.columns(table_name).each { |col|
record[col.name] = col.type_cast(record[col.name]) if record[col.name] record[col.name] = col.type_cast(record[col.name]) if record[col.name]
} }
hash["#{table_name}_#{i.succ!}"] = record hash["#{table_name}_#{i.succ!}"] = record
hash hash
}.to_yaml }.to_yaml
end end
end end
end end