esuls 0.1.16__tar.gz → 0.1.18__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: esuls
-Version: 0.1.16
+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.16 → esuls-0.1.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "esuls"
-version = "0.1.16"
+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.16 → 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}")
@@ -334,9 +346,11 @@ class AsyncDB(Generic[SchemaType]):
         columns = ','.join(field_names)
         placeholders = ','.join('?' for _ in field_names)
 
+        set_clause = ','.join(f'{col}=excluded.{col}' for col in field_names if col != 'created_at')
         return f"""
-            INSERT OR REPLACE INTO {self.table_name} ({columns},id)
+            INSERT INTO {self.table_name} ({columns},id)
             VALUES ({placeholders},?)
+            ON CONFLICT(id) DO UPDATE SET {set_clause}
         """
     
     def _prepare_item(self, item: SchemaType) -> Tuple[str, List[Any]]:

{esuls-0.1.16 → 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
@@ -256,6 +257,35 @@ async def close_shared_client() -> None:
         _domain_clients.clear()
 
 
+async def cleanup_all() -> None:
+    """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:
     """Close HTTP client for a specific domain. If http2 is None, closes both h1 and h2 clients."""
     domain = _extract_domain(url)

{esuls-0.1.16 → esuls-0.1.18}/src/esuls/tests/test_db_fixes.py

@@ -379,7 +379,7 @@ async def test_prepare_item_dedup(temp_db):
     try:
         item = TestItem(name="test", value=5)
         sql, values = db._prepare_item(item)
-        assert "INSERT OR REPLACE" in sql
+        assert "ON CONFLICT(id) DO UPDATE SET" in sql
         assert "test" in values
         assert 5 in values
 

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

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