sample_populate.rake

Bundler.require(:development, :test, :qa)

namespace :db do
  namespace :sample do
    desc "Fill database with sample data"
    task populate: :environment do
      ActiveRecord::Base.transaction do
        %w(projects attrs tasks).each { |table| populate(table) }
      end
    end

  private

    def populate(table)
      return unless ENV['all'].present? || ENV[table].present?

      if ENV[table].present?
        num = ENV[table]&.to_i
        if num.zero?
          puts "skip #{table}"
        else
          puts "populate #{num} #{table}"
          send("populate_#{table}", num)
        end
      elsif ENV['all'].present?
        puts "populate #{table}"
        send("populate_#{table}")
      end
    end

    def populate_projects(num = 2)
      Project.destroy_all
      FactoryGirl.create_list(:project, num)
    end
  end
end