documente_shared 0.1.180__tar.gz → 0.1.181__tar.gz

{documente_shared-0.1.180 → documente_shared-0.1.181}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: documente_shared
-Version: 0.1.180
+Version: 0.1.181
 Summary: Shared utilities for LlamitAI projects
 Author-email: Tech <tech@llamitai.com>
 License: MIT

{documente_shared-0.1.180 → documente_shared-0.1.181}/documente_shared/infrastructure/documente_client.py

@@ -1,7 +1,7 @@
-from typing import Callable, Optional
-
-from requests import Response, Session
+from contextlib import contextmanager
 from dataclasses import dataclass
+from requests import Response, Session
+from typing import Callable, Optional
 
 from documente_shared.application.payloads import camel_to_snake
 
@@ -14,9 +14,12 @@ class DocumenteClientMixin(object):
     session: Optional[Session] = None
 
     def __post_init__(self):
-        if self.session is None:
-            self.session = Session()
-        self.session.headers.update(self.get_common_headers())
+        self.session = self._get_session()
+
+    def _get_session(self) -> Session:
+        session = self.session or Session()
+        session.headers.update(self.get_common_headers())
+        return session
 
     def get_common_headers(self) -> dict:
         common_headers = {
@@ -27,6 +30,14 @@ class DocumenteClientMixin(object):
             common_headers.update({"X-Tenant": self.tenant})
         return common_headers
 
+    @contextmanager
+    def transactional_session(self):
+        session = self._get_session()
+        try:
+            yield session
+        finally:
+            session.close()
+
     @staticmethod
     def _is_success(response: Response) -> bool:
         return response.status_code in [200, 201]
@@ -41,4 +52,4 @@ class DocumenteClientMixin(object):
         return [
             builder(camel_to_snake(item))
             for item in response.json().get('data', [])
-        ]
+        ]

{documente_shared-0.1.180 → documente_shared-0.1.181}/documente_shared/infrastructure/repositories/http_processing_case.py

@@ -30,10 +30,11 @@ class HttpProcessingCaseRepository(
         params = {
             "view_format": str(view_format),
         }
-        response = self.session.get(
-            url=f"{self.api_url}/v1/processing-cases/{uuid}/",
-            params=params,
-        )
+        with self.transactional_session() as session:
+            response = session.get(
+                url=f"{self.api_url}/v1/processing-cases/{uuid}/",
+                params=params,
+            )
         if response.status_code not in [200, 201]:
             return None
         return self._build_processing_case(response)
@@ -49,14 +50,15 @@ class HttpProcessingCaseRepository(
             "persist_items": parse_bool_to_str(persist_items),
             "view_format": str(view_format),
         }
-        response = self.session.put(
-            url=f"{self.api_url}/v1/processing-cases/{instance.uuid}/",
-            params=params,
-            json=(
-                instance.to_dict
-                if persist_items else instance.to_queue_dict
-            ),
-        )
+        with self.transactional_session() as session:
+            response = session.put(
+                url=f"{self.api_url}/v1/processing-cases/{instance.uuid}/",
+                params=params,
+                json=(
+                    instance.to_dict
+                    if persist_items else instance.to_queue_dict
+                ),
+            )
         if response.status_code not in [200, 201]:
             raise Exception(f'Error persisting processing case: {response.text}')
         return self._build_processing_case(response)
@@ -65,12 +67,13 @@ class HttpProcessingCaseRepository(
         self.session.delete(f"{self.api_url}/v1/processing-cases/{instance.uuid}/")
 
     def filter(self, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
-        response = self.session.get(
-            url=f"{self.api_url}/v1/processing-cases/",
-            headers={
-                "X-Tenant": filters.tenant_slug,
-            }
-        )
+        with self.transactional_session() as session:
+            response = session.get(
+                url=f"{self.api_url}/v1/processing-cases/",
+                headers={
+                    "X-Tenant": filters.tenant_slug,
+                }
+            )
         if response.status_code not in [200, 201]:
             return []
         raw_response = response.json()

{documente_shared-0.1.180 → documente_shared-0.1.181}/documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -26,9 +26,10 @@ class HttpProcessingCaseItemRepository(
         read_bytes: bool = False,
         view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCaseItem]:
-        response = self.session.get(
-            url=f"{self.api_url}/v1/processing-case-items/{uuid}/?read_bytes={parse_bool_to_str(read_bytes)}",
-        )
+        with self.transactional_session() as session:
+            response = session.get(
+                url=f"{self.api_url}/v1/processing-case-items/{uuid}/?read_bytes={parse_bool_to_str(read_bytes)}",
+            )
         if response.status_code not in [200, 201]:
             return None
         return self._build_processing_case_item(response)
@@ -43,10 +44,11 @@ class HttpProcessingCaseItemRepository(
             "read_bytes": parse_bool_to_str(read_bytes),
             "view_format": str(view_format),
         }
-        response = self.session.get(
-            url=f"{self.api_url}/v1/processing-case-items/{digest}/",
-            params=params,
-        )
+        with self.transactional_session() as session:
+            response = session.get(
+                url=f"{self.api_url}/v1/processing-case-items/{digest}/",
+                params=params,
+            )
         if response.status_code not in [200, 201]:
             return None
         return self._build_processing_case_item(response)
@@ -63,14 +65,15 @@ class HttpProcessingCaseItemRepository(
             "read_bytes": parse_bool_to_str(read_bytes),
             "view_format": str(view_format),
         }
-        response: Response = self.session.put(
-            url=f"{self.api_url}/v1/processing-case-items/{instance.uuid}/",
-            params=params,
-            json=(
-                instance.to_persist_dict
-                if persist_bytes else instance.to_queue_dict
-            ),
-        )
+        with self.transactional_session() as session:
+            response: Response = session.put(
+                url=f"{self.api_url}/v1/processing-case-items/{instance.uuid}/",
+                params=params,
+                json=(
+                    instance.to_persist_dict
+                    if persist_bytes else instance.to_queue_dict
+                ),
+            )
         if response.status_code not in [200, 201]:
             logger.error("[shared.processing_case_item.persist_error]", response=response.text)
             return instance
@@ -87,10 +90,11 @@ class HttpProcessingCaseItemRepository(
         params = {
             "view_format": str(view_format),
         }
-        response = self.session.get(
-            url=f"{self.api_url}/v1/processing-case-items/",
-            params=params,
-        )
+        with self.transactional_session() as session:
+            response = session.get(
+                url=f"{self.api_url}/v1/processing-case-items/",
+                params=params,
+            )
         if response.status_code not in [200, 201]:
             return []
         raw_response = response.json()
@@ -104,7 +108,8 @@ class HttpProcessingCaseItemRepository(
         tenant_slug: str,
         filters: ProcessingCaseItemFilters,
     ) -> List[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/v1/processing-case-items/")
+        with self.transactional_session() as session:
+            response = session.get(f"{self.api_url}/v1/processing-case-items/")
         if response.status_code not in [200, 201]:
             return []
         raw_response = response.json()

{documente_shared-0.1.180 → documente_shared-0.1.181}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "documente_shared"
-version = "0.1.180"
+version = "0.1.181"
 description = "Shared utilities for LlamitAI projects"
 authors = [{ name = "Tech", email = "tech@llamitai.com" }]
 license = { text = "MIT" }

{documente_shared-0.1.180 → documente_shared-0.1.181}/uv.lock

@@ -138,7 +138,7 @@ wheels = [
 
 [[package]]
 name = "documente-shared"
-version = "0.1.180"
+version = "0.1.181"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },