添加自定义种子文件

时间:2013-11-09 04:12:33

标签: ruby-on-rails ruby ruby-on-rails-3.2 rake seed

我想用虚拟数据填充新功能,但不想使用db / seeds.rb文件,因为它已经有种子其他与此功能无关的数据。

要运行默认的seeds.rb文件,请运行命令rake db:seed

如果我在名为db的{​​{1}}目录中创建了一个文件,那么rake命令会如何(仅)运行该文件?

3 个答案:

答案 0 :(得分:161)

首先创建一个单独的目录来保存自定义种子 - 此示例使用db/seeds。然后,通过将rakefile添加到lib/tasks目录来创建自定义任务:

# lib/tasks/custom_seed.rake
namespace :db do
  namespace :seed do
    Dir[Rails.root.join('db', 'seeds', '*.rb')].each do |filename|
      task_name = File.basename(filename, '.rb')
      desc "Seed " + task_name + ", based on the file with the same name in `db/seeds/*.rb`"
      task task_name.to_sym => :environment do
        load(filename) if File.exist?(filename)
      end
    end
  end
end

此rakefile接受db/seeds目录中的种子文件的名称(.rb扩展名除外),然后按运行seeds.rb运行它。您可以通过从命令行发出以下命令来执行rake任务:

rake db:seed:file_name # Name of the file EXCLUDING the .rb extension 

更新:现在,它还应该在运行rake --tasksrake -T时列出种子任务。

答案 1 :(得分:12)

我尝试了zeantsoi的答案,但它没有给我我想要的东西,它完成了目录中的所有文件。劈开它并得到它。

namespace :db do
  namespace :seed do
    task :single => :environment do
      filename = Dir[File.join(Rails.root, 'db', 'seeds', "#{ENV['SEED']}.seeds.rb")][0]
      puts "Seeding #{filename}..."
      load(filename) if File.exist?(filename)
    end
  end
end

要使用此功能,请执行以下操作:

rake db:seed:single SEED=<seed_name_without_.seeds.rb>

这将在Rails.root / db / seeds文件夹中查找没有.seeds.rb的文件名(它会为您添加这些文件名)。

工作示例:

rake db:seed:single SEED=my_custom_seed

上面会播种Rails.root/db/seeds/my_custom_seed.seeds.rb文件

答案 2 :(得分:8)

太复杂了! 我只想要一个简单的任务来执行db / seeds目录下的每个文件而不传入任何文件名。

# lib/tasks/seed.rake
desc "Run all files in db/seeds directory"

namespace :db do
  task seed: :environment do
    Dir[File.join(Rails.root, 'db', 'seeds', '*.rb')].each do |filename|
      puts "seeding - #{filename}. for reals, yo!"
      load(filename)
    end
  end
end