worker-automate-hub 0.4.446__tar.gz → 0.4.447__tar.gz

{worker_automate_hub-0.4.446 → worker_automate_hub-0.4.447}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.4.446"
+version = "0.4.447"
 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.447/worker_automate_hub/api/webhook_service.py

@@ -0,0 +1,57 @@
+from worker_automate_hub.utils.logger import logger
+from rich.console import Console
+
+import aiohttp
+
+console = Console()
+
+async def send_to_webhook(
+    url_retorno: str,
+    status: str,
+    observacao: str
+) -> None:    
+    """
+    Envia uma notificacao para o endpoint do webhook passado como parametro.
+
+    Args:
+        url_retorno (str): URL do endpoint do webhook.
+        status (str): Status da notificacao.
+        observacao (str): Observacao da notificacao.
+
+    Raises:
+        ValueError: Se a URL, status ou observacao forem vazias.
+    """
+    if not url_retorno:
+        raise ValueError("URL do retorno esta vazia.")
+
+    if not status:
+        raise ValueError("Status da notificacao esta vazio.")
+
+    if not observacao:
+        raise ValueError("Observacao da notificacao esta vazia.")
+
+    data = {
+        "status": status,
+        "observacao": observacao,
+    }
+
+    try:
+        async with aiohttp.ClientSession(
+            connector=aiohttp.TCPConnector(verify_ssl=False)
+        ) as session:
+            async with session.post(f"{url_retorno}", data=data) as response:
+                if response.status != 200:
+                    raise Exception(f"Erro ao enviar notificacao: {response.text()}")
+
+                data = await response.text()
+                log_msg = f"\nSucesso ao enviar {data}\n para o webhook: {url_retorno}.\n"
+                console.print(
+                    log_msg,
+                    style="bold green",
+                )
+                logger.info(log_msg)
+
+    except Exception as e:
+        err_msg = f"Erro ao comunicar com endpoint do webhook: {e}"
+        console.print(f"\n{err_msg}\n", style="bold red")
+        logger.info(err_msg)

{worker_automate_hub-0.4.446 → worker_automate_hub-0.4.447}/worker_automate_hub/tasks/task_executor.py

@@ -1,6 +1,7 @@
 from datetime import datetime
 
 from pytz import timezone
+from worker_automate_hub.api.webhook_service import send_to_webhook
 from rich.console import Console
 
 from worker_automate_hub.api.client import (
@@ -37,6 +38,7 @@ async def perform_task(task: RpaProcessoEntradaDTO):
     console.print(f"\n{log_msg}\n", style="green")
     logger.info(log_msg)
     task_uuid = task.uuidProcesso
+    url_retorno = task.configEntrada.get("urlRetorno", None)
     processo: RpaProcesso = await get_processo(task_uuid)
     if processo is None:
         worker_config = load_worker_config()
@@ -74,6 +76,9 @@ async def perform_task(task: RpaProcessoEntradaDTO):
                 )
             else:
                 show_toast("Sucesso", f"Processo executado com sucesso: {result}")
+
+            if url_retorno is not None and result.sucesso == False:
+                await send_to_webhook(url_retorno, result.status, result.retorno)
             return result
         else:
             err_msg = f"Processo não encontrado: {task_uuid}"
@@ -93,6 +98,8 @@ async def perform_task(task: RpaProcessoEntradaDTO):
                     processo=processo,
                 )
             await unlock_queue(task.uuidFila)
+            if url_retorno is not None:
+                await send_to_webhook(url_retorno, RpaHistoricoStatusEnum.Falha, err_msg)
             return None
     except Exception as e:
         err_msg = f"Erro ao performar o processo: {e}"
@@ -112,6 +119,8 @@ async def perform_task(task: RpaProcessoEntradaDTO):
         await capture_and_send_screenshot(
             uuidRelacao=historico.uuidHistorico, desArquivo=err_msg
         )
+        if url_retorno is not None:
+            await send_to_webhook(url_retorno, RpaHistoricoStatusEnum.Falha, err_msg)
 
 
 async def _store_historico(