Add SQLAlchemy support

Makefile

@@ -130,7 +130,7 @@ test-release:
 	source $(VENV) && twine upload --repository testpypi dist/*
 
 mypy:
-	source $(VENV) && mypy .
+	source $(VENV) && mypy fake.py
 
 %:
 	@:

examples/sqlalchemy/article/factories.py

@@ -3,14 +3,15 @@
 from fake import (
     FACTORY,
     FileSystemStorage,
+    SQLAlchemyModelFactory,
     SubFactory,
     post_save,
     pre_save,
     trait,
 )
-from sqlalchemy_model_factory import SQLAlchemyModelFactory
 
 from article.models import Article, User
+from config import SESSION
 
 __author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"
 __copyright__ = "2023 Artur Barseghyan"
@@ -20,13 +21,16 @@
     "UserFactory",
 )
 
-# Build paths inside the project like this: BASE_DIR / 'subdir'.
+# Storage config. Build paths inside the project like this: BASE_DIR / 'subdir'
 BASE_DIR = Path(__file__).resolve().parent.parent
 MEDIA_ROOT = BASE_DIR / "media"
-
 STORAGE = FileSystemStorage(root_path=MEDIA_ROOT, rel_path="tmp")
 
 
+def get_session():
+    return SESSION()
+
+
 class UserFactory(SQLAlchemyModelFactory):
     """User factory."""
 
@@ -44,6 +48,9 @@ class Meta:
         model = User
         get_or_create = ("username",)
 
+    class MetaSQLAlchemy:
+        get_session = get_session
+
     @trait
     def is_admin_user(self, instance: User) -> None:
         instance.is_superuser = True
@@ -74,6 +81,9 @@ class ArticleFactory(SQLAlchemyModelFactory):
     class Meta:
         model = Article
 
+    class MetaSQLAlchemy:
+        get_session = get_session
+
     @pre_save
     def _pre_save_method(self, instance):
         instance.pre_save_called = True

examples/sqlalchemy/article/tests.py

@@ -1,8 +1,7 @@
 import unittest
 
-from sqlalchemy_model_factory import SESSION
-
 from article.factories import ArticleFactory, UserFactory
+from config import SESSION
 
 __author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"
 __copyright__ = "2023 Artur Barseghyan"

examples/sqlalchemy/config.py

@@ -0,0 +1,16 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import scoped_session, sessionmaker
+
+__author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"
+__copyright__ = "2023 Artur Barseghyan"
+__license__ = "MIT"
+__all__ = (
+    "DATABASE_URL",
+    "ENGINE",
+    "SESSION",
+)
+
+# SQLAlchemy
+DATABASE_URL = "sqlite:///test_database.db"
+ENGINE = create_engine(DATABASE_URL)
+SESSION = scoped_session(sessionmaker(bind=ENGINE))

examples/sqlalchemy/manage.py

@@ -19,19 +19,19 @@ def main():
     """Run administrative tasks based on command line arguments."""
     sys.path.insert(0, os.path.abspath(os.path.join("..", "..")))
     sys.path.insert(0, os.path.abspath("."))
+
+    from article.models import Base  # noqa
+    from config import ENGINE  # noqa
+
+    Base.metadata.create_all(ENGINE)
+
     parser = argparse.ArgumentParser(
         description="Management script for the project."
     )
     parser.add_argument("command", help="The command to run (test or shell)")
 
     args = parser.parse_args()
 
-    from sqlalchemy_model_factory import ENGINE
-
-    from article.models import Base
-
-    Base.metadata.create_all(ENGINE)
-
     if args.command == "test":
         run_tests()
     elif args.command == "shell":