28 lines
595 B
Ruby
28 lines
595 B
Ruby
|
require "seed-fu-mongoid/version"
|
||
|
|
require "seed-fu-mongoid/document_seeder"
|
||
|
|
require "seed-fu-mongoid/seed_proxy"
|
||
|
|
require 'mongoid/document'
|
||
|
|
|
||
|
|
if defined?(Rails::Railtie)
|
||
|
|
require 'seed-fu-mongoid/railtie'
|
||
|
|
end
|
||
|
|
|
||
|
|
# punch the duck out of Mongoid::Document
|
||
|
|
|
||
|
|
module Mongoid::Document
|
||
|
|
module ClassMethods
|
||
|
|
def seed_once(*constraints)
|
||
|
|
seeder = SeedFuMongoid::DocumentSeeder.new(self, constraints, block)
|
||
|
|
|
||
|
|
if seeder.new?
|
||
|
|
seeder.seed!
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def seed(*constraints, &block)
|
||
|
|
SeedFuMongoid::DocumentSeeder.new(self, constraints, block).seed!
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|