Defer sqlalchemy import

Author: adamchainz

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")

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: