Encryptors 2.59__tar.gz → 2.60__tar.gz

{encryptors-2.59 → encryptors-2.60}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Encryptors
-Version: 2.59
+Version: 2.60
 Summary: End-to-end algorithm library
 Author: OSDental LLC
 Author-email: support@osdental.ai

{encryptors-2.59 → encryptors-2.60}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 # ANDERSON REVISAR EL CACHE LOCAL DEL KEYVAULT PARA VALIDAR SI FUNCIONA
 setup(
     name="Encryptors", 
-    version="2.59",
+    version="2.60",
     author="OSDental LLC",
     author_email="support@osdental.ai",
     description="End-to-end algorithm library",

{encryptors-2.59 → encryptors-2.60}/src/Encryptors.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Encryptors
-Version: 2.59
+Version: 2.60
 Summary: End-to-end algorithm library
 Author: OSDental LLC
 Author-email: support@osdental.ai

{encryptors-2.59 → encryptors-2.60}/src/Encryptors.egg-info/SOURCES.txt

@@ -39,6 +39,7 @@ src/Osdental/Enums/__init__.py
 src/Osdental/Exception/ControlledException.py
 src/Osdental/Exception/__init__.py
 src/Osdental/Helpers/AuditHelper.py
+src/Osdental/Helpers/AuditQueue.py
 src/Osdental/Helpers/AzureClassifier.py
 src/Osdental/Helpers/GrpcConnection.py
 src/Osdental/Helpers/JwtTokenHelper.py

encryptors-2.60/src/Osdental/Helpers/AuditQueue.py

@@ -0,0 +1,45 @@
+import asyncio
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class AuditQueue:
+
+    def __init__(
+        self,
+        worker,
+        maxsize: int = 1000,
+        workers: int = 5
+    ):
+        self._worker = worker
+        self._queue = asyncio.Queue(maxsize=maxsize)
+        self._workers = workers
+        self._tasks: list[asyncio.Task] = []
+
+    async def emit(self, payload):
+        await self._queue.put(payload)
+
+    def start(self):
+        for _ in range(self._workers):
+            task = asyncio.create_task(self._consume())
+            self._tasks.append(task)
+
+    async def _consume(self):
+        while True:
+            payload = await self._queue.get()
+
+            try:
+                await self._worker(payload)
+
+            except Exception:
+                logger.exception("Error processing audit")
+
+            finally:
+                self._queue.task_done()
+
+    async def stop(self):
+        await self._queue.join()
+
+        for task in self._tasks:
+            task.cancel()

{encryptors-2.59 → encryptors-2.60}/src/Osdental/Http/APIClient.py

@@ -20,11 +20,11 @@ class APIClient:
 
     def __init__(
         self,
-        audit: Optional[IAuditEmitter] = None,
+        audit_emit: Optional[IAuditEmitter] = None,
         timeout: Optional[httpx.Timeout] = None,
         limits: Optional[httpx.Limits] = None
     ):
-        self._audit = audit
+        self._audit_emit = audit_emit
 
         self._client = httpx.AsyncClient(
             follow_redirects=True,
@@ -41,6 +41,9 @@ class APIClient:
     @rest_retry
     async def _request(self, method: HTTPMethod, url: str, **kwargs) -> httpx.Response:
 
+        audit_ctx = _ctx.get()            
+        audit_ctx.headers = kwargs.get("headers") or {}
+
         try:
 
             response = await self._client.request(
@@ -49,10 +52,6 @@ class APIClient:
                 **kwargs
             )
 
-            audit_ctx = _ctx.get()
-            
-            audit_ctx.headers = kwargs["headers"] or {}
-
             response.raise_for_status()
 
             api_res = ApiResponse(
@@ -125,6 +124,9 @@ class APIClient:
         headers: Optional[Dict[str, str]] = None
     ) -> Dict[str, Any]:
 
+        audit_ctx = _ctx.get()           
+        audit_ctx.headers = headers or {}
+
         payload = {
             "query": query,
             "variables": variables
@@ -139,10 +141,6 @@ class APIClient:
 
         data = response.json()
 
-        audit_ctx = _ctx.get()
-            
-        audit_ctx.headers = headers or {}
-
         _build_success_graphql_payload(audit_ctx, data)
 
         if "errors" in data:

{encryptors-2.59 → encryptors-2.60}/src/Osdental/Services/ServiceBusAuditEmitter.py

@@ -8,10 +8,14 @@ class ServiceBusAuditEmitter:
     def __init__(self, messaging: IMessageQueue, storage: IStorageService):
         self._messaging = messaging
         self._storage = storage
+        
 
-    async def emit(self, payload: Dict[str, Any]) -> None:
+    async def process(self, payload: Dict[str, Any]) -> None:
         message_log_id = payload.get("idMessageLog")
         data = json.dumps(payload)
         blob_name = f"audits/{message_log_id}.text"
-        await self._storage.upload(blob_name, data)
+        url = await self._storage.upload(blob_name, data)
+        if not url:
+            return
+        
         await self._messaging.send_message(blob_name)

{encryptors-2.59 → encryptors-2.60}/src/Osdental/Services/__init__.py

@@ -49,5 +49,5 @@ class IAuthTokenService(ABC):
 
 
 class IAuditEmitter(Protocol):
-    async def emit(self, payload: Dict[str, Any]) -> None:
+    async def process(self, payload: Dict[str, Any]) -> None:
         ...

{encryptors-2.59 → encryptors-2.60}/src/Osdental/Storage/__init__.py

@@ -6,13 +6,13 @@ from azure.storage.blob import BlobSasPermissions
 class IStorageService(ABC): 
 
     @abstractmethod
-    async def upload(self, blob_name: str, data: bytes | str) -> str | bool:
+    async def upload(self, blob_name: str, data: bytes | str) -> str:
         """ Upload a file to blob storage """
         pass
 
 
     @abstractmethod
-    async def download(self, blob_name: str) -> bytes | None:
+    async def download(self, blob_name: str) -> bytes:
         """ Download a file from blob storage """
         pass