worker-automate-hub 0.5.749__tar.gz → 0.5.751__tar.gz

{worker_automate_hub-0.5.749 → worker_automate_hub-0.5.751}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.749
+Version: 0.5.751
 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.749 → worker_automate_hub-0.5.751}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.749"
+version = "0.5.751"
 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.749 → worker_automate_hub-0.5.751}/worker_automate_hub/api/client.py

@@ -232,30 +232,34 @@ async def get_config_by_name(name: str) -> RpaConfiguracao:
     if "API_BASE_URL" not in env_config:
         raise ValueError("URL da API não encontrada na configuração do ambiente")
 
-    try:
-        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
-        timeout = aiohttp.ClientTimeout(total=600) 
-
-        async with aiohttp.ClientSession(
-            connector=aiohttp.TCPConnector(verify_ssl=True), timeout=timeout
-        ) as session:
-            async with session.get(
-                f"{env_config["API_BASE_URL"]}/configuracao/api/{name}",
-                headers=headers_basic,
-            ) as response:
-                if response.status != 200:
-                    raise Exception(f"Erro ao obter a configuração: {response.content}")
-                data = await response.json()
-                return RpaConfiguracao(**data)
-
-    except Exception as e:
-        err_msg = f"Erro ao obter a configuração: {e}"
-        logger.error(err_msg)
-        console.print(
-            f"{err_msg}\n",
-            style="bold red",
-        )
-        return None
+    x = 0
+    while x < 10:
+        try:
+            headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
+            timeout = aiohttp.ClientTimeout(total=600) 
+
+            async with aiohttp.ClientSession(
+                connector=aiohttp.TCPConnector(verify_ssl=True), timeout=timeout
+            ) as session:
+                async with session.get(
+                    f"{env_config["API_BASE_URL"]}/configuracao/api/{name}",
+                    headers=headers_basic,
+                ) as response:
+                    if response.status != 200:
+                        console.print(f"Erro ao obter a configuração: {response.content}")
+                        x += 1
+                    else:
+                        data = await response.json()
+                        return RpaConfiguracao(**data)
+        except Exception as e:
+            x += 1
+            err_msg = f"Erro ao obter a configuração: {e}"
+            logger.error(err_msg)
+            console.print(
+                f"{err_msg}\n",
+                style="bold red",
+            )
+    return None
 
 
 def sync_get_config_by_name(name: str) -> RpaConfiguracao:

{worker_automate_hub-0.5.749 → worker_automate_hub-0.5.751}/worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py

@@ -153,9 +153,7 @@ async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
         await worker_sleep(2)
 
         # Clicar em Salvar
-        main_window.child_window(
-            class_name="TBitBtn", found_index=1
-        ).click_input()
+        main_window.child_window(class_name="TBitBtn", found_index=1).click_input()
 
         await worker_sleep(5)
 
@@ -176,16 +174,14 @@ async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
         await worker_sleep(2)
 
         # Clicar em ok para salvar
-        main_window.child_window(
-            class_name="Button", found_index=0
-        ).click_input()
+        main_window.child_window(class_name="Button", found_index=0).click_input()
 
         await worker_sleep(20)
 
         # caminho_img = r"assets\\extracao_relatorios\\janela_printing.png"
 
         # Aguarda até a janela com título "Printing" (ou "Salvando...") fechar
-        
+
         try:
             app = Application().connect(title_re="Printing")  # conecta se existir
             janela = app.window(title_re="Printing")
@@ -214,6 +210,11 @@ async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
             )
             console.print("Removendo arquivo XLS da pasta downloads")
             os.remove(f"{caminho_arquivo}")
+            return RpaRetornoProcessoDTO(
+                sucesso=True,
+                retorno="Relatório gerado com sucesso",
+                status=RpaHistoricoStatusEnum.Sucesso,
+            )
 
         except Exception as e:
             console.print(f"Erro ao enviar o arquivo: {e}", style="bold red")