maleo-database 0.0.1__tar.gz → 0.0.3__tar.gz

{maleo_database-0.0.1 → maleo_database-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maleo-database
-Version: 0.0.1
+Version: 0.0.3
 Summary: Database package for MaleoSuite
 Author-email: Agra Bima Yuda <agra@nexmedis.com>
 License: Proprietary
@@ -16,6 +16,7 @@ Requires-Dist: cachetools>=5.5.2
 Requires-Dist: certifi>=2025.8.3
 Requires-Dist: cffi>=1.17.1
 Requires-Dist: cfgv>=3.4.0
+Requires-Dist: charset-normalizer>=3.4.3
 Requires-Dist: click>=8.2.1
 Requires-Dist: cryptography>=45.0.6
 Requires-Dist: distlib>=0.4.0
@@ -24,23 +25,37 @@ Requires-Dist: elastic-transport>=9.1.0
 Requires-Dist: elasticsearch>=9.1.0
 Requires-Dist: fastapi>=0.116.1
 Requires-Dist: filelock>=3.19.1
+Requires-Dist: google-api-core>=2.25.1
 Requires-Dist: google-auth>=2.40.3
+Requires-Dist: google-cloud-appengine-logging>=1.6.2
+Requires-Dist: google-cloud-audit-log>=0.3.2
+Requires-Dist: google-cloud-core>=2.4.3
+Requires-Dist: google-cloud-logging>=3.12.1
+Requires-Dist: googleapis-common-protos>=1.70.0
 Requires-Dist: greenlet>=3.2.4
+Requires-Dist: grpc-google-iam-v1>=0.14.2
+Requires-Dist: grpcio>=1.74.0
+Requires-Dist: grpcio-status>=1.74.0
 Requires-Dist: identify>=2.6.13
 Requires-Dist: idna>=3.10
+Requires-Dist: importlib_metadata>=8.7.0
 Requires-Dist: maleo-constants>=0.0.2
 Requires-Dist: maleo-enums>=0.0.2
-Requires-Dist: maleo-mixins>=0.0.2
+Requires-Dist: maleo-logging>=0.0.1
+Requires-Dist: maleo-mixins>=0.0.5
 Requires-Dist: maleo-types-base>=0.0.2
 Requires-Dist: maleo-types-enums>=0.0.2
 Requires-Dist: maleo-utils>=0.0.3
 Requires-Dist: motor>=3.7.1
 Requires-Dist: mypy_extensions>=1.1.0
 Requires-Dist: nodeenv>=1.9.1
+Requires-Dist: opentelemetry-api>=1.36.0
 Requires-Dist: packaging>=25.0
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: platformdirs>=4.4.0
 Requires-Dist: pre_commit>=4.3.0
+Requires-Dist: proto-plus>=1.26.1
+Requires-Dist: protobuf>=6.32.0
 Requires-Dist: pyasn1>=0.6.1
 Requires-Dist: pyasn1_modules>=0.4.2
 Requires-Dist: pycparser>=2.22
@@ -51,6 +66,7 @@ Requires-Dist: pymongo>=4.14.1
 Requires-Dist: python-dateutil>=2.9.0.post0
 Requires-Dist: PyYAML>=6.0.2
 Requires-Dist: redis>=6.4.0
+Requires-Dist: requests>=2.32.5
 Requires-Dist: rsa>=4.9.1
 Requires-Dist: six>=1.17.0
 Requires-Dist: sniffio>=1.3.1
@@ -60,6 +76,7 @@ Requires-Dist: typing-inspection>=0.4.1
 Requires-Dist: typing_extensions>=4.15.0
 Requires-Dist: urllib3>=2.5.0
 Requires-Dist: virtualenv>=20.34.0
+Requires-Dist: zipp>=3.23.0
 Dynamic: license-file
 
 # README #

{maleo_database-0.0.1 → maleo_database-0.0.3}/maleo_database.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maleo-database
-Version: 0.0.1
+Version: 0.0.3
 Summary: Database package for MaleoSuite
 Author-email: Agra Bima Yuda <agra@nexmedis.com>
 License: Proprietary
@@ -16,6 +16,7 @@ Requires-Dist: cachetools>=5.5.2
 Requires-Dist: certifi>=2025.8.3
 Requires-Dist: cffi>=1.17.1
 Requires-Dist: cfgv>=3.4.0
+Requires-Dist: charset-normalizer>=3.4.3
 Requires-Dist: click>=8.2.1
 Requires-Dist: cryptography>=45.0.6
 Requires-Dist: distlib>=0.4.0
@@ -24,23 +25,37 @@ Requires-Dist: elastic-transport>=9.1.0
 Requires-Dist: elasticsearch>=9.1.0
 Requires-Dist: fastapi>=0.116.1
 Requires-Dist: filelock>=3.19.1
+Requires-Dist: google-api-core>=2.25.1
 Requires-Dist: google-auth>=2.40.3
+Requires-Dist: google-cloud-appengine-logging>=1.6.2
+Requires-Dist: google-cloud-audit-log>=0.3.2
+Requires-Dist: google-cloud-core>=2.4.3
+Requires-Dist: google-cloud-logging>=3.12.1
+Requires-Dist: googleapis-common-protos>=1.70.0
 Requires-Dist: greenlet>=3.2.4
+Requires-Dist: grpc-google-iam-v1>=0.14.2
+Requires-Dist: grpcio>=1.74.0
+Requires-Dist: grpcio-status>=1.74.0
 Requires-Dist: identify>=2.6.13
 Requires-Dist: idna>=3.10
+Requires-Dist: importlib_metadata>=8.7.0
 Requires-Dist: maleo-constants>=0.0.2
 Requires-Dist: maleo-enums>=0.0.2
-Requires-Dist: maleo-mixins>=0.0.2
+Requires-Dist: maleo-logging>=0.0.1
+Requires-Dist: maleo-mixins>=0.0.5
 Requires-Dist: maleo-types-base>=0.0.2
 Requires-Dist: maleo-types-enums>=0.0.2
 Requires-Dist: maleo-utils>=0.0.3
 Requires-Dist: motor>=3.7.1
 Requires-Dist: mypy_extensions>=1.1.0
 Requires-Dist: nodeenv>=1.9.1
+Requires-Dist: opentelemetry-api>=1.36.0
 Requires-Dist: packaging>=25.0
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: platformdirs>=4.4.0
 Requires-Dist: pre_commit>=4.3.0
+Requires-Dist: proto-plus>=1.26.1
+Requires-Dist: protobuf>=6.32.0
 Requires-Dist: pyasn1>=0.6.1
 Requires-Dist: pyasn1_modules>=0.4.2
 Requires-Dist: pycparser>=2.22
@@ -51,6 +66,7 @@ Requires-Dist: pymongo>=4.14.1
 Requires-Dist: python-dateutil>=2.9.0.post0
 Requires-Dist: PyYAML>=6.0.2
 Requires-Dist: redis>=6.4.0
+Requires-Dist: requests>=2.32.5
 Requires-Dist: rsa>=4.9.1
 Requires-Dist: six>=1.17.0
 Requires-Dist: sniffio>=1.3.1
@@ -60,6 +76,7 @@ Requires-Dist: typing-inspection>=0.4.1
 Requires-Dist: typing_extensions>=4.15.0
 Requires-Dist: urllib3>=2.5.0
 Requires-Dist: virtualenv>=20.34.0
+Requires-Dist: zipp>=3.23.0
 Dynamic: license-file
 
 # README #

{maleo_database-0.0.1 → maleo_database-0.0.3}/maleo_database.egg-info/requires.txt

@@ -7,6 +7,7 @@ cachetools>=5.5.2
 certifi>=2025.8.3
 cffi>=1.17.1
 cfgv>=3.4.0
+charset-normalizer>=3.4.3
 click>=8.2.1
 cryptography>=45.0.6
 distlib>=0.4.0
@@ -15,23 +16,37 @@ elastic-transport>=9.1.0
 elasticsearch>=9.1.0
 fastapi>=0.116.1
 filelock>=3.19.1
+google-api-core>=2.25.1
 google-auth>=2.40.3
+google-cloud-appengine-logging>=1.6.2
+google-cloud-audit-log>=0.3.2
+google-cloud-core>=2.4.3
+google-cloud-logging>=3.12.1
+googleapis-common-protos>=1.70.0
 greenlet>=3.2.4
+grpc-google-iam-v1>=0.14.2
+grpcio>=1.74.0
+grpcio-status>=1.74.0
 identify>=2.6.13
 idna>=3.10
+importlib_metadata>=8.7.0
 maleo-constants>=0.0.2
 maleo-enums>=0.0.2
-maleo-mixins>=0.0.2
+maleo-logging>=0.0.1
+maleo-mixins>=0.0.5
 maleo-types-base>=0.0.2
 maleo-types-enums>=0.0.2
 maleo-utils>=0.0.3
 motor>=3.7.1
 mypy_extensions>=1.1.0
 nodeenv>=1.9.1
+opentelemetry-api>=1.36.0
 packaging>=25.0
 pathspec>=0.12.1
 platformdirs>=4.4.0
 pre_commit>=4.3.0
+proto-plus>=1.26.1
+protobuf>=6.32.0
 pyasn1>=0.6.1
 pyasn1_modules>=0.4.2
 pycparser>=2.22
@@ -42,6 +57,7 @@ pymongo>=4.14.1
 python-dateutil>=2.9.0.post0
 PyYAML>=6.0.2
 redis>=6.4.0
+requests>=2.32.5
 rsa>=4.9.1
 six>=1.17.0
 sniffio>=1.3.1
@@ -51,3 +67,4 @@ typing-inspection>=0.4.1
 typing_extensions>=4.15.0
 urllib3>=2.5.0
 virtualenv>=20.34.0
+zipp>=3.23.0

{maleo_database-0.0.1 → maleo_database-0.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "maleo-database"
-version = "0.0.1"
+version = "0.0.3"
 description = "Database package for MaleoSuite"
 authors = [
     { name = "Agra Bima Yuda", email = "agra@nexmedis.com" }
@@ -22,6 +22,7 @@ dependencies = [
     "certifi>=2025.8.3",
     "cffi>=1.17.1",
     "cfgv>=3.4.0",
+    "charset-normalizer>=3.4.3",
     "click>=8.2.1",
     "cryptography>=45.0.6",
     "distlib>=0.4.0",
@@ -30,23 +31,37 @@ dependencies = [
     "elasticsearch>=9.1.0",
     "fastapi>=0.116.1",
     "filelock>=3.19.1",
+    "google-api-core>=2.25.1",
     "google-auth>=2.40.3",
+    "google-cloud-appengine-logging>=1.6.2",
+    "google-cloud-audit-log>=0.3.2",
+    "google-cloud-core>=2.4.3",
+    "google-cloud-logging>=3.12.1",
+    "googleapis-common-protos>=1.70.0",
     "greenlet>=3.2.4",
+    "grpc-google-iam-v1>=0.14.2",
+    "grpcio>=1.74.0",
+    "grpcio-status>=1.74.0",
     "identify>=2.6.13",
     "idna>=3.10",
+    "importlib_metadata>=8.7.0",
     "maleo-constants>=0.0.2",
     "maleo-enums>=0.0.2",
-    "maleo-mixins>=0.0.2",
+    "maleo-logging>=0.0.1",
+    "maleo-mixins>=0.0.5",
     "maleo-types-base>=0.0.2",
     "maleo-types-enums>=0.0.2",
     "maleo-utils>=0.0.3",
     "motor>=3.7.1",
     "mypy_extensions>=1.1.0",
     "nodeenv>=1.9.1",
+    "opentelemetry-api>=1.36.0",
     "packaging>=25.0",
     "pathspec>=0.12.1",
     "platformdirs>=4.4.0",
     "pre_commit>=4.3.0",
+    "proto-plus>=1.26.1",
+    "protobuf>=6.32.0",
     "pyasn1>=0.6.1",
     "pyasn1_modules>=0.4.2",
     "pycparser>=2.22",
@@ -57,6 +72,7 @@ dependencies = [
     "python-dateutil>=2.9.0.post0",
     "PyYAML>=6.0.2",
     "redis>=6.4.0",
+    "requests>=2.32.5",
     "rsa>=4.9.1",
     "six>=1.17.0",
     "sniffio>=1.3.1",
@@ -66,6 +82,7 @@ dependencies = [
     "typing_extensions>=4.15.0",
     "urllib3>=2.5.0",
     "virtualenv>=20.34.0",
+    "zipp>=3.23.0",
 ]
 
 [tool.setuptools]

maleo_database-0.0.3/src/managers/session.py

@@ -0,0 +1,256 @@
+from contextlib import (
+    AbstractAsyncContextManager,
+    AbstractContextManager,
+    asynccontextmanager,
+    contextmanager,
+)
+from datetime import datetime, timezone
+from pydantic import ValidationError
+from sqlalchemy.engine import Engine
+from sqlalchemy.exc import SQLAlchemyError
+from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, AsyncSession
+from sqlalchemy.orm import sessionmaker, Session
+from typing import AsyncGenerator, Generator, Literal, Tuple, Union, overload
+from uuid import UUID, uuid4
+from maleo.mixins.timestamp import OperationTimestamp
+from maleo.types.base.uuid import OptionalUUID
+from maleo.logging.logger import Database
+from ..enums import Connection
+from ..config import (
+    PostgreSQLDatabaseConfig,
+    MySQLDatabaseConfig,
+    SQLiteDatabaseConfig,
+    SQLServerDatabaseConfig,
+)
+
+
+class SessionManager:
+    def __init__(
+        self,
+        config: Union[
+            PostgreSQLDatabaseConfig,
+            MySQLDatabaseConfig,
+            SQLiteDatabaseConfig,
+            SQLServerDatabaseConfig,
+        ],
+        engines: Tuple[AsyncEngine, Engine],
+        logger: Database,
+    ) -> None:
+        self._config = config
+        self._async_engine, self._sync_engine = engines
+        self._logger = logger
+        self._async_sessionmaker: async_sessionmaker[AsyncSession] = async_sessionmaker[
+            AsyncSession
+        ](bind=self._async_engine, expire_on_commit=True)
+        self._sync_sessionmaker: sessionmaker[Session] = sessionmaker[Session](
+            bind=self._sync_engine, expire_on_commit=True
+        )
+
+    async def _async_session_handler(
+        self, operation_id: UUID
+    ) -> AsyncGenerator[AsyncSession, None]:
+        """Async session handler with proper error handling."""
+        executed_at = datetime.now(tz=timezone.utc)
+        session = self._async_sessionmaker()
+        try:
+            yield session
+            await session.commit()
+            completed_at = datetime.now(tz=timezone.utc)
+            self._logger.info(
+                f"Operation {operation_id} - success - Committed async database transaction",
+                extra={
+                    "json_fields": {
+                        "config": self._config.model_dump(mode="json"),
+                        "operation_id": operation_id,
+                        "success": True,
+                        "timestamp": OperationTimestamp(
+                            executed_at=executed_at,
+                            completed_at=completed_at,
+                            duration=(completed_at - executed_at).total_seconds(),
+                        ).model_dump(mode="json"),
+                    },
+                },
+            )
+        except (SQLAlchemyError, ValidationError, Exception):
+            await session.rollback()
+            completed_at = datetime.now(tz=timezone.utc)
+            self._logger.error(
+                f"Operation {operation_id} - failed - Error handling async database session",
+                exc_info=True,
+                extra={
+                    "json_fields": {
+                        "config": self._config.model_dump(mode="json"),
+                        "operation_id": operation_id,
+                        "success": False,
+                        "timestamp": OperationTimestamp(
+                            executed_at=executed_at,
+                            completed_at=completed_at,
+                            duration=(completed_at - executed_at).total_seconds(),
+                        ).model_dump(mode="json"),
+                    },
+                },
+            )
+            raise
+        finally:
+            await session.close()
+            completed_at = datetime.now(tz=timezone.utc)
+            self._logger.info(
+                f"Operation {operation_id} - success - Closed async database session",
+                extra={
+                    "json_fields": {
+                        "config": self._config.model_dump(mode="json"),
+                        "operation_id": operation_id,
+                        "success": True,
+                        "timestamp": OperationTimestamp(
+                            executed_at=executed_at,
+                            completed_at=completed_at,
+                            duration=(completed_at - executed_at).total_seconds(),
+                        ).model_dump(mode="json"),
+                    },
+                },
+            )
+
+    def _sync_session_handler(
+        self, operation_id: UUID
+    ) -> Generator[Session, None, None]:
+        """Sync session handler with proper error handling."""
+        executed_at = datetime.now(tz=timezone.utc)
+        session = self._sync_sessionmaker()
+        try:
+            yield session
+            session.commit()
+            completed_at = datetime.now(tz=timezone.utc)
+            self._logger.info(
+                f"Operation {operation_id} - success - Committed sync database transaction",
+                extra={
+                    "json_fields": {
+                        "config": self._config.model_dump(mode="json"),
+                        "operation_id": operation_id,
+                        "success": True,
+                        "timestamp": OperationTimestamp(
+                            executed_at=executed_at,
+                            completed_at=completed_at,
+                            duration=(completed_at - executed_at).total_seconds(),
+                        ).model_dump(mode="json"),
+                    },
+                },
+            )
+        except (SQLAlchemyError, ValidationError, Exception):
+            session.rollback()
+            completed_at = datetime.now(tz=timezone.utc)
+            self._logger.error(
+                f"Operation {operation_id} - failed - Error handling sync database session",
+                exc_info=True,
+                extra={
+                    "json_fields": {
+                        "config": self._config.model_dump(mode="json"),
+                        "operation_id": operation_id,
+                        "success": False,
+                        "timestamp": OperationTimestamp(
+                            executed_at=executed_at,
+                            completed_at=completed_at,
+                            duration=(completed_at - executed_at).total_seconds(),
+                        ).model_dump(mode="json"),
+                    },
+                },
+            )
+            raise
+        finally:
+            session.close()
+            completed_at = datetime.now(tz=timezone.utc)
+            self._logger.info(
+                f"Operation {operation_id} - success - Closed sync database session",
+                extra={
+                    "json_fields": {
+                        "config": self._config.model_dump(mode="json"),
+                        "operation_id": operation_id,
+                        "success": True,
+                        "timestamp": OperationTimestamp(
+                            executed_at=executed_at,
+                            completed_at=completed_at,
+                            duration=(completed_at - executed_at).total_seconds(),
+                        ).model_dump(mode="json"),
+                    },
+                },
+            )
+
+    @asynccontextmanager
+    async def _async_context_manager(
+        self, operation_id: UUID
+    ) -> AsyncGenerator[AsyncSession, None]:
+        """Async context manager implementation."""
+        async for session in self._async_session_handler(operation_id):
+            yield session
+
+    @contextmanager
+    def _sync_context_manager(
+        self, operation_id: UUID
+    ) -> Generator[Session, None, None]:
+        """Sync context manager implementation."""
+        yield from self._sync_session_handler(operation_id)
+
+    # Overloaded context manager methods
+    @overload
+    def get(
+        self, connection: Literal[Connection.ASYNC], operation_id: OptionalUUID = None
+    ) -> AbstractAsyncContextManager[AsyncSession]: ...
+
+    @overload
+    def get(
+        self, connection: Literal[Connection.SYNC], operation_id: OptionalUUID = None
+    ) -> AbstractContextManager[Session]: ...
+
+    def get(
+        self,
+        connection: Connection = Connection.ASYNC,
+        operation_id: OptionalUUID = None,
+    ) -> Union[
+        AbstractAsyncContextManager[AsyncSession], AbstractContextManager[Session]
+    ]:
+        """Context manager for manual session handling."""
+        if operation_id is None:
+            operation_id = uuid4()
+        if connection is Connection.ASYNC:
+            return self._async_context_manager(operation_id)
+        else:
+            return self._sync_context_manager(operation_id)
+
+    # Alternative: More explicit methods
+    @asynccontextmanager
+    async def get_async(
+        self, operation_id: OptionalUUID = None
+    ) -> AsyncGenerator[AsyncSession, None]:
+        """Explicit async context manager."""
+        if operation_id is None:
+            operation_id = uuid4()
+        async for session in self._async_session_handler(operation_id):
+            yield session
+
+    @contextmanager
+    def get_sync(
+        self, operation_id: OptionalUUID = None
+    ) -> Generator[Session, None, None]:
+        """Explicit sync context manager."""
+        if operation_id is None:
+            operation_id = uuid4()
+        yield from self._sync_session_handler(operation_id)
+
+    def as_async_dependency(self, operation_id: OptionalUUID = None):
+        """Explicit async dependency injection."""
+        if operation_id is None:
+            operation_id = uuid4()
+
+        def dependency() -> AsyncGenerator[AsyncSession, None]:
+            return self._async_session_handler(operation_id)
+
+        return dependency
+
+    def as_sync_dependency(self, operation_id: OptionalUUID = None):
+        """Explicit sync dependency injection."""
+        if operation_id is None:
+            operation_id = uuid4()
+
+        def dependency() -> AsyncGenerator[AsyncSession, None]:
+            return self._async_session_handler(operation_id)
+
+        return dependency

maleo_database-0.0.1/src/managers/session.py

@@ -1,123 +0,0 @@
-from contextlib import (
-    AbstractAsyncContextManager,
-    AbstractContextManager,
-    asynccontextmanager,
-    contextmanager,
-)
-from pydantic import ValidationError
-from sqlalchemy.engine import Engine
-from sqlalchemy.exc import SQLAlchemyError
-from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, AsyncSession
-from sqlalchemy.orm import sessionmaker, Session
-from typing import AsyncGenerator, Generator, Literal, Tuple, Union, overload
-from ..enums import Connection
-
-
-class SessionManager:
-    def __init__(self, engines: Tuple[AsyncEngine, Engine]) -> None:
-        self._async_engine, self._sync_engine = engines
-        self._async_sessionmaker: async_sessionmaker[AsyncSession] = async_sessionmaker[
-            AsyncSession
-        ](bind=self._async_engine, expire_on_commit=True)
-        self._sync_sessionmaker: sessionmaker[Session] = sessionmaker[Session](
-            bind=self._sync_engine, expire_on_commit=True
-        )
-
-    async def _async_session_handler(self) -> AsyncGenerator[AsyncSession, None]:
-        """Async session handler with proper error handling."""
-        session = self._async_sessionmaker()
-        try:
-            yield session
-            await session.commit()
-        except (SQLAlchemyError, ValidationError, Exception):
-            await session.rollback()
-            raise
-        finally:
-            await session.close()
-
-    def _sync_session_handler(self) -> Generator[Session, None, None]:
-        """Sync session handler with proper error handling."""
-        session = self._sync_sessionmaker()
-        try:
-            yield session
-            session.commit()
-        except (SQLAlchemyError, ValidationError, Exception):
-            session.rollback()
-            raise
-        finally:
-            session.close()
-
-    # Overloaded inject methods
-    @overload
-    def inject(
-        self, connection: Literal[Connection.ASYNC]
-    ) -> AsyncGenerator[AsyncSession, None]: ...
-
-    @overload
-    def inject(
-        self, connection: Literal[Connection.SYNC]
-    ) -> Generator[Session, None, None]: ...
-
-    def inject(
-        self, connection: Connection = Connection.ASYNC
-    ) -> Union[AsyncGenerator[AsyncSession, None], Generator[Session, None, None]]:
-        """Returns a generator for dependency injection."""
-        if connection is Connection.ASYNC:
-            return self._async_session_handler()
-        else:
-            return self._sync_session_handler()
-
-    # Overloaded context manager methods
-    @overload
-    def get(
-        self, connection: Literal[Connection.ASYNC]
-    ) -> AbstractAsyncContextManager[AsyncSession]: ...
-
-    @overload
-    def get(
-        self, connection: Literal[Connection.SYNC]
-    ) -> AbstractContextManager[Session]: ...
-
-    def get(
-        self, connection: Connection = Connection.ASYNC
-    ) -> Union[
-        AbstractAsyncContextManager[AsyncSession], AbstractContextManager[Session]
-    ]:
-        """Context manager for manual session handling."""
-        if connection is Connection.ASYNC:
-            return self._async_context_manager()
-        else:
-            return self._sync_context_manager()
-
-    @asynccontextmanager
-    async def _async_context_manager(self) -> AsyncGenerator[AsyncSession, None]:
-        """Async context manager implementation."""
-        async for session in self._async_session_handler():
-            yield session
-
-    @contextmanager
-    def _sync_context_manager(self) -> Generator[Session, None, None]:
-        """Sync context manager implementation."""
-        yield from self._sync_session_handler()
-
-    # Alternative: More explicit methods
-    @asynccontextmanager
-    async def get_async(self) -> AsyncGenerator[AsyncSession, None]:
-        """Explicit async context manager."""
-        async for session in self._async_session_handler():
-            yield session
-
-    @contextmanager
-    def get_sync(self) -> Generator[Session, None, None]:
-        """Explicit sync context manager."""
-        yield from self._sync_session_handler()
-        # with self._sync_session_handler() as session:
-        #     yield session
-
-    def inject_async(self) -> AsyncGenerator[AsyncSession, None]:
-        """Explicit async dependency injection."""
-        return self._async_session_handler()
-
-    def inject_sync(self) -> Generator[Session, None, None]:
-        """Explicit sync dependency injection."""
-        return self._sync_session_handler()