diff options
Diffstat (limited to 'factory')
| -rw-r--r-- | factory/__init__.py | 1 | ||||
| -rw-r--r-- | factory/mongoengine.py | 45 |
2 files changed, 46 insertions, 0 deletions
diff --git a/factory/__init__.py b/factory/__init__.py index ea45a9a..a27eb40 100644 --- a/factory/__init__.py +++ b/factory/__init__.py @@ -38,6 +38,7 @@ from .base import ( use_strategy, ) +# Backward compatibility; this should be removed soon. from .mogo import MogoFactory from .django import DjangoModelFactory diff --git a/factory/mongoengine.py b/factory/mongoengine.py new file mode 100644 index 0000000..8cd3a67 --- /dev/null +++ b/factory/mongoengine.py @@ -0,0 +1,45 @@ +# -*- coding: utf-8 -*- +# Copyright (c) 2010 Mark Sandstrom +# Copyright (c) 2011-2013 Raphaël Barrois +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +# THE SOFTWARE. + + +from __future__ import unicode_literals + + +"""factory_boy extensions for use with the mongoengine library (pymongo wrapper).""" + + +from . import base + + +class MongoEngineFactory(base.Factory): + """Factory for mongoengine objects.""" + ABSTRACT_FACTORY = True + + @classmethod + def _build(cls, target_class, *args, **kwargs): + return target_class(*args, **kwargs) + + @classmethod + def _create(cls, target_class, *args, **kwargs): + instance = target_class(*args, **kwargs) + instance.save() + return instance |