nlbone 0.1.21__py3-none-any.whl → 0.1.23__py3-none-any.whl

nlbone/adapters/db/sqlalchemy/engine.py

@@ -0,0 +1,118 @@
+from contextlib import asynccontextmanager, contextmanager
+from typing import Generator, Optional, Any, AsyncGenerator
+
+from sqlalchemy import create_engine, text
+from sqlalchemy.engine import Engine
+from sqlalchemy.orm import Session, sessionmaker
+
+from sqlalchemy.ext.asyncio import (
+    AsyncEngine,
+    AsyncSession,
+    async_sessionmaker,
+    create_async_engine,
+)
+
+from nlbone.config.settings import get_settings
+
+_settings = get_settings()
+
+ASYNC_DSN: str = _settings.POSTGRES_DB_DSN
+
+if "+asyncpg" in ASYNC_DSN:
+    SYNC_DSN: str = ASYNC_DSN.replace("+asyncpg", "+psycopg")
+else:
+    SYNC_DSN = ASYNC_DSN
+
+_async_engine: Optional[AsyncEngine] = None
+_async_session_factory: Optional[async_sessionmaker[AsyncSession]] = None
+
+_sync_engine: Optional[Engine] = None
+_sync_session_factory: Optional[sessionmaker[Session]] = None
+
+
+def init_async_engine(echo: Optional[bool] = None) -> AsyncEngine:
+    global _async_engine, _async_session_factory
+    if _async_engine is not None:
+        return _async_engine
+
+    _async_engine = create_async_engine(
+        ASYNC_DSN,
+        echo=_settings.DEBUG if echo is None else echo,
+        pool_pre_ping=True,
+        pool_size=5,
+        max_overflow=10,
+    )
+    _async_session_factory = async_sessionmaker(
+        bind=_async_engine,
+        expire_on_commit=False,
+        autoflush=False,
+    )
+    return _async_engine
+
+
+@asynccontextmanager
+async def async_session() -> AsyncGenerator[AsyncSession, Any]:
+    if _async_session_factory is None:
+        init_async_engine()
+    assert _async_session_factory is not None
+    session = _async_session_factory()
+    try:
+        yield session
+        await session.commit()
+    except Exception:
+        await session.rollback()
+        raise
+    finally:
+        await session.close()
+
+
+async def async_ping() -> None:
+    eng = init_async_engine()
+    async with eng.connect() as conn:
+        await conn.execute(text("SELECT 1"))
+
+
+def init_sync_engine(echo: Optional[bool] = None) -> Engine:
+    global _sync_engine, _sync_session_factory
+    if _sync_engine is not None:
+        return _sync_engine
+
+    _sync_engine = create_engine(
+        SYNC_DSN,
+        echo=_settings.DEBUG if echo is None else echo,
+        pool_pre_ping=True,
+        pool_size=5,
+        max_overflow=10,
+        future=True,
+    )
+    _sync_session_factory = sessionmaker(
+        bind=_sync_engine,
+        autocommit=False,
+        autoflush=False,
+        expire_on_commit=False,
+        future=True,
+    )
+    return _sync_engine
+
+
+@contextmanager
+def sync_session() -> Generator[Session, None, None]:
+    if _sync_session_factory is None:
+        init_sync_engine()
+    assert _sync_session_factory is not None
+    s = _sync_session_factory()
+    try:
+        yield s
+        s.commit()
+    except Exception:
+        s.rollback()
+        raise
+    finally:
+        s.close()
+
+
+def sync_ping() -> None:
+    """Health check برای اتصال sync."""
+    eng = init_sync_engine()
+    with eng.connect() as conn:
+        conn.execute(text("SELECT 1"))

nlbone/config/settings.py

@@ -12,7 +12,7 @@ class Settings(BaseSettings):
     ENV: Literal["local", "dev", "staging", "prod"] = Field(default="local")
     DEBUG: bool = Field(default=False)
     LOG_LEVEL: Literal["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"] = Field(default="INFO")
-    LOG_JSON: bool = Field(default=True)  # اگر False باشد، فرمت متنی ساده می‌شود.
+    LOG_JSON: bool = Field(default=True)
 
     # ---------------------------
     # HTTP / Timeouts
@@ -30,10 +30,10 @@ class Settings(BaseSettings):
     # ---------------------------
     # Database
     # ---------------------------
-    DB_DSN: str = Field(default="postgresql+asyncpg://user:pass@localhost:5432/nlbone")
+    POSTGRES_DB_DSN: str = Field(default="postgresql+asyncpg://user:pass@localhost:5432/nlbone")
 
     # ---------------------------
-    # Messaging / Cache (نمونه)
+    # Messaging / Cache
     # ---------------------------
     REDIS_URL: str = Field(default="redis://localhost:6379/0")
 

nlbone/interfaces/api/dependencies.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+from typing import AsyncGenerator, Generator
+
+from sqlalchemy.orm import Session
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from nlbone.adapters.db.sqlalchemy.engine import async_session, sync_session
+
+
+async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
+    async with async_session() as s:
+        yield s
+
+def get_session() -> Generator[Session, None, None]:
+    with sync_session() as s:
+        yield s

{nlbone-0.1.21.dist-info → nlbone-0.1.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nlbone
-Version: 0.1.21
+Version: 0.1.23
 Summary: Backbone package for interfaces and infrastructure in Python projects
 Author-email: Amir Hosein Kahkbazzadeh <a.khakbazzadeh@gmail.com>
 License: MIT
@@ -21,6 +21,7 @@ Requires-Dist: pre-commit>=3.7; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
 Requires-Dist: pytest>=8.0; extra == 'dev'
 Requires-Dist: ruff>=0.5; extra == 'dev'
+Requires-Dist: tomli; extra == 'dev'
 Requires-Dist: twine; extra == 'dev'
 Description-Content-Type: text/markdown
 

{nlbone-0.1.21.dist-info → nlbone-0.1.23.dist-info}/RECORD

@@ -8,6 +8,7 @@ nlbone/adapters/db/memory.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 nlbone/adapters/db/postgres.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/adapters/db/query_builder.py,sha256=qL2Ppe39X7pnKpHfWZZanxfEKqQW3grZv5bQLS52mfU,6066
 nlbone/adapters/db/sqlalchemy/__init__.py,sha256=bq_dIuk4JqT-68k6p0aMb7DY2GqcsvvsuZT5uZDuTBU,127
+nlbone/adapters/db/sqlalchemy/engine.py,sha256=baEByNlB8NFuGVnMMz7xJqNa1_1E7DbLJ0vUo9vFarg,3045
 nlbone/adapters/db/sqlalchemy/query/__init__.py,sha256=9SFkoNkklaji6kZt_Nl6X6vqMFE92GiqW36Zezr5PuE,129
 nlbone/adapters/db/sqlalchemy/query/builder.py,sha256=AzUX2MlfrMGY-EITzVJoq9naM5sTL9h9yzI7cCeEBRk,886
 nlbone/adapters/db/sqlalchemy/query/coercion.py,sha256=uYClfEuew_W4r-vbwj2c5MVpayejhpOMl6h4iUrMotQ,2142
@@ -20,7 +21,7 @@ nlbone/adapters/messaging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 nlbone/adapters/messaging/redis.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/config/logging.py,sha256=68WRQejEpL6eHEY_cOgdlOjndKc-RWth0n4YmXnceC8,5041
-nlbone/config/settings.py,sha256=N3o7FWAEx_aYyOzmoRH6muq4K39rQC-mWJJuHz6Ps90,1864
+nlbone/config/settings.py,sha256=f9zIRsWce8v8ZFI0dd4vBY9O-Oo7TKEBDT4wOA8bKcs,1791
 nlbone/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/core/{__init__.py},sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/core/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -38,6 +39,7 @@ nlbone/di/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/di/container.py,sha256=7-lcrH1q9NmNLY7VMpLfPJZYKsvNuXP2U6gOYcCwgPo,304
 nlbone/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/interfaces/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nlbone/interfaces/api/dependencies.py,sha256=IqDVq1lcCCxd22FBUg523lVANM_j71BYAQtsbrHc4M8,465
 nlbone/interfaces/api/exceptions.py,sha256=OczR1FND2nbCngwbfgaPrgDbDaz4Pc47mFSHgtL7tW8,313
 nlbone/interfaces/api/routers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/interfaces/api/schemas.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -49,7 +51,7 @@ nlbone/interfaces/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 nlbone/interfaces/jobs/sync_tokens.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/utils/time.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nlbone-0.1.21.dist-info/METADATA,sha256=L89KPfF_G7h-Oh5_VdrING8ycju2yIphCNYKAcKS554,2189
-nlbone-0.1.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nlbone-0.1.21.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nlbone-0.1.21.dist-info/RECORD,,
+nlbone-0.1.23.dist-info/METADATA,sha256=3hwLuYJE0xbwf5ZZXeubHEO9F45CT4FDiCgMdklj81w,2226
+nlbone-0.1.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nlbone-0.1.23.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nlbone-0.1.23.dist-info/RECORD,,