Contents

namespace :db do
  desc <<-EOS
    Loads seed data for the current environment. It will look for
    ruby seed files in <RAILS_ROOT>/db/fixtures/ and 
    <RAILS_ROOT>/db/fixtures/<RAILS_ENV>/.

    By default it will load any ruby files found. You can filter the files
    loaded by passing in the SEED environment variable with a comma-delimited
    list of patterns to include. Any files not matching the pattern will
    not be loaded.
    
    You can also change the directory where seed files are looked for
    with the FIXTURE_PATH environment variable. 
    
    Examples:
      # default, to load all seed files for the current environment
      rake db:seed
      
      # to load seed files matching orders or customers
      rake db:seed SEED=orders,customers
      
      # to load files from RAILS_ROOT/features/fixtures
      rake db:seed FIXTURE_PATH=features/fixtures 
  EOS
  task :seed => :environment do
    fixture_path = ENV["FIXTURE_PATH"] ? ENV["FIXTURE_PATH"] : "db/fixtures"

    global_seed_files = Dir[File.join(RAILS_ROOT, fixture_path, '*.rb')].sort
    env_specific_seed_files = Dir[File.join(RAILS_ROOT, fixture_path, RAILS_ENV, '*.rb')].sort
    potential_seed_files = (global_seed_files + env_specific_seed_files).uniq
    
    if ENV["SEED"]
      filter = ENV["SEED"].gsub(/,/, "|")
      potential_seed_files.reject!{ |file| true unless file =~ /#{filter}/ }
      puts "\n == Filtering seed files against regexp: #{filter}"
    end

    potential_seed_files.each do |file|
      pretty_name = file.sub("#{RAILS_ROOT}/", "")
      puts "\n== Seed from #{pretty_name} " + ("=" * (60 - (17 + File.split(file).last.length)))

      old_level = ActiveRecord::Base.logger.level
      begin
        ActiveRecord::Base.validation_disabled = true
        ActiveRecord::Base.logger.level = 7
        ActiveRecord::Base.transaction do
          load file
        end
      ensure
        ActiveRecord::Base.validation_disabled = false
        ActiveRecord::Base.logger.level = old_level
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
mixonic-seed-fu-1.1.0	tasks/seed_fu_tasks.rake