maleo-database 0.0.21__py3-none-any.whl → 0.0.24__py3-none-any.whl

maleo/database/handlers.py

@@ -1,8 +1,5 @@
 from pydantic import BaseModel, ConfigDict, Field
-from sqlalchemy import MetaData
 from typing import Generic, Optional, TypeVar
-from maleo.dtos.contexts.service import ServiceContext
-from maleo.logging.logger import Database
 from .config import (
     ElasticsearchConfig,
     MongoConfig,
@@ -23,6 +20,7 @@ from .managers import (
     SQLServerManager,
     ManagerT,
 )
+from .types import DeclarativeBaseT
 
 
 class Handler(
@@ -42,17 +40,7 @@ HandlerT = TypeVar("HandlerT", bound=Handler)
 
 
 class ElasticsearchHandler(Handler[ElasticsearchConfig, ElasticsearchManager]):
-    @classmethod
-    def new(
-        cls,
-        config: ElasticsearchConfig,
-        logger: Database,
-        service_context: Optional[ServiceContext] = None,
-    ) -> "ElasticsearchHandler":
-        manager = ElasticsearchManager(
-            config=config, logger=logger, service_context=service_context
-        )
-        return cls(config=config, manager=manager)
+    pass
 
 
 class MongoHandler(Handler[MongoConfig, MongoManager]):
@@ -63,33 +51,29 @@ class RedisHandler(Handler[RedisConfig, RedisManager]):
     pass
 
 
-class MySQLHandler(Handler[MySQLConfig, MySQLManager]):
+class MySQLHandler(
+    Handler[MySQLConfig, MySQLManager[DeclarativeBaseT]], Generic[DeclarativeBaseT]
+):
+    pass
+
+
+class PostgreSQLHandler(
+    Handler[PostgreSQLConfig, PostgreSQLManager[DeclarativeBaseT]],
+    Generic[DeclarativeBaseT],
+):
     pass
 
 
-class PostgreSQLHandler(Handler[PostgreSQLConfig, PostgreSQLManager]):
-    @classmethod
-    def new(
-        cls,
-        config: PostgreSQLConfig,
-        logger: Database,
-        metadata: MetaData,
-        service_context: Optional[ServiceContext] = None,
-    ) -> "PostgreSQLHandler":
-        manager = PostgreSQLManager(
-            config=config,
-            logger=logger,
-            metadata=metadata,
-            service_context=service_context,
-        )
-        return cls(config=config, manager=manager)
-
-
-class SQLiteHandler(Handler[SQLiteConfig, SQLiteManager]):
+class SQLiteHandler(
+    Handler[SQLiteConfig, SQLiteManager[DeclarativeBaseT]], Generic[DeclarativeBaseT]
+):
     pass
 
 
-class SQLServerHandler(Handler[SQLServerConfig, SQLServerManager]):
+class SQLServerHandler(
+    Handler[SQLServerConfig, SQLServerManager[DeclarativeBaseT]],
+    Generic[DeclarativeBaseT],
+):
     pass
 
 
@@ -190,7 +174,7 @@ class Handlers(
     ],
 ):
     nosql: NoSQLHandlersT = Field(..., description="NoSQL handlers")
-    sql: NoSQLHandlersT = Field(..., description="SQL handlers")
+    sql: SQLHandlersT = Field(..., description="SQL handlers")
 
 
 HandlersT = TypeVar("HandlersT", bound=Optional[Handlers])

maleo/database/managers/__init__.py

@@ -6,8 +6,8 @@ from motor.motor_asyncio import AsyncIOMotorClient
 from pymongo import MongoClient
 from redis.asyncio import Redis as AsyncRedis
 from redis import Redis as SyncRedis
-from sqlalchemy import MetaData, text
-from typing import Generic, Optional, TypeVar
+from sqlalchemy import text
+from typing import Generic, Optional, Type, TypeVar
 from uuid import uuid4
 from maleo.dtos.authentication import GenericAuthentication
 from maleo.dtos.contexts.operation import generate_operation_context
@@ -43,6 +43,7 @@ from ..config import (
     ConfigT,
 )
 from ..enums import Connection
+from ..types import DeclarativeBaseT
 from .client import (
     AsyncClientT,
     SyncClientT,
@@ -96,16 +97,15 @@ class Manager(ABC, Generic[ConfigT]):
         pass
 
 
-class SQLManager(Manager[SQLConfigT], Generic[SQLConfigT]):
+class SQLManager(Manager[SQLConfigT], Generic[SQLConfigT, DeclarativeBaseT]):
     def __init__(
         self,
+        Base: Type[DeclarativeBaseT],
         config: SQLConfigT,
         logger: Database,
-        metadata: MetaData,
         service_context: Optional[ServiceContext] = None,
     ) -> None:
         super().__init__(config, logger, service_context)
-        self._metadata = metadata
         self._operation_context.target.details = self._config.model_dump()
         self._engine_manager = EngineManager[SQLConfigT](config)
         self._session_manager = SessionManager(
@@ -114,7 +114,8 @@ class SQLManager(Manager[SQLConfigT], Generic[SQLConfigT]):
             logger=self._logger,
             service_context=self._service_context,
         )
-        self._metadata.create_all(bind=self._engine_manager.get(Connection.SYNC))
+        self.Base = Base
+        self.Base.metadata.create_all(bind=self._engine_manager.get(Connection.SYNC))
 
     @property
     def engine(self) -> EngineManager[SQLConfigT]:
@@ -251,19 +252,27 @@ class SQLManager(Manager[SQLConfigT], Generic[SQLConfigT]):
         await self._engine_manager.dispose()
 
 
-class MySQLManager(SQLManager[MySQLConfig]):
+class MySQLManager(
+    SQLManager[MySQLConfig, DeclarativeBaseT], Generic[DeclarativeBaseT]
+):
     pass
 
 
-class PostgreSQLManager(SQLManager[PostgreSQLConfig]):
+class PostgreSQLManager(
+    SQLManager[PostgreSQLConfig, DeclarativeBaseT], Generic[DeclarativeBaseT]
+):
     pass
 
 
-class SQLiteManager(SQLManager[SQLiteConfig]):
+class SQLiteManager(
+    SQLManager[SQLiteConfig, DeclarativeBaseT], Generic[DeclarativeBaseT]
+):
     pass
 
 
-class SQLServerManager(SQLManager[SQLServerConfig]):
+class SQLServerManager(
+    SQLManager[SQLServerConfig, DeclarativeBaseT], Generic[DeclarativeBaseT]
+):
     pass
 
 

maleo/database/orm/queries.py

@@ -1,9 +1,9 @@
 from sqlalchemy import asc, cast, desc, or_, select
-from sqlalchemy.orm import DeclarativeBase, Session, aliased
+from sqlalchemy.orm import Session, aliased
 from sqlalchemy.orm.attributes import InstrumentedAttribute
 from sqlalchemy.sql import Select
 from sqlalchemy.types import DATE, String, TEXT, TIMESTAMP
-from typing import Sequence, Type, TypeVar
+from typing import Any, Sequence, Tuple, Type, TypeVar
 from maleo.enums.sort import Order
 from maleo.mixins.general import DateFilter, SortColumn
 from maleo.types.base.any import OptionalAny
@@ -11,18 +11,21 @@ from maleo.types.base.boolean import OptionalBoolean
 from maleo.types.base.integer import OptionalListOfIntegers
 from maleo.types.base.string import OptionalListOfStrings, OptionalString
 from maleo.types.enums.status import OptionalListOfDataStatuses
+from ..types import DeclarativeBaseT
 
 
-T = TypeVar("T", bound=DeclarativeBase)
+R = TypeVar(
+    "R", bound=Tuple[Any, ...]
+)  # result shape inferred from Select (tuple[...])
 
 
 def filter_column(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     column: str,
     value: OptionalAny = None,
     include_null: bool = False,
-) -> Select:
+) -> Select[R]:
     column_attr = getattr(table, column, None)
     if column_attr is None or not isinstance(column_attr, InstrumentedAttribute):
         return stmt
@@ -40,12 +43,12 @@ def filter_column(
 
 
 def filter_ids(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     column: str,
     ids: OptionalListOfIntegers = None,
     include_null: bool = False,
-) -> Select:
+) -> Select[R]:
     column_attr = getattr(table, column, None)
     if column_attr is None or not isinstance(column_attr, InstrumentedAttribute):
         return stmt
@@ -63,10 +66,10 @@ def filter_ids(
 
 
 def filter_timestamps(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     date_filters: Sequence[DateFilter],
-) -> Select:
+) -> Select[R]:
     if date_filters:
         for date_filter in date_filters:
             try:
@@ -90,10 +93,10 @@ def filter_timestamps(
 
 
 def filter_statuses(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     statuses: OptionalListOfDataStatuses,
-) -> Select:
+) -> Select[R]:
     if statuses is not None:
         status_filters = [table.status == status for status in statuses]  # type: ignore
         stmt = stmt.filter(or_(*status_filters))
@@ -101,11 +104,11 @@ def filter_statuses(
 
 
 def filter_is_root(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     parent_column: str = "parent_id",
     is_root: OptionalBoolean = None,
-) -> Select:
+) -> Select[R]:
     parent_attr = getattr(table, parent_column, None)
     if parent_attr is None or not isinstance(parent_attr, InstrumentedAttribute):
         return stmt
@@ -118,12 +121,12 @@ def filter_is_root(
 
 def filter_is_parent(
     session: Session,
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     id_column: str = "id",
     parent_column: str = "parent_id",
     is_parent: OptionalBoolean = None,
-) -> Select:
+) -> Select[R]:
     id_attr = getattr(table, id_column, None)
     if id_attr is None or not isinstance(id_attr, InstrumentedAttribute):
         return stmt
@@ -139,11 +142,11 @@ def filter_is_parent(
 
 
 def filter_is_child(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     parent_column: str = "parent_id",
     is_child: OptionalBoolean = None,
-) -> Select:
+) -> Select[R]:
     parent_attr = getattr(table, parent_column, None)
     if parent_attr is None or not isinstance(parent_attr, InstrumentedAttribute):
         return stmt
@@ -156,12 +159,12 @@ def filter_is_child(
 
 def filter_is_leaf(
     session: Session,
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     id_column: str = "id",
     parent_column: str = "parent_id",
     is_leaf: OptionalBoolean = None,
-) -> Select:
+) -> Select[R]:
     id_attr = getattr(table, id_column, None)
     if id_attr is None or not isinstance(id_attr, InstrumentedAttribute):
         return stmt
@@ -177,11 +180,11 @@ def filter_is_leaf(
 
 
 def search(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     search: OptionalString = None,
     columns: OptionalListOfStrings = None,
-) -> Select:
+) -> Select[R]:
     if search is None:
         return stmt
 
@@ -211,10 +214,10 @@ def search(
 
 
 def sort(
-    stmt: Select,
-    table: Type[T],
+    stmt: Select[R],
+    table: Type[DeclarativeBaseT],
     sort_columns: Sequence[SortColumn],
-) -> Select:
+) -> Select[R]:
     for sort_column in sort_columns:
         try:
             sort_col = getattr(table, sort_column.name)
@@ -227,7 +230,7 @@ def sort(
     return stmt
 
 
-def paginate(stmt: Select, page: int, limit: int) -> Select:
+def paginate(stmt: Select[R], page: int, limit: int) -> Select[R]:
     offset: int = int((page - 1) * limit)
     stmt = stmt.limit(limit).offset(offset)
     return stmt

maleo/database/types.py

@@ -0,0 +1,5 @@
+from sqlalchemy.orm import DeclarativeBase
+from typing import TypeVar
+
+
+DeclarativeBaseT = TypeVar("DeclarativeBaseT", bound=DeclarativeBase)

{maleo_database-0.0.21.dist-info → maleo_database-0.0.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maleo-database
-Version: 0.0.21
+Version: 0.0.24
 Summary: Database package for MaleoSuite
 Author-email: Agra Bima Yuda <agra@nexmedis.com>
 License: Proprietary

{maleo_database-0.0.21.dist-info → maleo_database-0.0.24.dist-info}/RECORD

@@ -1,21 +1,22 @@
 maleo/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 maleo/database/dtos.py,sha256=DArPtSXzPJd896i-t4OmoqBP1mW-it8NVxvfWq7wq8Y,220
 maleo/database/enums.py,sha256=tHJugOgGWEarwyHbUXmqK-AFXO-D-pmhIdXHkE81ik8,1691
-maleo/database/handlers.py,sha256=JsFlxvt9idY3cMXhXu_eBtcKGm9l4nyd8Gzw1WOR9vc,4722
+maleo/database/handlers.py,sha256=A8cY9yyqNhFO0L6kC1P3XdWO4xC9JXZQhK1JQs6AU2Q,4040
+maleo/database/types.py,sha256=FFosx0E46pq7423Fc85P1uA8CRPaJsY0m88KG86oLjY,142
 maleo/database/config/__init__.py,sha256=ygBWG1iq9tPCKo_mtcWgd5jcJ6gKhHmxXuZOmLuL2Hc,11102
 maleo/database/config/additional.py,sha256=8uTEQDoKjAlXif-1njKuicIT07H_slTWr9pE0OENAU0,1098
 maleo/database/config/connection.py,sha256=0Cg0XUxf20Q_RVqFmzOtBUXl9m7I-x9BEyEhZVoWEh4,22869
 maleo/database/config/identifier.py,sha256=b1MjhoKl3h7xJe1eVIj1wjvYH9BrjHzKnjzdjaEaTeQ,626
 maleo/database/config/pooling.py,sha256=4Q0rkM4FAt6xAbBG1PItHmMSzS7aCZNBgFPeaEjKwTo,11552
-maleo/database/managers/__init__.py,sha256=45un_rhnAeSFQLxmEsEXNsYLfg6kpWWljsXpqDFWZQk,13073
+maleo/database/managers/__init__.py,sha256=9fITbrn9qjmzWp2oe5KmYVa2KZZxf-7INbiXd9ctxic,13333
 maleo/database/managers/client.py,sha256=hdfPn8Ae8G78yZYDxI1FylcM-rPMIL8Ty4XmsASDrt0,3065
 maleo/database/managers/engine.py,sha256=aF7JCjSYujodbn2QQrT_OeLX-WjUdIv5cBgWtvLgHI0,1725
 maleo/database/managers/session.py,sha256=aKyuiHiA171Vptr92AA5U-mRwzwbKXb9RAZ9Wq5Wo4s,19557
 maleo/database/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 maleo/database/orm/mixins.py,sha256=rW-FVOrRm_R8yJyrcwGoVIPPV1oAOhSVFmANHR7bowU,2372
-maleo/database/orm/queries.py,sha256=o_6KHMfQeSp_3sJWusXll-NQBxPm8XUJepnWBahoyv4,7152
-maleo_database-0.0.21.dist-info/licenses/LICENSE,sha256=aftGsecnk7TWVX-7KW94FqK4Syy6YSZ8PZEF7EcIp3M,2621
-maleo_database-0.0.21.dist-info/METADATA,sha256=YPQk2kQ8dTfnfnnwwvgm7M23GLDgWEnJDpENItdGTt0,3837
-maleo_database-0.0.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-maleo_database-0.0.21.dist-info/top_level.txt,sha256=3Tpd1siVsfYoeI9FEOJNYnffx_shzZ3wsPpTvz5bljc,6
-maleo_database-0.0.21.dist-info/RECORD,,
+maleo/database/orm/queries.py,sha256=GGt9Ca92IQwQ6YTjmx03xULQ7UPKkFckbHwZCzjdZDY,7456
+maleo_database-0.0.24.dist-info/licenses/LICENSE,sha256=aftGsecnk7TWVX-7KW94FqK4Syy6YSZ8PZEF7EcIp3M,2621
+maleo_database-0.0.24.dist-info/METADATA,sha256=tqqD2BBgJgJSWMuQhREnMtIa4kSyxBiDyDaGwwKiBJQ,3837
+maleo_database-0.0.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+maleo_database-0.0.24.dist-info/top_level.txt,sha256=3Tpd1siVsfYoeI9FEOJNYnffx_shzZ3wsPpTvz5bljc,6
+maleo_database-0.0.24.dist-info/RECORD,,