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
commit d3f99bc66f
1 changed files with 3 additions and 3 deletions

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