sotkalib 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

sotkalib/sqla/db.py

@@ -0,0 +1,101 @@
+from pydantic import BaseModel, Field
+from sqlalchemy import Engine, create_engine
+from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, create_async_engine
+from sqlalchemy.ext.asyncio.session import AsyncSession
+from sqlalchemy.orm import Session, sessionmaker
+
+from sotkalib.log import get_logger
+
+
+class ConnectionTimeoutError(Exception): pass
+
+class DatabaseSettings(BaseModel):
+    uri: str = Field(examples=[
+        "postgresql://username:password@localhost:5432/database"
+    ])
+    async_driver: str = "asyncpg"
+    echo: bool = False
+    pool_size: int = 10
+
+    @property
+    def async_uri(self) -> str:
+        return self.uri.replace("postgresql://", "postgresql" + self.async_driver + "://")
+
+class Database:
+    _sync_engine: Engine | None
+    _async_engine: AsyncEngine | None
+    _sync_session_factory: sessionmaker = None
+    _async_session_factory: async_sessionmaker = None
+
+    logger = get_logger("sqldb.instance")
+
+    def __init__(self, settings: DatabaseSettings):
+        self.__async_uri = settings.async_uri
+        self.__sync_uri = settings.uri
+        self.echo = settings.echo
+        self.pool_size = settings.pool_size
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self._sync_engine:
+            self._sync_engine.dispose()
+            self.logger.info("closed sync db connection")
+
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, *args):
+        if self._async_engine:
+            await self._async_engine.dispose()
+            self.logger.info("closed async db connection")
+
+    def __async_init(self):
+        self._async_engine = create_async_engine(
+            url=self.__async_uri,
+            echo=self.echo,
+            pool_size=self.pool_size,
+        )
+        self._async_session_factory = async_sessionmaker(bind=self._async_engine, expire_on_commit=False)
+        self.logger.debug(  # noqa: PLE1205
+            "successfully initialized async db connection, engine.status = {} sessionmaker.status = {}",
+            self._async_engine.name is not None,
+            self._async_session_factory is not None,
+            )
+
+    @property
+    def async_session(self) -> async_sessionmaker[AsyncSession]:
+        if self._async_engine is None or self._async_session_factory is None:
+            self.logger.debug("async_sf not found, initializing")
+            self.__async_init()
+            if self._async_engine is None or self._async_session_factory is None:
+                self.logger.error(c := "could not asynchronously connect to pgsql")
+                raise ConnectionTimeoutError(c)
+        self.logger.debug("success getting (asyncmaker)")
+        return self._async_session_factory
+
+    def __sync_init(self):
+        self._sync_engine = create_engine(
+            url=self.__sync_uri,
+            echo=self.echo,
+            pool_size=self.pool_size,
+        )
+        self._sync_session_factory = sessionmaker(bind=self._sync_engine, expire_on_commit=False)
+        self.logger.debug(  # noqa
+            " -> (__sync_init) successfully initialized sync db connection,\n"
+            "\t\t\t\tengine.status = {} sessionmaker.status = {}",
+            self._sync_engine.name is not None,
+            self._sync_session_factory is not None,
+            )
+
+    @property
+    def session(self) -> sessionmaker[Session]:
+        if self._sync_engine is None or self._sync_session_factory is None:
+            self.logger.debug("not found, initializing...")
+            self.__sync_init()
+            if self._sync_engine is None or self._sync_session_factory is None:
+                self.logger.error(c := "could not synchronously connect to pgsql")
+                raise ConnectionTimeoutError(c)
+        self.logger.debug("success getting (syncmaker)")
+        return self._sync_session_factory

{sotkalib-0.0.2.dist-info → sotkalib-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sotkalib
-Version: 0.0.2
+Version: 0.0.4
 Summary: 
 Author: alexey
 Author-email: alexey <me@pyrorhythm.dev>
@@ -8,6 +8,8 @@ Requires-Dist: aiohttp>=3.13.3
 Requires-Dist: dotenv>=0.9.9
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: pydantic>=2.12.5
+Requires-Dist: redis>=7.1.0
+Requires-Dist: sqlalchemy[asyncio]>=2.0.46
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 

sotkalib-0.0.4.dist-info/RECORD

@@ -0,0 +1,25 @@
+sotkalib/__init__.py,sha256=t3dEAlrtHABFRZdF7RM5K2N1fDUTVwUPmBPG4jFdvSY,138
+sotkalib/config/__init__.py,sha256=CSjn02NCnBPO14QOg4OzKI-lTxyKoBxQ4ODsiWamlIM,102
+sotkalib/config/field.py,sha256=596_6DLnUIMN86h2uob9YN5IrHnc7fguWScSbIYsTH4,326
+sotkalib/config/struct.py,sha256=gv1jFrSRytMC6bZTUDOUQf00Zo1iKxQvuTxGv9qnyHI,5679
+sotkalib/enum/__init__.py,sha256=pKpLPm8fqHO4Et21TWIybIPRiehN1KrmxcBh6hPRsxM,127
+sotkalib/enum/mixins.py,sha256=rgXb0eXaBSozrviOMJo1671x4DiN9SELtw3-x6PvhDM,1821
+sotkalib/exceptions/__init__.py,sha256=H2h-yW1o0_X1Z9O-hWgj6h2spFxpzJJQ_N2ITMr600A,58
+sotkalib/exceptions/api/__init__.py,sha256=tIFOiRlbPkgCRNv5OPZ1M98nRnAMkFIuqSK7dZpKMRI,38
+sotkalib/exceptions/api/exc.py,sha256=gqx4GrHXUvKcR7tEmJpRqPbDWOT2AgKoyck8-FovQCc,1329
+sotkalib/exceptions/handlers/__init__.py,sha256=rA6o6_LVa-0TToyPhT1vB_Lz2E0U8EUfHICAYgAUd78,178
+sotkalib/exceptions/handlers/args_incl_error.py,sha256=DB8TMhZdSkwZwEahOJ99zXWknqBTkSC4IKZsZ5psxdg,535
+sotkalib/exceptions/handlers/core.py,sha256=5fhusoxBhUz59TaVWobplBvD-sbkZKBnmmu-fcSyRk4,836
+sotkalib/http/__init__.py,sha256=HxOuGbHbz39MD0ICjOSh4zv-nGPZca9TgxP4rCceamw,241
+sotkalib/http/client_session.py,sha256=OtGTpwr8I7tsgOxRGS8c7tby5sGT_T6B0Zc8HTC9Sxs,8838
+sotkalib/log/__init__.py,sha256=xrBx--c8QU5xkb3_n61LuqF8ySUaxlQkHCxHyH_D8aE,58
+sotkalib/log/factory.py,sha256=Wl8qY2-vimpctRlRYSWPLjC0KgeEGgSSuDDJaxWtvK8,309
+sotkalib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sotkalib/redis/__init__.py,sha256=wv3AIRw3aXXLgTTrt6my8C8jSBl4-R3dWRfGLbGTHng,167
+sotkalib/redis/client.py,sha256=0TWe-gYqFiuCjqimCQrVrnTHSM0EROIoJL36M3qwOtQ,1118
+sotkalib/redis/lock.py,sha256=nEZjIyXmgq3vH-Urs8qXC_N8lmXNho00SaTZ7wJIEIo,2528
+sotkalib/sqla/__init__.py,sha256=fYT8O-bPcdXxJ3QVu3KbypwbZ_hpm8Eq1CuQgjvyNJ8,86
+sotkalib/sqla/db.py,sha256=lxE6XLgX-CowyhfRwTAA_FGJybiihaGWWzlNe8M41CE,3866
+sotkalib-0.0.4.dist-info/WHEEL,sha256=5DEXXimM34_d4Gx1AuF9ysMr1_maoEtGKjaILM3s4w4,80
+sotkalib-0.0.4.dist-info/METADATA,sha256=4OxxOFpmiDXvjlOjQveYAlGtkXS1P8WXNZz0jchKDVY,380
+sotkalib-0.0.4.dist-info/RECORD,,

{sotkalib-0.0.2.dist-info → sotkalib-0.0.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.28
+Generator: uv 0.9.29
 Root-Is-Purelib: true
 Tag: py3-none-any

sotkalib-0.0.2.dist-info/RECORD

@@ -1,12 +0,0 @@
-sotkalib/__init__.py,sha256=ebrSgK8GtSh2aZNdP3vKA6A4b1iHLFNWzpgznyOOn8Q,48
-sotkalib/config/__init__.py,sha256=CSjn02NCnBPO14QOg4OzKI-lTxyKoBxQ4ODsiWamlIM,102
-sotkalib/config/field.py,sha256=UaIiZBa4TUOB-akEqIvFJKxj2QjjsTaAjL2r163N-V4,790
-sotkalib/config/struct.py,sha256=kUabPNOv9uA8d0SzBOfr-w08YOYKy-D8LN0sof_ZQ5Q,6453
-sotkalib/http/__init__.py,sha256=gkuf1ZM7qXRmnm-yDUMmxxbCuu7bLek_PL3UmMQBtlM,317
-sotkalib/http/client_session.py,sha256=9AyOLFQFXf_lkirJcfp0NBIzx5owdRtDJZM-PEcCKtI,8952
-sotkalib/log/__init__.py,sha256=xrBx--c8QU5xkb3_n61LuqF8ySUaxlQkHCxHyH_D8aE,58
-sotkalib/log/factory.py,sha256=N8SZNvCdBLmsM0ES38x2BxZbJD1GgdMHtJ8h8lZjv7A,806
-sotkalib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sotkalib-0.0.2.dist-info/WHEEL,sha256=fAguSjoiATBe7TNBkJwOjyL1Tt4wwiaQGtNtjRPNMQA,80
-sotkalib-0.0.2.dist-info/METADATA,sha256=TOF1FK1cVfMXt1kXc4gVjFsFsGj8hi5uMdQkude7sMQ,309
-sotkalib-0.0.2.dist-info/RECORD,,