austial-orm 0.1.5__py3-none-any.whl

austial/orm/__init__.py

@@ -0,0 +1,3 @@
+from austial.orm.database_module import DATABASE_ENGINE, DATABASE_SESSION_FACTORY, DatabaseModule
+
+__all__ = ["DATABASE_ENGINE", "DATABASE_SESSION_FACTORY", "DatabaseModule"]

austial/orm/database_module.py

@@ -0,0 +1,61 @@
+"""``DatabaseModule`` -- mirrors ``@nestjs/typeorm``'s ``TypeOrmModule.forRoot[Async]()``.
+
+Backed by SQLAlchemy's async engine; defaults to a local sqlite file so a
+freshly generated project works with zero external services.
+"""
+
+from __future__ import annotations
+
+from collections.abc import Callable, Sequence
+from typing import Any
+
+from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
+
+from austial.common.decorators.modules import Module
+from austial.core.dynamic_module import DynamicModule
+
+DATABASE_ENGINE = "DATABASE_ENGINE"
+DATABASE_SESSION_FACTORY = "DATABASE_SESSION_FACTORY"
+
+
+@Module()
+class DatabaseModule:
+    """Import via ``DatabaseModule.for_root(url=...)`` or
+    ``DatabaseModule.for_root_async(use_factory=..., inject=[ConfigService])``,
+    mirroring Nest's TypeORM dynamic-module conventions. Exposes two tokens
+    other providers can ``Inject(...)``: ``DATABASE_ENGINE`` and
+    ``DATABASE_SESSION_FACTORY``."""
+
+    @staticmethod
+    def for_root(url: str = "sqlite+aiosqlite:///./austial.db", **engine_kwargs: Any) -> DynamicModule:
+        engine = create_async_engine(url, **engine_kwargs)
+        session_factory = async_sessionmaker(engine, expire_on_commit=False)
+        return DynamicModule(
+            module=DatabaseModule,
+            providers=[
+                {"provide": DATABASE_ENGINE, "useValue": engine},
+                {"provide": DATABASE_SESSION_FACTORY, "useValue": session_factory},
+            ],
+            exports=[DATABASE_ENGINE, DATABASE_SESSION_FACTORY],
+        )
+
+    @staticmethod
+    def for_root_async(*, use_factory: Callable[..., str], inject: Sequence[Any] = ()) -> DynamicModule:
+        def _build_engine(*deps: Any):
+            return create_async_engine(use_factory(*deps))
+
+        def _build_session_factory(engine: Any):
+            return async_sessionmaker(engine, expire_on_commit=False)
+
+        return DynamicModule(
+            module=DatabaseModule,
+            providers=[
+                {"provide": DATABASE_ENGINE, "useFactory": _build_engine, "inject": list(inject)},
+                {
+                    "provide": DATABASE_SESSION_FACTORY,
+                    "useFactory": _build_session_factory,
+                    "inject": [DATABASE_ENGINE],
+                },
+            ],
+            exports=[DATABASE_ENGINE, DATABASE_SESSION_FACTORY],
+        )

austial_orm-0.1.5.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: austial-orm
+Version: 0.1.5
+Summary: DatabaseModule wrapping SQLAlchemy's async engine/session -- mirrors @nestjs/typeorm (no main-monorepo equivalent).
+Requires-Python: >=3.11
+Requires-Dist: aiosqlite>=0.20.0
+Requires-Dist: austial-common
+Requires-Dist: austial-core
+Requires-Dist: sqlalchemy>=2.0.30

austial_orm-0.1.5.dist-info/RECORD

@@ -0,0 +1,5 @@
+austial/orm/__init__.py,sha256=2K_ltYgBDoyRm7RyrN91Kwxt-Zn69fdTPqUUz_-R8kA,175
+austial/orm/database_module.py,sha256=XWMGBNIi2GjiCCR7dZn07x9AF7UCkp8VS9ZZhyHcSbY,2373
+austial_orm-0.1.5.dist-info/METADATA,sha256=XNqS2AhxGP7T1IlcqDDr95zIChaW9hBIMhsVZoMd-bA,329
+austial_orm-0.1.5.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+austial_orm-0.1.5.dist-info/RECORD,,

austial_orm-0.1.5.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.31.0
+Root-Is-Purelib: true
+Tag: py3-none-any