esuls 0.1.17__tar.gz → 0.1.18__tar.gz

{esuls-0.1.17/src/esuls.egg-info → esuls-0.1.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: esuls
-Version: 0.1.17
+Version: 0.1.18
 Summary: Utility library for async database operations, HTTP requests, and parallel execution
 Author-email: IperGiove <ipergiove@gmail.com>
 License: MIT

{esuls-0.1.17 → esuls-0.1.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "esuls"
-version = "0.1.17"
+version = "0.1.18"
 description = "Utility library for async database operations, HTTP requests, and parallel execution"
 readme = "README.md"
 requires-python = ">=3.14"

{esuls-0.1.17 → esuls-0.1.18}/src/esuls/db_cli.py

@@ -46,6 +46,8 @@ class AsyncDB(Generic[SchemaType]):
     _schema_init_lock: asyncio.Lock = None
     # Threading lock to guard class-level dict mutations
     _db_locks_guard = threading.Lock()
+    # Track all instances for cleanup
+    _instances: list['AsyncDB'] = []
 
     def __init__(self, db_path: Union[str, Path], table_name: str, schema_class: Type[SchemaType]):
         """Initialize AsyncDB with a path and schema dataclass."""
@@ -76,6 +78,9 @@ class AsyncDB(Generic[SchemaType]):
         # Persistent connection (lazy init)
         self._connection: Optional[aiosqlite.Connection] = None
 
+        # Track instance for cleanup
+        AsyncDB._instances.append(self)
+
         # Use a class-level set to track initialized schemas
         if not hasattr(AsyncDB, '_initialized_schemas'):
             AsyncDB._initialized_schemas = set()
@@ -138,6 +143,13 @@ class AsyncDB(Generic[SchemaType]):
                 pass
             self._connection = None
 
+    @classmethod
+    async def close_all(cls) -> None:
+        """Close all AsyncDB instances."""
+        for instance in cls._instances:
+            await instance.close()
+        cls._instances.clear()
+
     async def _init_schema(self, db: aiosqlite.Connection) -> None:
         """Generate schema from dataclass structure with support for field additions."""
         logger.debug(f"Initializing schema for {self.schema_class.__name__} in table {self.table_name}")

{esuls-0.1.17 → esuls-0.1.18}/src/esuls/request_cli.py

@@ -3,6 +3,7 @@ from functools import lru_cache
 from typing import TypeAlias, Union, Optional, Dict, Any, AsyncContextManager, Literal
 from urllib.parse import urlparse
 import asyncio
+import atexit
 import json
 import random
 import ssl
@@ -257,12 +258,32 @@ async def close_shared_client() -> None:
 
 
 async def cleanup_all() -> None:
-    """Close all global HTTP resources (domain clients + cffi session)."""
+    """Close all global HTTP resources (domain clients + cffi session) and DB connections."""
     await close_shared_client()
     if _get_session_cffi.cache_info().currsize > 0:
         cffi_session = _get_session_cffi()
         await cffi_session.close()
         _get_session_cffi.cache_clear()
+    # Close all AsyncDB instances
+    from esuls.db_cli import AsyncDB
+    await AsyncDB.close_all()
+
+
+def _atexit_cleanup() -> None:
+    """Run async cleanup at process exit."""
+    try:
+        loop = asyncio.get_event_loop()
+        if loop.is_running():
+            loop.create_task(cleanup_all())
+        else:
+            loop.run_until_complete(cleanup_all())
+    except RuntimeError:
+        loop = asyncio.new_event_loop()
+        loop.run_until_complete(cleanup_all())
+        loop.close()
+
+
+atexit.register(_atexit_cleanup)
 
 
 async def close_domain_client(url: str, http2: Optional[bool] = None) -> None:

{esuls-0.1.17 → esuls-0.1.18/src/esuls.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: esuls
-Version: 0.1.17
+Version: 0.1.18
 Summary: Utility library for async database operations, HTTP requests, and parallel execution
 Author-email: IperGiove <ipergiove@gmail.com>
 License: MIT