wheke-sqlmodel 0.3.0__tar.gz

wheke_sqlmodel-0.3.0/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,3 @@
+# Wheke SQLModel
+
+Add sql capabilities to wheke with SQLModel

wheke_sqlmodel-0.3.0/pyproject.toml

@@ -0,0 +1,129 @@
+[project]
+name = "wheke-sqlmodel"
+version = "0.3.0"
+description = "Add sql capabilities to wheke with SQLModel"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+  "aiosqlite",
+  "anyio",
+  "sqlmodel",
+  "wheke",
+]
+
+[dependency-groups]
+dev = [
+  "httpx",
+  "isort",
+  "mypy",
+  "pytest",
+  "pytest-cov",
+  "ruff",
+]
+
+[tool.pytest.ini_options]
+pythonpath = [
+  "src"
+]
+
+[tool.coverage.run]
+relative_files = true
+branch = true
+parallel = true
+omit = []
+
+[tool.coverage.report]
+exclude_lines = [
+  "no cov",
+  "if __name__ == .__main__.:",
+  "if TYPE_CHECKING:",
+]
+
+[tool.isort]
+profile = "black"
+skip = [
+  ".env",
+  ".hatch",
+  ".mypy_cache",
+  ".venv",
+  "__pycache__",
+  "env",
+  "venv",
+]
+
+[tool.mypy]
+disallow_incomplete_defs = true
+check_untyped_defs = true
+warn_unused_ignores = true
+exclude = """
+.env
+| .hatch
+| .mypy_cache
+| .venv
+| __pycache__
+| env
+| venv
+"""
+
+[tool.ruff]
+exclude = [
+  ".env",
+  ".hatch",
+  ".mypy_cache",
+  ".venv",
+  "__pycache__",
+  "env",
+  "venv",
+]
+
+[tool.ruff.lint]
+select = [
+  "A",
+  "ARG",
+  "B",
+  "C",
+  "DTZ",
+  "E",
+  "EM",
+  "F",
+  "FBT",
+  "I",
+  "ICN",
+  "ISC",
+  "N",
+  "PLC",
+  "PLE",
+  "PLR",
+  "PLW",
+  "Q",
+  "RUF",
+  "S",
+  "T",
+  "TID",
+  "UP",
+  "W",
+  "YTT",
+]
+ignore = [
+  # Same line string implicit string concatenation
+  "ISC001",
+  # Allow non-abstract empty methods in abstract base classes
+  "B027",
+  # Ignore checks for possible passwords
+  "S105", "S106", "S107",
+  # Ignore complexity
+  "C901", "PLR0911", "PLR0912", "PLR0913", "PLR0915",
+  # Relative imports
+  "TID252",
+]
+unfixable = [
+  # Don't touch unused imports
+  "F401",
+]
+
+[tool.ruff.lint.isort]
+known-first-party = ["wheke_sqlmodel"]
+
+[tool.ruff.lint.per-file-ignores]
+# Tests can use magic values, assertions, and relative imports
+"tests/**/*" = ["PLR2004", "S101", "TID252"]

wheke_sqlmodel-0.3.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

wheke_sqlmodel-0.3.0/src/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-0.3.0/src/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-0.3.0/src/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-0.3.0/src/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-0.3.0/src/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/src/wheke_sqlmodel.egg-info/PKG-INFO

@@ -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/src/wheke_sqlmodel.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+README.md
+pyproject.toml
+src/wheke_sqlmodel/__init__.py
+src/wheke_sqlmodel/_cli.py
+src/wheke_sqlmodel/_pod.py
+src/wheke_sqlmodel/_service.py
+src/wheke_sqlmodel/_settings.py
+src/wheke_sqlmodel/py.typed
+src/wheke_sqlmodel.egg-info/PKG-INFO
+src/wheke_sqlmodel.egg-info/SOURCES.txt
+src/wheke_sqlmodel.egg-info/dependency_links.txt
+src/wheke_sqlmodel.egg-info/requires.txt
+src/wheke_sqlmodel.egg-info/top_level.txt
+tests/test_app.py

wheke_sqlmodel-0.3.0/src/wheke_sqlmodel.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

wheke_sqlmodel-0.3.0/src/wheke_sqlmodel.egg-info/requires.txt

@@ -0,0 +1,4 @@
+aiosqlite
+anyio
+sqlmodel
+wheke

wheke_sqlmodel-0.3.0/src/wheke_sqlmodel.egg-info/top_level.txt

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

wheke_sqlmodel-0.3.0/tests/test_app.py

@@ -0,0 +1,12 @@
+from fastapi import status
+from fastapi.testclient import TestClient
+
+
+def test_list_entries(client: TestClient) -> None:
+    response = client.post("/entries", json={"value": "test"})
+    assert response.status_code == status.HTTP_201_CREATED
+
+    response = client.get("/entries")
+    assert response.status_code == status.HTTP_200_OK
+
+    assert len(response.json()) == 1