wheke-sqlmodel 0.3.0__py3-none-any.whl

wheke_sqlmodel/__init__.py

@@ -0,0 +1,11 @@
+from ._pod import sqlmodel_pod
+from ._service import SQLModelService, get_sqlmodel_service
+from ._settings import SQLITE_DRIVER, SQLModelSettings
+
+__all__ = [
+    "SQLITE_DRIVER",
+    "SQLModelService",
+    "SQLModelSettings",
+    "get_sqlmodel_service",
+    "sqlmodel_pod",
+]

wheke_sqlmodel/_cli.py

@@ -0,0 +1,30 @@
+import anyio
+import typer
+from rich.console import Console
+from typer import Context
+from wheke import get_container
+
+from ._service import get_sqlmodel_service
+
+cli = typer.Typer(short_help="SQLModel commands")
+console = Console()
+
+
+@cli.command()
+def create_db(ctx: Context) -> None:
+    container = get_container(ctx)
+    service = get_sqlmodel_service(container)
+
+    console.print("Creating database...")
+
+    anyio.run(service.create_db)
+
+
+@cli.command()
+def drop_db(ctx: Context) -> None:
+    container = get_container(ctx)
+    service = get_sqlmodel_service(container)
+
+    console.print("Droping database...")
+
+    anyio.run(service.drop_db)

wheke_sqlmodel/_pod.py

@@ -0,0 +1,17 @@
+from wheke import Pod, ServiceConfig
+
+from ._cli import cli
+from ._service import SQLModelService, sqlmodel_service_factory
+
+sqlmodel_pod = Pod(
+    "sqlmodel",
+    services=[
+        ServiceConfig(
+            SQLModelService,
+            sqlmodel_service_factory,
+            is_singleton=True,
+            singleton_cleanup_method="dispose",
+        ),
+    ],
+    cli=cli,
+)

wheke_sqlmodel/_service.py

@@ -0,0 +1,50 @@
+from collections.abc import AsyncGenerator
+from contextlib import asynccontextmanager
+
+from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
+from sqlmodel import SQLModel
+from sqlmodel.ext.asyncio.session import AsyncSession
+from svcs import Container
+from wheke import WhekeSettings, get_service, get_settings
+
+from ._settings import SQLModelSettings
+
+
+class SQLModelService:
+    engine: AsyncEngine
+
+    def __init__(self, *, settings: SQLModelSettings) -> None:
+        self.engine = create_async_engine(
+            settings.connection_string,
+            connect_args={
+                "check_same_thread": False,
+            },
+            echo=settings.echo_operations,
+        )
+
+    @property
+    @asynccontextmanager
+    async def session(self) -> AsyncGenerator[AsyncSession]:
+        async with AsyncSession(self.engine) as _session:
+            yield _session
+
+    async def create_db(self) -> None:
+        async with self.engine.begin() as conn:
+            await conn.run_sync(SQLModel.metadata.create_all)
+
+    async def drop_db(self) -> None:
+        async with self.engine.begin() as conn:
+            await conn.run_sync(SQLModel.metadata.drop_all)
+
+    async def dispose(self) -> None:
+        await self.engine.dispose()
+
+
+def sqlmodel_service_factory(container: Container) -> SQLModelService:
+    settings = get_settings(container, WhekeSettings).get_feature(SQLModelSettings)
+
+    return SQLModelService(settings=settings)
+
+
+def get_sqlmodel_service(container: Container) -> SQLModelService:
+    return get_service(container, SQLModelService)

wheke_sqlmodel/_settings.py

@@ -0,0 +1,13 @@
+from typing import ClassVar
+
+from wheke import FeatureSettings
+
+SQLITE_DRIVER = "sqlite+aiosqlite"
+
+
+class SQLModelSettings(FeatureSettings):
+    __feature_name__: ClassVar[str] = "sqlmodel"
+
+    connection_string: str = f"{SQLITE_DRIVER}:///database.db"
+
+    echo_operations: bool = False

wheke_sqlmodel-0.3.0.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: wheke-sqlmodel
+Version: 0.3.0
+Summary: Add sql capabilities to wheke with SQLModel
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: aiosqlite
+Requires-Dist: anyio
+Requires-Dist: sqlmodel
+Requires-Dist: wheke
+
+# Wheke SQLModel
+
+Add sql capabilities to wheke with SQLModel

wheke_sqlmodel-0.3.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+wheke_sqlmodel/__init__.py,sha256=gm5QT4sfPbaleU3XoeHHEnIdrKJj7jiZy_FNxmBPRyg,277
+wheke_sqlmodel/_cli.py,sha256=kEdsK-bzrY59B0XhmIvVSbWRCaFosYlVb9DstJjPa_c,651
+wheke_sqlmodel/_pod.py,sha256=41x9ReGxjH3DFt4jSYpwORpTaF-qWl_P2-bpW0klEk0,378
+wheke_sqlmodel/_service.py,sha256=Mmp2sd9-7cm8HbGD1FjD62P39mwkgrrE_dk_gIWNI4g,1581
+wheke_sqlmodel/_settings.py,sha256=7ibjtz8X3O8otRGuyIj62lswrcVROztnyHxzDhjKvI0,290
+wheke_sqlmodel/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wheke_sqlmodel-0.3.0.dist-info/METADATA,sha256=-X_C9MDbFE7D0OF5jGLNZrcZA-Kc04usT5t6CtGhZ0I,329
+wheke_sqlmodel-0.3.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+wheke_sqlmodel-0.3.0.dist-info/top_level.txt,sha256=xn3IbC1JQzfXO8TkB3mB5AdkRJ53uNhOCawQmj_Kh9E,15
+wheke_sqlmodel-0.3.0.dist-info/RECORD,,

wheke_sqlmodel-0.3.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

wheke_sqlmodel-0.3.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+wheke_sqlmodel