github-migration
/
seed-fu-mongoid
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Merge pull request #1 from bopm/bopm-patch-1 

Fix seeding for a first time and duplication on second run
This commit is contained in:
Sergey Moiseev 2017-04-11 03:05:50 +03:00 committed by GitHub
parent 3ed5aa9e10 8256e05d68
commit d3f99bc66f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/seed-fu-mongoid/document_seeder.rb
View File

@ -48,7 +48,7 @@ module SeedFuMongoid
def document
@doc ||= begin
@klass.find_by(constraint_search)
@klass.find_by(constraint_search) || @klass.new
rescue => e
@klass.new
end
@ -58,9 +58,9 @@ module SeedFuMongoid
return @constraint_search if @constraint_search
@constraint_search = {}
constraints.each do |constraint|
@constraints.each do |constraint|
if data[constraint]
@constraint_search[constraint] = proxy[constraint]
@constraint_search[constraint] = data[constraint]
else
raise ConstraintNotDefined.new(constraint)
end