worker-automate-hub 0.5.37__tar.gz → 0.5.38__tar.gz

{worker_automate_hub-0.5.37 → worker_automate_hub-0.5.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.37
+Version: 0.5.38
 Summary: Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta.
 Author: Joel Paim
 Requires-Python: >=3.12,<4.0

{worker_automate_hub-0.5.37 → worker_automate_hub-0.5.38}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.37"
+version = "0.5.38"
 description = "Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta."
 authors = ["Joel Paim"]
 readme = "README.md"

{worker_automate_hub-0.5.37 → worker_automate_hub-0.5.38}/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