worker-automate-hub 0.5.37__py3-none-any.whl → 0.5.39__py3-none-any.whl

worker_automate_hub/api/ahead_service.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 from pathlib import Path
 import getpass
@@ -5,6 +6,7 @@ import aiohttp
 from worker_automate_hub.config.settings import load_env_config
 from worker_automate_hub.utils.logger import logger
 
+
 async def get_xml(chave_acesso: str):
     env_config, _ = load_env_config()
     try:
@@ -31,26 +33,33 @@ async def get_xml(chave_acesso: str):
         logger.error(err_msg)
         return None
 
+
 async def save_xml_to_downloads(chave_acesso: str):
-    xml_content = await get_xml(chave_acesso)
-    if xml_content:
-        try:
-            downloads_path = Path.home() / "Downloads"
-            downloads_path.mkdir(parents=True, exist_ok=True)
-
-            file_name = f"{chave_acesso}.xml"
-            file_path = downloads_path / file_name
-
-            with open(file_path, 'w', encoding='utf-8') as f:
-                f.write(xml_content)
-
-            logger.info(f"XML salvo em {file_path}")
-            return True
-        except Exception as e:
-            err_msg = f"Erro ao salvar o XML: {e}"
+    MAX_RETRIES = 3
+
+    for attempt in range(1, MAX_RETRIES + 1):
+        xml_content = await get_xml(chave_acesso)
+        if xml_content:
+            try:
+                downloads_path = Path.home() / "Downloads"
+                downloads_path.mkdir(parents=True, exist_ok=True)
+
+                file_name = f"{chave_acesso}.xml"
+                file_path = downloads_path / file_name
+
+                with open(file_path, "w", encoding="utf-8") as f:
+                    f.write(xml_content)
+
+                logger.info(f"XML salvo em {file_path}")
+                return True
+            except Exception as e:
+                err_msg = f"Erro ao salvar o XML: {e}"
+                logger.error(err_msg)
+        else:
+            err_msg = "Não foi possível obter o XML."
             logger.error(err_msg)
-            return False
-    else:
-        err_msg = "Não foi possível obter o XML."
-        logger.error(err_msg)
-        return False
+
+        if attempt < MAX_RETRIES:
+            await asyncio.sleep(2)
+
+    return False

worker_automate_hub/api/client.py

@@ -460,6 +460,9 @@ async def send_file(uuidRelacao: str, desArquivo: str, tipo: str, file: bytes, f
         if file_extension == "txt":
             filename = "text.txt"
             content_type = "text/plain"
+        elif file_extension == "pdf":
+            filename = "file.pdf"
+            content_type = "application/pdf"
         else:
             filename = "file.jpg"
             content_type = "image/jpeg"