Defer sqlalchemy import

Author: adamchainz

factory/__init__.py

@@ -54,22 +54,13 @@
     stub_batch,
 )
 
-try:
-    from . import alchemy
-except ImportError:
-    pass
+from . import alchemy
 try:
     from . import django
 except ImportError:
     pass
-try:
-    from . import mogo
-except ImportError:
-    pass
-try:
-    from . import mongoengine
-except ImportError:
-    pass
+from . import mogo
+from . import mongoengine
 
 __author__ = 'Raphaël Barrois <raphael.barrois+fboy@polytechnique.org>'
 __version__ = importlib.metadata.version("factory_boy")

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: