diff options
Diffstat (limited to 'factory/mongoengine.py')
-rw-r--r-- | factory/mongoengine.py | 12 |
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 |