YAML data seeds importer for Rails 3

db.rake

require 'active_record/fixtures'
 
namespace :db do
  desc 'Seed the database with once/ and always/ fixtures.'
  task :seed => :environment do
    load_fixtures 'seed/once'
    load_fixtures 'seed/always', :always
  end
 
  desc 'Seed the database with develop/ fixtures.'
  task :develop => :environment do
    load_fixtures 'seed/develop', :always
  end
 
  private
 
  def load_fixtures(dir, always = false)
    Dir.glob(File.join(Rails.root, 'db', dir, '*.yml')).each do |fixture_file|
      table_name = File.basename(fixture_file, '.yml')
      
      if always || table_empty?(table_name)
        ActiveRecord::Fixtures.create_fixtures(File.join('db', dir), table_name)
      end
    end
  end
 
  def table_empty?(table_name)
    quoted = connection.quote_table_name(table_name)
    connection.select_value("SELECT COUNT(*) FROM #{quoted}").to_i.zero?
  end
 
  def connection
    ActiveRecord::Base.connection
  end
end