milaniliev / db_fixtures_dump_enhanced

A Rake task to create fixture files that allows for specifying tables, conditions and limits

This URL has Read+Write access

db_fixtures_dump_enhanced / fixtures.rake
100644 27 lines (25 sloc) 1.06 kb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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 => :environment do
      sql = "SELECT * FROM %s"
      sql << " WHERE " + ENV['WHERE'] if ENV['WHERE']
      ActiveRecord::Base.connection.add_limit_offset!(sql, { :limit => ENV['LIMIT'].to_i, :offset => ENV['OFFSET'].to_i }) if ENV['LIMIT'] || ENV['OFFSET']
      skip_tables = ["schema_info"]
      ActiveRecord::Base.establish_connection(RAILS_ENV)
      tables = ENV['TABLES'].split(',')
      tables ||= (ActiveRecord::Base.connection.tables - skip_tables)
      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