Allow the definition of 'abstract' factories.

Needs more testing.

Signed-off-by: Raphaël Barrois <raphael.barrois@polyconseil.fr>

1 files changed, 4 insertions, 2 deletions

diff --git a/factory/base.py b/factory/base.py
index c474348..cfbfae8 100644
--- a/factory/base.py
+++ b/factory/base.py
@@ -73,7 +73,7 @@ class BaseFactoryMetaClass(type):
         an associated class instance at a later time.'''
 
         base = get_factory_base(bases)
-        if not base:
+        if not base or attrs.get('ABSTRACT_FACTORY', False):
             # If this isn't a subclass of Factory, don't do anything special.
             return super(BaseFactoryMetaClass, cls).__new__(cls, class_name, bases, attrs)
 
@@ -100,7 +100,7 @@ class FactoryMetaClass(BaseFactoryMetaClass):
         for construction of an associated class instance at a later time.'''
 
         base = get_factory_base(bases)
-        if not base:
+        if not base or attrs.get('ABSTRACT_FACTORY', False):
             # If this isn't a subclass of Factory, don't do anything special.
             return super(FactoryMetaClass, cls).__new__(cls, class_name, bases, attrs)
 
@@ -253,6 +253,8 @@ class DjangoModelFactory(Factory):
     handle those for non-numerical primary keys.
     """
 
+    ABSTRACT_FACTORY = True
+
     @classmethod
     def _setup_next_sequence(cls):
         try: