summaryrefslogtreecommitdiff
path: root/factory/mongoengine.py
diff options
context:
space:
mode:
Diffstat (limited to 'factory/mongoengine.py')
-rw-r--r--factory/mongoengine.py12
1 files changed, 7 insertions, 5 deletions
diff --git a/factory/mongoengine.py b/factory/mongoengine.py
index 462f5f2..e3ab99c 100644
--- a/factory/mongoengine.py
+++ b/factory/mongoengine.py
@@ -32,15 +32,17 @@ from . import base
class MongoEngineFactory(base.Factory):
"""Factory for mongoengine objects."""
- ABSTRACT_FACTORY = True
+
+ class Meta:
+ abstract = True
@classmethod
- def _build(cls, target_class, *args, **kwargs):
- return target_class(*args, **kwargs)
+ def _build(cls, model_class, *args, **kwargs):
+ return model_class(*args, **kwargs)
@classmethod
- def _create(cls, target_class, *args, **kwargs):
- instance = target_class(*args, **kwargs)
+ def _create(cls, model_class, *args, **kwargs):
+ instance = model_class(*args, **kwargs)
if instance._is_document:
instance.save()
return instance