worker-automate-hub 0.4.440__tar.gz → 0.4.442__tar.gz

{worker_automate_hub-0.4.440 → worker_automate_hub-0.4.442}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: worker-automate-hub
-Version: 0.4.440
+Version: 0.4.442
 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.4.440 → worker_automate_hub-0.4.442}/pyproject.toml

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

@@ -45,33 +45,6 @@ async def get_new_task(stop_event: threading.Event) -> RpaProcessoEntradaDTO:
         )
         return None
 
-async def burnQueue(id_fila: str):
-    env_config, _ = load_env_config()
-    try:       
-
-        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
-        
-
-        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=False)) as session:
-            async with session.delete(
-                f"{env_config["API_BASE_URL"]}/fila/burn-queue/{id_fila}",
-                headers=headers_basic,
-            ) as response:
-                if response.status == 200:
-                    logger.info("Fila excluida com sucesso.")
-                    console.print("\nFila excluida com sucesso.\n", style="bold green")    
-                else:
-                    logger.error(f"Erro ao excluir a fila: {response.content}") 
-                    console.print(f"Erro ao excluir a fila: {response.content}", style="bold red")          
-
-    except Exception as e:
-        err_msg = f"Erro remover registro da fila: {e}"
-        logger.error(err_msg)
-        console.print(
-            f"{err_msg}\n",
-            style="bold red",
-        )
-        return None
     
 async def notify_is_alive(stop_event: threading.Event):
     env_config, _ = load_env_config()
@@ -348,28 +321,7 @@ async def send_gchat_message(message: str) -> None:
             style="bold red",
         )
         return None
-    
-async def unlock_queue(id: str):
-    env_config, _ = load_env_config()
-    try:      
-        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}       
-        
 
-        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=False)) as session:
-            async with session.get(
-                f"{env_config["API_BASE_URL"]}/fila/unlock-queue/{id}",                
-                headers=headers_basic,
-            ) as response:
-                return await response.text()
-
-    except Exception as e:
-        err_msg = f"Erro ao desbloquear a fila: {e}"
-        logger.error(err_msg)
-        console.print(
-            f"{err_msg}\n",
-            style="bold red",
-        )
-        return None
 
 
 def read_secret(path: str, vault_token: str):

worker_automate_hub-0.4.442/worker_automate_hub/api/rpa_fila_service.py

@@ -0,0 +1,58 @@
+from worker_automate_hub.config.settings import load_env_config
+from worker_automate_hub.utils.logger import logger
+from rich.console import Console
+
+import aiohttp
+
+console = Console()
+
+
+async def burn_queue(id_fila: str):
+    env_config, _ = load_env_config()
+    try:       
+
+        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
+        
+
+        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=False)) as session:
+            async with session.delete(
+                f"{env_config["API_BASE_URL"]}/fila/burn-queue/{id_fila}",
+                headers=headers_basic,
+            ) as response:
+                if response.status == 200:
+                    logger.info("Fila excluida com sucesso.")
+                    console.print("\nFila excluida com sucesso.\n", style="bold green")    
+                else:
+                    logger.error(f"Erro ao excluir a fila: {response.content}") 
+                    console.print(f"Erro ao excluir a fila: {response.content}", style="bold red")          
+
+    except Exception as e:
+        err_msg = f"Erro remover registro da fila: {e}"
+        logger.error(err_msg)
+        console.print(
+            f"{err_msg}\n",
+            style="bold red",
+        )
+        return None
+    
+async def unlock_queue(id: str):
+    env_config, _ = load_env_config()
+    try:      
+        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}       
+        
+
+        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=False)) as session:
+            async with session.get(
+                f"{env_config["API_BASE_URL"]}/fila/unlock-queue/{id}",                
+                headers=headers_basic,
+            ) as response:
+                return await response.text()
+
+    except Exception as e:
+        err_msg = f"Erro ao desbloquear a fila: {e}"
+        logger.error(err_msg)
+        console.print(
+            f"{err_msg}\n",
+            style="bold red",
+        )
+        return None

{worker_automate_hub-0.4.440 → worker_automate_hub-0.4.442}/worker_automate_hub/tasks/task_executor.py

@@ -6,8 +6,8 @@ from rich.console import Console
 from worker_automate_hub.api.client import (
     get_processo,
     send_gchat_message,
-    unlock_queue,
 )
+from worker_automate_hub.api.rpa_fila_service import unlock_queue
 from worker_automate_hub.api.rpa_historico_service import store, update
 from worker_automate_hub.models.dao.rpa_historico import RpaHistorico
 from worker_automate_hub.models.dao.rpa_processo import RpaProcesso

{worker_automate_hub-0.4.440 → worker_automate_hub-0.4.442}/worker_automate_hub/worker.py

@@ -3,11 +3,11 @@ import os
 import threading
 from pathlib import Path
 
+from worker_automate_hub.api.rpa_fila_service import burn_queue, unlock_queue
 import pyfiglet
 from rich.console import Console
 
 from worker_automate_hub.api.client import (
-    burnQueue,
     get_new_task,
     notify_is_alive,
     send_gchat_message,
@@ -41,11 +41,12 @@ async def check_and_execute_tasks(stop_event: threading.Event):
             if task is not None:
                 processo_existe = await is_uuid_in_tasks(task.uuidProcesso)
                 if processo_existe:
-                    await burnQueue(task.uuidFila)
+                    await burn_queue(task.uuidFila)
                     logger.info(f"Executando a task: {task.nomProcesso}")
                     await perform_task(task)
                 else:
-                    log_message = f"O processo [{task.nomProcesso}] não existe no Worker [{worker_config['NOME_ROBO']}] e não foi removido da fila."
+                    await unlock_queue(task.uuidFila)
+                    log_message = f"O processo [{task.nomProcesso}] não existe no Worker [{worker_config['NOME_ROBO']}] e foi devolvido para a fila."
                     console.print(f"\n{log_message}\n", style="yellow")
                     logger.error(log_message)
                     await send_gchat_message(log_message)