summaryrefslogtreecommitdiff
path: root/factory/base.py
diff options
context:
space:
mode:
authorRaphaël Barrois <raphael.barrois@polytechnique.org>2013-08-28 00:37:32 +0200
committerRaphaël Barrois <raphael.barrois@polytechnique.org>2013-08-28 00:38:24 +0200
commit7fc3e4cbdae050dcde49ea3101636ddf57d6c96d (patch)
tree119a7a3effecc95ef657a0c3fe673466e830d365 /factory/base.py
parentac1322cf12fc79e5d11abf1aa95b01413b312258 (diff)
downloadfactory-boy-7fc3e4cbdae050dcde49ea3101636ddf57d6c96d.tar
factory-boy-7fc3e4cbdae050dcde49ea3101636ddf57d6c96d.tar.gz
Remove duplicate SQLAlchemyModelFactory (Closes #83).
Diffstat (limited to 'factory/base.py')
-rw-r--r--factory/base.py33
1 files changed, 0 insertions, 33 deletions
diff --git a/factory/base.py b/factory/base.py
index 0db8249..ac906de 100644
--- a/factory/base.py
+++ b/factory/base.py
@@ -640,36 +640,3 @@ def use_strategy(new_strategy):
klass.FACTORY_STRATEGY = new_strategy
return klass
return wrapped_class
-
-
-class SQLAlchemyModelFactory(Factory):
- """Factory for SQLAlchemy models. """
-
- ABSTRACT_FACTORY = True
- FACTORY_HIDDEN_ARGS=('SESSION',)
-
- def __init__(self, session):
- self.session = session
-
- @classmethod
- def _get_function(cls, function_name):
- session = cls._declarations['SESSION']
- sqlalchemy = __import__(session.__module__)
- max = getattr(sqlalchemy.sql.functions, function_name)
-
- @classmethod
- def _setup_next_sequence(cls, *args, **kwargs):
- """Compute the next available PK, based on the 'pk' database field."""
- max = cls._get_function('max')
- session = cls._declarations['SESSION']
- pk = cls.FACTORY_FOR.__table__.primary_key.columns.values()[0].key
- max_pk = session.query(max(getattr(cls.FACTORY_FOR, pk))).one()
- return max_pk[0] + 1 if max_pk[0] else 1
-
- @classmethod
- def _create(cls, target_class, *args, **kwargs):
- """Create an instance of the model, and save it to the database."""
- session = cls._declarations['SESSION']
- obj = target_class(*args, **kwargs)
- session.add(obj)
- return obj