Defer sqlalchemy import

FactoryBoy · Mar 29, 2024 · ee8d278 · ee8d278
1 parent dca3a16
commit ee8d278
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 15 deletions.
diff --git a/factory/__init__.py b/factory/__init__.py
@@ -2,6 +2,7 @@
 
 import importlib.metadata
 
+from . import alchemy, mogo, mongoengine
 from .base import (
  BaseDictFactory,
  BaseListFactory,
@@ -54,22 +55,10 @@
  stub_batch,
 )
 
-try:
- from . import alchemy
-except ImportError:
- pass
 try:
  from . import django
 except ImportError:
  pass
-try:
- from . import mogo
-except ImportError:
- pass
-try:
- from . import mongoengine
-except ImportError:
- pass
 
 __author__ = 'Raphaël Barrois <raphael.barrois+fboy@polytechnique.org>'
 __version__ = importlib.metadata.version("factory_boy")
diff --git a/factory/alchemy.py b/factory/alchemy.py
@@ -1,8 +1,5 @@
 # Copyright: See the LICENSE file.
 
-from sqlalchemy.exc import IntegrityError
-from sqlalchemy.orm.exc import NoResultFound
-
 from . import base, errors
 
 SESSION_PERSISTENCE_COMMIT = 'commit'
@@ -64,6 +61,9 @@ def _generate(cls, strategy, params):
 
  @classmethod
  def _get_or_create(cls, model_class, session, args, kwargs):
+ from sqlalchemy.exc import IntegrityError
+ from sqlalchemy.orm.exc import NoResultFound
+
  key_fields = {}
  for field in cls._meta.sqlalchemy_get_or_create:
  if field not in kwargs: