Encryptors 2.41__tar.gz → 2.42__tar.gz

{encryptors-2.41 → encryptors-2.42}/PKG-INFO

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

{encryptors-2.41 → encryptors-2.42}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 # ANDERSON ESTO YA SE SUBIO Y ESTA ESTABLE, AUN TE QUEDA PENDIENTE LA AUDITORIA CON RSA Y AES DE ACCESSTOKEN
 setup(
     name="Encryptors", 
-    version="2.41",
+    version="2.42",
     author="OSDental LLC",
     author_email="support@osdental.ai",
     description="End-to-end algorithm library",

{encryptors-2.41 → encryptors-2.42}/src/Encryptors.egg-info/PKG-INFO

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

{encryptors-2.41 → encryptors-2.42}/src/Osdental/Graphql/Extensions/AuditExtension.py

@@ -1,4 +1,5 @@
 import json
+import copy
 from graphql.pyutils import is_awaitable
 from ariadne.types import Extension
 from Osdental.Shared.Logger import logger
@@ -9,6 +10,7 @@ from Osdental.Rest.Context.RequestContext import (
     RequestContext
 )
 from Osdental.Graphql.Models import BaseGraphQLContext
+from Osdental.Helpers.ResponseDecryptor import encryptor_data, VALID_TYPES
 from Osdental.Shared.Enums.Constant import Constant
 
 class AuditExtension(Extension):
@@ -154,7 +156,11 @@ class AuditExtension(Extension):
             self.result = result
 
             if not getattr(context, "audit_plain_response", None):
-                context.audit_plain_response = result
+                if result.encryption_type in VALID_TYPES and result.key:
+                    result_copy = copy.deepcopy(result)
+                    result = encryptor_data(result.encryption_type, result.key, result.data)
+
+                context.audit_plain_response = result_copy
 
         return result
 

{encryptors-2.41 → encryptors-2.42}/src/Osdental/Helpers/AuditDispatcher.py

@@ -127,7 +127,7 @@ class AuditDispatcher:
             audit_message = request_audit_payload | payload
 
         else:
-            if encryption_type in VALID_TYPES and key is not None:
+            if encryption_type in VALID_TYPES and key:
                 data = decryptor_data(encryption_type, key, data)
 
             payload = AuditHelper.build_final_payload(

encryptors-2.42/src/Osdental/Helpers/ResponseDecryptor.py

@@ -0,0 +1,33 @@
+from typing import Dict, Any
+from Osdental.Encryptor.Aes import AES
+from Osdental.Encryptor.Rsa import RSAEncryptor
+from Osdental.Shared.Logger import logger
+
+VALID_TYPES = {"RSA", "AES"}
+
+def encryptor_data(encryption_type: str, key: str, data: Dict[str, Any] | str) -> Any:
+    match encryption_type:
+
+        case "AES":
+            return AES.encrypt(key, data)
+
+        case "RSA":
+            return RSAEncryptor.encrypt(data, key)
+        
+        case _:
+            logger.warning(f"Encryption type '{encryption_type}' not supported.")
+            return data
+        
+def decryptor_data(encryption_type: str, key: str, data: Dict[str, Any] | str) -> Any:
+    
+    match encryption_type:
+
+        case "AES":
+            return AES.decrypt(key, data)
+
+        case "RSA":
+            return RSAEncryptor.decrypt(data, key)
+        
+        case _:
+            logger.warning(f"Encryption type '{encryption_type}' not supported.")
+            return data

encryptors-2.41/src/Osdental/Helpers/ResponseDecryptor.py

@@ -1,15 +0,0 @@
-from typing import Dict, Any
-from Osdental.Encryptor.Aes import AES
-from Osdental.Encryptor.Rsa import RSAEncryptor
-
-VALID_TYPES = {"RSA", "AES"}
-
-def decryptor_data(encryption_type: str, key: str, data: Dict[str, Any] | str) -> Any:
-    
-    match encryption_type:
-
-        case "AES":
-            return AES.decrypt(key, data)
-
-        case "RSA":
-            return RSAEncryptor.decrypt(data, key)