hajimi-sqlalchemy 0.0.1__py3-none-any.whl

hajimi_sqlalchemy/__init__.py

@@ -0,0 +1,4 @@
+from .base import BaseModel
+from .config import HajimiSQLAlchemyConfig
+from .dependencies import SessionDep
+from .plugin import HajimiSQLAlchemyPlugin

hajimi_sqlalchemy/base.py

@@ -0,0 +1,5 @@
+from sqlalchemy.orm import DeclarativeBase
+
+
+class BaseModel(DeclarativeBase):
+    __abstract__ = True

hajimi_sqlalchemy/config.py

@@ -0,0 +1,29 @@
+from dataclasses import dataclass
+
+from sqlalchemy import URL
+
+
+@dataclass(frozen=True, kw_only=True)
+class HajimiSQLAlchemyConfig:
+    username: str
+    password: str
+    host: str
+    port: int = 5432
+    database: str
+
+    echo: bool = False
+    pool_size: int = 3
+    max_overflow: int = 2
+    pool_timeout: int = 30
+    pool_recycle: int = 1800
+
+    @property
+    def url(self) -> URL:
+        return URL.create(
+            drivername='postgresql+psycopg',
+            username=self.username,
+            password=self.password,
+            host=self.host,
+            port=self.port,
+            database=self.database,
+        )

hajimi_sqlalchemy/dependencies.py

@@ -0,0 +1,20 @@
+"""FastAPI dependencies exposing the SQLAlchemy async session."""
+
+from typing import Annotated, AsyncGenerator
+
+from fastapi import Depends, Request
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from .plugin import HajimiSQLAlchemyPlugin
+
+
+async def get_session(request: Request) -> AsyncGenerator[AsyncSession, None]:
+    """Yield an ``AsyncSession`` scoped to the request, opened from the registered plugin."""
+    app_state = request.app.state._state['__app_state__']
+    plugin: HajimiSQLAlchemyPlugin = getattr(app_state, HajimiSQLAlchemyPlugin.__name__)
+
+    async with plugin.session_maker.begin() as session:
+        yield session
+
+
+SessionDep = Annotated[AsyncSession, Depends(get_session)]

hajimi_sqlalchemy/plugin.py

@@ -0,0 +1,44 @@
+from fastapi import FastAPI
+from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker, create_async_engine
+
+from .config import HajimiSQLAlchemyConfig
+
+
+class HajimiSQLAlchemyPlugin:
+    def __init__(self, config: HajimiSQLAlchemyConfig) -> None:
+        self._config = config
+        self._engine: AsyncEngine | None = None
+        self._session_maker: async_sessionmaker[AsyncSession] | None = None
+
+    @property
+    def session_maker(self) -> async_sessionmaker[AsyncSession]:
+        if not isinstance(self._session_maker, async_sessionmaker):
+            raise RuntimeError(
+                'HajimiSQLAlchemyPlugin.session_maker is unavailable: '
+                'startup() has not been called or shutdown() has already run.'
+            )
+    
+        return self._session_maker
+
+    async def startup(self, app: FastAPI) -> None:
+        self._engine = create_async_engine(
+            url=self._config.url,
+            echo=self._config.echo,
+            pool_size=self._config.pool_size,
+            max_overflow=self._config.max_overflow,
+            pool_timeout=self._config.pool_timeout,
+            pool_recycle=self._config.pool_recycle,
+            pool_pre_ping=True,
+            connect_args={'connect_timeout': 6},
+        )
+        self._session_maker = async_sessionmaker(engine=self._engine)
+        app.state.__app__state[self.__name__] = self
+
+    async def shutdown(self, app: FastAPI) -> None:
+        if self._session_maker is not None:
+            self._session_maker = None
+        if self._engine is not None:
+            await self._engine.dispose()
+            self._engine = None
+
+        app.state.__app__state.pop(self.__name__, None)

hajimi_sqlalchemy-0.0.1.dist-info/METADATA

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: hajimi-sqlalchemy
+Version: 0.0.1
+Requires-Python: >=3.14
+License-File: LICENSE
+Requires-Dist: hajimi-core>=0.1.0
+Requires-Dist: sqlalchemy[asyncio,postgresql-psycopg]>=2.0.49
+Dynamic: license-file

hajimi_sqlalchemy-0.0.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+hajimi_sqlalchemy/__init__.py,sha256=zNelzQf6mpE3P9adR7oWuDMQ0QJI9LFVRtOAXWFpNWE,151
+hajimi_sqlalchemy/base.py,sha256=atu01ommpKNd6MbNlta39RpRr4gsit67eyp9kmWnYZ8,103
+hajimi_sqlalchemy/config.py,sha256=A_8mvzlKk0uzWGWe-P9z_clq11LvG8dKyk_gYfjJJu4,637
+hajimi_sqlalchemy/dependencies.py,sha256=VJBYJ7sQt8h6qGf3xqa0OoQ6UwkUOUh6HzkihSsIIlI,704
+hajimi_sqlalchemy/plugin.py,sha256=EE0Ce4rAkJz-NGY0ReS_jYY47XFi7SFsWm6RZwMpvUI,1694
+hajimi_sqlalchemy-0.0.1.dist-info/licenses/LICENSE,sha256=2N6aWoJwaqUS051iv3_th5CYm77Wv2yjZ7hfk4q1SXI,1063
+hajimi_sqlalchemy-0.0.1.dist-info/METADATA,sha256=ciuDzjYAtUWvNHJ06VXhaQfZNtPjdNxZ76lSJZ4Yt7Q,225
+hajimi_sqlalchemy-0.0.1.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+hajimi_sqlalchemy-0.0.1.dist-info/top_level.txt,sha256=ttsxvMfRWlYAHZMZWBe4tnVu8pQiBAEJVyH7HkaT3L0,18
+hajimi_sqlalchemy-0.0.1.dist-info/RECORD,,

hajimi_sqlalchemy-0.0.1.dist-info/WHEEL

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

hajimi_sqlalchemy-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 huruhi
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

hajimi_sqlalchemy-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+hajimi_sqlalchemy