MetaSkills.net
 

Custom Rake Task To Unload Fixtures

| delicious | digg | technorati | reddit |
Posted On: October 2nd, 2006 by kencollins

I made a revised version of a rake task that I have used quite often, for unloading existing DB tables into fixture data and thought I would share. This rake task is the in correct name space and adds a "rake db:fixtures:unload" command to your rails project when you put this in "lib/tasks/foo.rake". It can take an optional TABLES variable or if none is present the whole array of DB tables are converted. I find this rake task helpful when dealing with LARGE databases.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21


namespace :db do
  namespace :fixtures do
    desc 'Create YAML test fixtures from data in specifed tables. Set table names by TABLES=foos,bars,etc'
    task :unload => :environment do
      sql  = "SELECT * FROM %s"
      tables = ["schema_info"]
      ActiveRecord::Base.establish_connection
      (ENV['TABLES'] ? ENV['TABLES'].split(/,/) : ActiveRecord::Base.connection.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
end