worker-automate-hub 0.5.9__py3-none-any.whl → 0.5.12__py3-none-any.whl

worker_automate_hub/api/rdp_service.py

@@ -0,0 +1,38 @@
+import asyncio
+import aiohttp
+from rich.console import Console
+from worker_automate_hub.config.settings import load_env_config
+from worker_automate_hub.utils.logger import logger
+
+console = Console()
+
+async def send_rdp_action(uuid_robo: str, action: str):
+    env_config, _ = load_env_config()
+    
+    body = {
+        "uuidRobo": uuid_robo, 
+        "action": action
+    }
+    
+    headers_basic = {"Authorization": f"Basic {env_config['API_AUTHORIZATION']}"}
+    url = f"{env_config['API_BASE_URL']}/robo/api/manage"
+
+    try:
+        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=True)) as session:
+            async with session.post(url, json=body, headers=headers_basic) as response:
+                if response.status == 200:
+                    data = await response.json()
+                    logger.info(f"Ação '{action}' enviada com sucesso: {data}")
+                    console.print(f"Ação '{action}' enviada com sucesso.", style="bold green")
+                    return data
+                else:
+                    err_msg = f"Erro ao enviar ação '{action}': {response.status} - {await response.text()}"
+                    logger.error(err_msg)
+                    console.print(err_msg, style="bold red")
+                    return {"error": err_msg}
+
+    except Exception as e:
+        err_msg = f"Erro ao comunicar com o endpoint RDP: {e}"
+        logger.error(err_msg)
+        console.print(err_msg, style="bold red")
+        return {"error": str(e)}

worker_automate_hub/api/rpa_fila_service.py

@@ -1,30 +1,34 @@
-from worker_automate_hub.config.settings import load_env_config
-from worker_automate_hub.utils.logger import logger
+import aiohttp
 from rich.console import Console
 
-import aiohttp
+from worker_automate_hub.config.settings import load_env_config
+from worker_automate_hub.models.dao.rpa_fila import RpaFila
+from worker_automate_hub.utils.logger import logger
 
 console = Console()
 
 
 async def burn_queue(id_fila: str):
     env_config, _ = load_env_config()
-    try:       
+    try:
 
         headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
-        
 
-        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=True)) as session:
+        async with aiohttp.ClientSession(
+            connector=aiohttp.TCPConnector(verify_ssl=True)
+        ) 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")    
+                    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")          
+                    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}"
@@ -34,16 +38,18 @@ async def burn_queue(id_fila: str):
             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"]}"}       
-        
+    try:
+        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
 
-        async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=True)) as session:
+        async with aiohttp.ClientSession(
+            connector=aiohttp.TCPConnector(verify_ssl=True)
+        ) as session:
             async with session.get(
-                f"{env_config["API_BASE_URL"]}/fila/unlock-queue/{id}",                
+                f"{env_config["API_BASE_URL"]}/fila/unlock-queue/{id}",
                 headers=headers_basic,
             ) as response:
                 return await response.text()
@@ -55,4 +61,4 @@ async def unlock_queue(id: str):
             f"{err_msg}\n",
             style="bold red",
         )
-        return None
+        return None

worker_automate_hub/api/rpa_historico_service.py

@@ -1,12 +1,147 @@
+from datetime import datetime
+
 import aiohttp
+from pytz import timezone
+from rich.console import Console
 
 from worker_automate_hub.config.settings import load_env_config
 from worker_automate_hub.models.dao.rpa_historico import RpaHistorico
+from worker_automate_hub.models.dao.rpa_processo import RpaProcesso
 from worker_automate_hub.models.dto.rpa_historico_request_dto import (
     RpaHistoricoRequestDTO,
+    RpaHistoricoStatusEnum,
+    RpaRetornoProcessoDTO,
+)
+from worker_automate_hub.models.dto.rpa_processo_entrada_dto import (
+    RpaProcessoEntradaDTO,
 )
 from worker_automate_hub.utils.logger import logger
 
+console = Console()
+
+
+async def create_store_historico(
+    task: RpaProcessoEntradaDTO,
+    processo: RpaProcesso,
+    status: RpaHistoricoStatusEnum,
+    retorno_processo: RpaRetornoProcessoDTO = None,
+) -> RpaHistorico:
+    """
+    Salva o histórico de um processo com status Processando.
+
+    Recebe um RpaProcessoEntradaDTO e um RpaProcesso como parâmetro e salva o
+    histórico com status Processando. Retorna um dicionário com o uuid do
+    histórico salvo.
+
+    Args:
+        task (RpaProcessoEntradaDTO): O processo a ser salvo.
+        processo (RpaProcesso): O processo que está sendo executado.
+
+    Returns:
+        RpaHistorico: Dicionário com o uuid do histórico salvo.
+    """
+    try:
+        from worker_automate_hub.config.settings import load_worker_config
+
+        worker_config = load_worker_config()
+        tz = timezone("America/Sao_Paulo")
+        start_time = datetime.now(tz).isoformat()
+
+        identificador_processo = (
+            task.configEntrada.get("nfe") or task.configEntrada.get("empresa") or ""
+        )
+
+        # Armazenar início da operação no histórico
+        start_data = RpaHistoricoRequestDTO(
+            uuidProcesso=task.uuidProcesso,
+            uuidRobo=worker_config["UUID_ROBO"],
+            prioridade=processo.prioridade,
+            desStatus=status,
+            configEntrada=task.configEntrada,
+            datInicioExecucao=start_time,
+            datEntradaFila=task.datEntradaFila,
+            identificador=identificador_processo,
+            retorno=retorno_processo,
+        )
+
+        store_response: RpaHistorico = await store(start_data)
+        console.print(
+            f"\nHistorico salvo com o uuid: {store_response.uuidHistorico}\n",
+            style="green",
+        )
+        return store_response
+    except Exception as e:
+        err_msg = f"Erro ao salvar o registro no histórico: {e}"
+        console.print(f"\n{err_msg}\n", style="red")
+        logger.error(f"{err_msg}")
+
+
+async def create_update_historico(
+    historico_uuid: str,
+    task: RpaProcessoEntradaDTO,
+    retorno_processo: RpaRetornoProcessoDTO,
+    processo: RpaProcesso,
+):
+    """
+    Atualiza o histórico de um processo com o status de sucesso ou falha.
+
+    Recebe o uuid do histórico, o RpaProcessoEntradaDTO do processo, um booleano
+    indicando se o processo foi um sucesso ou não, o RpaRetornoProcessoDTO do
+    processo e o RpaProcesso do processo como parâmetro e atualiza o histórico
+    com o status de sucesso ou falha. Retorna um dicionário com o uuid do
+    histórico atualizado.
+
+    Args:
+        historico_uuid (str): O uuid do histórico.
+        task (RpaProcessoEntradaDTO): O RpaProcessoEntradaDTO do processo.
+        sucesso (bool): Um booleano indicando se o processo foi um sucesso ou não.
+        retorno_processo (RpaRetornoProcessoDTO): O RpaRetornoProcessoDTO do processo.
+        processo (RpaProcesso): O RpaProcesso do processo.
+
+    Returns:
+        RpaHistorico: Dicionário com o uuid do histórico atualizado.
+    """
+
+    try:
+        from worker_automate_hub.config.settings import load_worker_config
+
+        worker_config = load_worker_config()
+        tz = timezone("America/Sao_Paulo")
+        des_status: RpaHistoricoStatusEnum = retorno_processo.status
+        end_time = datetime.now(tz).isoformat()
+
+        identificador_processo = (
+            task.configEntrada.get("nfe") or task.configEntrada.get("empresa") or ""
+        )
+        if not retorno_processo.tags:
+            retorno_processo.tags = []
+
+        # Armazenar fim da operação no histórico
+        end_data = RpaHistoricoRequestDTO(
+            uuidHistorico=historico_uuid,
+            uuidProcesso=task.uuidProcesso,
+            uuidRobo=worker_config["UUID_ROBO"],
+            prioridade=processo.prioridade,
+            desStatus=des_status,
+            configEntrada=task.configEntrada,
+            retorno=retorno_processo,
+            datFimExecucao=end_time,
+            identificador=identificador_processo,
+            tags=retorno_processo.tags,
+        )
+
+        update_response: RpaHistorico = await update(end_data)
+        console.print(
+            f"\nHistorico atualizado com o uuid: {update_response.uuidHistorico}\n",
+            style="green",
+        )
+        return update_response
+
+    except Exception as e:
+        err_msg = f"Erro ao atualizar o histórico do processo: {e}"
+        console.print(f"\n{err_msg}\n", style="red")
+        logger.error(err_msg)
+
 
 async def store(data: RpaHistoricoRequestDTO) -> dict:
     """

worker_automate_hub/models/dao/rpa_fila.py

@@ -0,0 +1,18 @@
+from datetime import datetime
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class RpaFila(BaseModel):
+    uuidFila: Optional[str] = Field(None, alias="uuidFila")
+    uuidRobo: Optional[str] = Field(None, alias="uuidRobo")
+    uuidProcesso: str = Field(..., alias="uuidProcesso")
+    prioridade: int = Field(..., alias="prioridade")
+    configEntrada: Optional[dict] = Field(None, alias="configEntrada")
+    dtLeituraFila: Optional[datetime] = Field(None, alias="dtLeituraFila")
+    lock: Optional[bool] = Field(None, alias="lock")
+    mutarNotificacao: Optional[int] = Field(None, alias="mutarNotificacao")
+
+    class Config:
+        populate_by_name = True

worker_automate_hub/tasks/jobs/fechar_conexao_rdp.py

@@ -0,0 +1,131 @@
+import asyncio
+import platform
+import subprocess
+import socket
+import pyautogui
+import pygetwindow as gw
+from rich.console import Console
+from pywinauto import Application
+from worker_automate_hub.api.rdp_service import send_rdp_action
+from worker_automate_hub.models.dto.rpa_historico_request_dto import RpaHistoricoStatusEnum, RpaRetornoProcessoDTO
+from worker_automate_hub.models.dto.rpa_processo_rdp_dto import RpaProcessoRdpDTO
+from worker_automate_hub.utils.logger import logger
+from worker_automate_hub.utils.util import worker_sleep
+
+console = Console()
+
+class RDPConnection:
+    def __init__(self, task: RpaProcessoRdpDTO):
+        self.task = task
+        self.ip = task.configEntrada.get("ip")
+        self.user = task.configEntrada.get("user")
+        self.password = task.configEntrada.get("password")
+        self.processo = task.configEntrada.get("processo")
+        self.uuid_robo = task.configEntrada.get("uuidRobo")
+
+    async def verificar_conexao(self) -> bool:
+        """
+        Verifica a conectividade com o host remoto via ping e verificação da porta RDP (3389).
+        """
+        sistema_operacional = platform.system().lower()
+        console.print(f"Sistema operacional detectado: {sistema_operacional}")
+
+        comando_ping = (
+            ["ping", "-n", "1", "-w", "1000", self.ip] if sistema_operacional == "windows" else ["ping", "-c", "1", "-W", "1", self.ip]
+        )
+        console.print(f"Executando comando de ping: {' '.join(comando_ping)}")
+
+        try:
+            resposta_ping = subprocess.run(comando_ping, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            ping_alcancado = resposta_ping.returncode == 0
+            console.print(f"Ping {'sucesso' if ping_alcancado else 'falhou'}")
+        except Exception as e:
+            console.print(f"Erro ao executar ping: {e}")
+            ping_alcancado = False
+
+        porta_aberta = False
+        try:
+            with socket.create_connection((self.ip, 3389), timeout=10):
+                console.print(f"Porta 3389 aberta em {self.ip}")
+                porta_aberta = True
+        except (socket.timeout, OSError) as e:
+            console.print(f"Erro ao verificar a porta RDP: {e}")
+
+        return ping_alcancado and porta_aberta
+
+    async def desligar(self):
+        """
+        Fecha a conexão RDP interagindo com a janela do cliente RDP.
+        """
+        janelas_rdp = [
+            win
+            for win in gw.getAllTitles()
+            if self.ip in win
+        ]
+
+        if not janelas_rdp:
+            console.print(f"Nenhuma janela RDP encontrada para o IP: {self.ip}")
+            return
+
+        for titulo in janelas_rdp:
+            janela = gw.getWindowsWithTitle(titulo)[0]
+            if not janela:
+                console.print(f"Erro ao localizar a janela: {titulo}")
+                continue
+
+            try:
+                janela = gw.getWindowsWithTitle(titulo)[0]
+                console.print(f"Processando janela: {titulo}")
+
+                app = Application(backend="uia").connect(title=titulo)
+                app_window = app.window(title=titulo)
+                console.print(f"Janela encontrada: {titulo}")
+                
+                if not app or not app_window:
+                    raise Exception("Nenhuma janela com título correspondente foi encontrada.")
+
+                app_window.set_focus()
+                console.print("Janela RDP ativada.")
+                
+                x, y = janela.left, janela.top
+
+                pyautogui.moveTo(x + 2, y + 2)
+                pyautogui.hotkey("alt", "space")
+                await worker_sleep(2)
+                pyautogui.press("down", presses=10, interval=0.1)
+                await worker_sleep(2)
+                pyautogui.press("enter")
+                await worker_sleep(2)
+                pyautogui.press("enter")
+                break
+
+            except Exception as e:
+                console.print(f"Erro ao interagir com a janela {titulo}: {e}")
+                raise(e)
+
+async def fechar_conexao_rdp(task: RpaProcessoRdpDTO) -> RpaRetornoProcessoDTO:
+        """
+        Gerencia o processo de fechamento da conexão RDP e retorna um status do processo.
+        """
+        try:
+            rdp_connection = RDPConnection(task)
+            console.print("Iniciando o fechamento da conexão RDP.")
+            conectado = await rdp_connection.verificar_conexao()
+
+            if not conectado:
+                msg = f"A máquina informada não está ligada. Verifique o IP: {rdp_connection.ip} e a disponibilidade da porta."
+                logger.warning(msg)
+                return RpaRetornoProcessoDTO(sucesso=False, retorno=msg, status=RpaHistoricoStatusEnum.Falha)
+
+            await rdp_connection.desligar()
+            await send_rdp_action(rdp_connection.uuid_robo, "stop")
+
+            return RpaRetornoProcessoDTO(
+                sucesso=True,
+                retorno="Conexão RDP encerrada com sucesso.",
+                status=RpaHistoricoStatusEnum.Sucesso,
+            )
+        except Exception as ex:
+            err_msg = f"Erro ao executar o fechamento da conexão RDP para a máquina {rdp_connection.ip}: {ex}"
+            console.print(err_msg)
+            return RpaRetornoProcessoDTO(sucesso=False, retorno=err_msg, status=RpaHistoricoStatusEnum.Falha)

worker_automate_hub/tasks/task_definitions.py

@@ -4,6 +4,7 @@ from worker_automate_hub.tasks.jobs.coleta_dje_process import (
     coleta_dje_start_update,
 )
 from worker_automate_hub.tasks.jobs.conexao_rdp import conexao_rdp
+from worker_automate_hub.tasks.jobs.fechar_conexao_rdp import fechar_conexao_rdp
 from worker_automate_hub.tasks.jobs.descartes import descartes
 from worker_automate_hub.tasks.jobs.ecac_estadual_main import (
     ecac_estadual_main,
@@ -61,6 +62,7 @@ task_definitions = {
     "abcfa1ba-d580-465a-aefb-c15ac4514407": descartes,
     "2c8ee738-7447-4517-aee7-ce2c9d25cea9": transferencias,
     "855f9e0f-e972-4f52-bc1a-60d1fc244e79": conexao_rdp,
+    "d36b0c83-9cc3-465f-ac80-934099a0e661": fechar_conexao_rdp,
     "81785803-0594-4bba-9aa0-7f220c200296": coleta_dje_start_update,
     "3907c8d4-d05b-4d92-b19a-2c4e934f1d78": ecac_estadual_main,
     "81d2d6e6-e9eb-414d-a939-d220476d2bab": ecac_federal,

worker_automate_hub/tasks/task_executor.py

@@ -1,19 +1,17 @@
-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 (
-    get_processo,
-    send_gchat_message,
+from worker_automate_hub.api.client import get_processo, send_gchat_message
+from worker_automate_hub.api.rpa_fila_service import burn_queue
+from worker_automate_hub.api.rpa_historico_service import (
+    create_store_historico,
+    create_update_historico,
 )
-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.api.webhook_service import send_to_webhook
+from worker_automate_hub.config.settings import load_worker_config
 from worker_automate_hub.models.dao.rpa_historico import RpaHistorico
 from worker_automate_hub.models.dao.rpa_processo import RpaProcesso
 from worker_automate_hub.models.dto.rpa_historico_request_dto import (
-    RpaHistoricoRequestDTO,
     RpaHistoricoStatusEnum,
     RpaRetornoProcessoDTO,
 )
@@ -24,9 +22,6 @@ from worker_automate_hub.tasks.task_definitions import task_definitions
 from worker_automate_hub.utils.logger import logger
 from worker_automate_hub.utils.toast import show_toast
 from worker_automate_hub.utils.util import capture_and_send_screenshot
-from worker_automate_hub.config.settings import (
-    load_worker_config,
-)
 
 console = Console()
 
@@ -52,7 +47,9 @@ async def perform_task(task: RpaProcessoEntradaDTO):
         registrar_historico = True
 
     if registrar_historico:
-        historico: RpaHistorico = await _store_historico(task, processo)
+        historico: RpaHistorico = await create_store_historico(
+            task, processo, RpaHistoricoStatusEnum.Processando
+        )
     try:
         if task_uuid in task_definitions:
             # Executar a task
@@ -61,7 +58,7 @@ async def perform_task(task: RpaProcessoEntradaDTO):
 
             result: RpaRetornoProcessoDTO = await task_definitions[task_uuid](task)
             if registrar_historico:
-                await _update_historico(
+                await create_update_historico(
                     historico_uuid=historico.uuidHistorico,
                     task=task,
                     retorno_processo=result,
@@ -87,19 +84,21 @@ async def perform_task(task: RpaProcessoEntradaDTO):
             show_toast("Erro", err_msg)
 
             if registrar_historico:
-                await _update_historico(
+                await create_update_historico(
                     historico_uuid=historico.uuidHistorico,
                     task=task,
                     retorno_processo=RpaRetornoProcessoDTO(
                         sucesso=False,
                         retorno=err_msg,
-                        status=RpaHistoricoStatusEnum.Falha,
+                        status=RpaHistoricoStatusEnum.Descartado,
                     ),
                     processo=processo,
                 )
-            await unlock_queue(task.uuidFila)
+            await burn_queue(task.uuidFila)
             if url_retorno is not None:
-                await send_to_webhook(url_retorno, RpaHistoricoStatusEnum.Falha, err_msg)
+                await send_to_webhook(
+                    url_retorno, RpaHistoricoStatusEnum.Descartado, err_msg
+                )
             return None
     except Exception as e:
         err_msg = f"Erro ao performar o processo: {e}"
@@ -108,7 +107,7 @@ async def perform_task(task: RpaProcessoEntradaDTO):
         show_toast("Erro", err_msg)
 
         if registrar_historico:
-            await _update_historico(
+            await create_update_historico(
                 historico_uuid=historico.uuidHistorico,
                 task=task,
                 retorno_processo=RpaRetornoProcessoDTO(
@@ -121,127 +120,3 @@ async def perform_task(task: RpaProcessoEntradaDTO):
         )
         if url_retorno is not None:
             await send_to_webhook(url_retorno, RpaHistoricoStatusEnum.Falha, err_msg)
-
-
-async def _store_historico(
-    task: RpaProcessoEntradaDTO, processo: RpaProcesso
-) -> RpaHistorico:
-    """
-    Salva o histórico de um processo com status Processando.
-
-    Recebe um RpaProcessoEntradaDTO e um RpaProcesso como parâmetro e salva o
-    histórico com status Processando. Retorna um dicionário com o uuid do
-    histórico salvo.
-
-    Args:
-        task (RpaProcessoEntradaDTO): O processo a ser salvo.
-        processo (RpaProcesso): O processo que está sendo executado.
-
-    Returns:
-        RpaHistorico: Dicionário com o uuid do histórico salvo.
-    """
-    try:
-        from worker_automate_hub.config.settings import load_worker_config
-
-        worker_config = load_worker_config()
-        tz = timezone("America/Sao_Paulo")
-        start_time = datetime.now(tz).isoformat()
-
-        identificador_processo = (
-            task.configEntrada.get("nfe") or 
-            task.configEntrada.get("empresa") or 
-            ""
-        )
-
-
-        # Armazenar início da operação no histórico
-        start_data = RpaHistoricoRequestDTO(
-            uuidProcesso=task.uuidProcesso,
-            uuidRobo=worker_config["UUID_ROBO"],
-            prioridade=processo.prioridade,
-            desStatus=RpaHistoricoStatusEnum.Processando,
-            configEntrada=task.configEntrada,
-            datInicioExecucao=start_time,
-            datEntradaFila=task.datEntradaFila,
-            identificador=identificador_processo,
-        )
-
-        store_response: RpaHistorico = await store(start_data)
-        console.print(
-            f"\nHistorico salvo com o uuid: {store_response.uuidHistorico}\n",
-            style="green",
-        )
-        return store_response
-    except Exception as e:
-        err_msg = f"Erro ao salvar o registro no histórico: {e}"
-        console.print(f"\n{err_msg}\n", style="red")
-        logger.error(f"{err_msg}")
-
-
-async def _update_historico(
-    historico_uuid: str,
-    task: RpaProcessoEntradaDTO,
-    retorno_processo: RpaRetornoProcessoDTO,
-    processo: RpaProcesso,
-):
-    """
-    Atualiza o histórico de um processo com o status de sucesso ou falha.
-
-    Recebe o uuid do histórico, o RpaProcessoEntradaDTO do processo, um booleano
-    indicando se o processo foi um sucesso ou não, o RpaRetornoProcessoDTO do
-    processo e o RpaProcesso do processo como parâmetro e atualiza o histórico
-    com o status de sucesso ou falha. Retorna um dicionário com o uuid do
-    histórico atualizado.
-
-    Args:
-        historico_uuid (str): O uuid do histórico.
-        task (RpaProcessoEntradaDTO): O RpaProcessoEntradaDTO do processo.
-        sucesso (bool): Um booleano indicando se o processo foi um sucesso ou não.
-        retorno_processo (RpaRetornoProcessoDTO): O RpaRetornoProcessoDTO do processo.
-        processo (RpaProcesso): O RpaProcesso do processo.
-
-    Returns:
-        RpaHistorico: Dicionário com o uuid do histórico atualizado.
-    """
-
-    try:
-        from worker_automate_hub.config.settings import load_worker_config
-
-        worker_config = load_worker_config()
-        tz = timezone("America/Sao_Paulo")
-        des_status: RpaHistoricoStatusEnum = retorno_processo.status
-        end_time = datetime.now(tz).isoformat()
-
-        identificador_processo = (
-            task.configEntrada.get("nfe") or 
-            task.configEntrada.get("empresa") or 
-            ""
-        )
-        if not retorno_processo.tags:
-            retorno_processo.tags = []
-
-        # Armazenar fim da operação no histórico
-        end_data = RpaHistoricoRequestDTO(
-            uuidHistorico=historico_uuid,
-            uuidProcesso=task.uuidProcesso,
-            uuidRobo=worker_config["UUID_ROBO"],
-            prioridade=processo.prioridade,
-            desStatus=des_status,
-            configEntrada=task.configEntrada,
-            retorno=retorno_processo,
-            datFimExecucao=end_time,
-            identificador=identificador_processo,
-            tags=retorno_processo.tags,
-        )
-
-        update_response: RpaHistorico = await update(end_data)
-        console.print(
-            f"\nHistorico atualizado com o uuid: {update_response.uuidHistorico}\n",
-            style="green",
-        )
-        return update_response
-
-    except Exception as e:
-        err_msg = f"Erro ao atualizar o histórico do processo: {e}"
-        console.print(f"\n{err_msg}\n", style="red")
-        logger.error(err_msg)

worker_automate_hub/worker.py

@@ -3,19 +3,30 @@ 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 (
     get_new_task,
+    get_processo,
     notify_is_alive,
     send_gchat_message,
 )
+from worker_automate_hub.api.rpa_fila_service import burn_queue
+from worker_automate_hub.api.rpa_historico_service import (
+    create_store_historico,
+)
 from worker_automate_hub.config.settings import (
     load_env_config,
     load_worker_config,
 )
+from worker_automate_hub.models.dao.rpa_processo import RpaProcesso
+from worker_automate_hub.models.dto.rpa_historico_request_dto import (
+    RpaHistoricoStatusEnum,
+    RpaRetornoProcessoDTO,
+    RpaTagDTO,
+    RpaTagEnum,
+)
 from worker_automate_hub.models.dto.rpa_processo_entrada_dto import (
     RpaProcessoEntradaDTO,
 )
@@ -40,15 +51,29 @@ async def check_and_execute_tasks(stop_event: threading.Event):
             worker_config = load_worker_config()
             if task is not None:
                 processo_existe = await is_uuid_in_tasks(task.uuidProcesso)
+                await burn_queue(task.uuidFila)
                 if processo_existe:
-                    await burn_queue(task.uuidFila)
                     logger.info(f"Executando a task: {task.nomProcesso}")
                     await perform_task(task)
                 else:
-                    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."
+                    log_message = f"O processo [{task.nomProcesso}] não existe no Worker [{worker_config['NOME_ROBO']}] e foi removido da fila."
                     console.print(f"\n{log_message}\n", style="yellow")
                     logger.error(log_message)
+                    try:
+                        processo: RpaProcesso = await get_processo(task.uuidProcesso)
+                        await create_store_historico(
+                            task,
+                            processo,
+                            RpaHistoricoStatusEnum.Descartado,
+                            retorno_processo=RpaRetornoProcessoDTO(
+                                sucesso=False,
+                                retorno=log_message,
+                                status=RpaHistoricoStatusEnum.Descartado,
+                                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+                            ),
+                        )
+                    except Exception as e:
+                        console.print(f"Erro ao salvar histórico de processo não implementado: {e}", style="bold red")
                     await send_gchat_message(log_message)
             else:
                 await asyncio.sleep(5)

{worker_automate_hub-0.5.9.dist-info → worker_automate_hub-0.5.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: worker-automate-hub
-Version: 0.5.9
+Version: 0.5.12
 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.9.dist-info → worker_automate_hub-0.5.12.dist-info}/RECORD

@@ -4,8 +4,9 @@ worker_automate_hub/api/ahead_service.py,sha256=0tX-i1ACRg3_yOI-_AfFEZ6FNU3L8Zb3
 worker_automate_hub/api/client.py,sha256=tFrIr4mpjvUJ1uU--BCiWa3NT2Ru_yliBsWmfiGt1LA,19720
 worker_automate_hub/api/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/api/helpers/api_helpers.py,sha256=SkheO2fXexeh-a4shr8Qzsz_kZhuSG0DJ7kbODctRbM,3696
-worker_automate_hub/api/rpa_fila_service.py,sha256=K_8EL5P6Y_CHs-f9ZGTsuUUjMd2fu4wN4NieHvMijKs,2100
-worker_automate_hub/api/rpa_historico_service.py,sha256=GZ3umxvSGNecQsffIRwgfmgpXHDFWq8EiBuAYqtrMNQ,4761
+worker_automate_hub/api/rdp_service.py,sha256=cDG3RP59U1m2rRX51IMqqKwkQ2-Rgl0PRJ8e4cXyhjs,1547
+worker_automate_hub/api/rpa_fila_service.py,sha256=6RyzCvj2qODifz3cIwOg5kuUpk8cyN_yFmLV5YOJAoQ,2186
+worker_automate_hub/api/rpa_historico_service.py,sha256=wQw8_k8i0S-r7qrrn4BSoztkgtiuH_bHRSEytbZebDg,9690
 worker_automate_hub/api/webhook_service.py,sha256=9KvTv1U02CSYpf_vLnqs4RZqZzR7Mn0v1OVOI451yOM,1762
 worker_automate_hub/cli.py,sha256=JB45pjPJ8_E-4xw0OjqDMcAw-tpDV0mjmvwJRTnTzY0,6862
 worker_automate_hub/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,6 +23,7 @@ worker_automate_hub/decorators/timeit.py,sha256=lL2bK8TbF4GE5Ma3LI5N5iJxGF1vwlZ4
 worker_automate_hub/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/models/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/models/dao/rpa_configuracao.py,sha256=pftPsgbqsDjN8pRrzcjBKKfXaHPFA7KkJx9gl8LW7lk,328
+worker_automate_hub/models/dao/rpa_fila.py,sha256=ZV-0ykAPKjMRTvgmts3WwPQQlCRv6h4mYr6Ki8IeoEE,699
 worker_automate_hub/models/dao/rpa_historico.py,sha256=zKo_Wz_qMdB-cmFkJY1ckFx63wBPUumir37OxFijhes,1071
 worker_automate_hub/models/dao/rpa_processo.py,sha256=AfOUhJKL-AAOyjON4g7f6B7Bxa174FNC1Eh7oItK2kA,1233
 worker_automate_hub/models/dto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -53,14 +55,15 @@ worker_automate_hub/tasks/jobs/entrada_de_notas_505.py,sha256=jIml8gjXPdI6_x7S9V
 worker_automate_hub/tasks/jobs/entrada_de_notas_7139.py,sha256=oxH2CUJj3UyhYoSxEY2hUr_6wFgZPLcHswEbApIu8VY,15558
 worker_automate_hub/tasks/jobs/entrada_de_notas_9.py,sha256=Urt_7dZNuyUkSyG9fYb2ArBXzEONXlrqWIesoXeL1EI,50984
 worker_automate_hub/tasks/jobs/exemplo_processo.py,sha256=3-zxbb-9YHPmSA_K1Qgxp_FwSqg2QDjGBRCLxDZ8QoQ,3451
+worker_automate_hub/tasks/jobs/fechar_conexao_rdp.py,sha256=80FnDN6n_NMmDdQn6mmXEExRpK4BiYPrF0-C48Y1MBc,5445
 worker_automate_hub/tasks/jobs/fidc_gerar_nosso_numero.py,sha256=mZ6mCaz1lYmyFxwC8MVLXPsgynE1VkNKlEzYGsciDiY,10179
 worker_automate_hub/tasks/jobs/fidc_remessa_cobranca_cnab240.py,sha256=QBGm6eS5JghgNWNqZlk1g2a2iV8LnBLiOTBBL3Giet0,4181
 worker_automate_hub/tasks/jobs/login_emsys.py,sha256=IoGCIvO4UwmuxOZEn3cvYJlKyhsWvtHvbFk8vwjTroQ,5620
 worker_automate_hub/tasks/jobs/playground.py,sha256=bdnXv3C7WLQUxt4edGZDfAbRJJ2-q4zuIQaK3GLnaUc,1765
 worker_automate_hub/tasks/jobs/sped_fiscal.py,sha256=_byvD7i_N3pgPjQd1lc0XNKCFONIPbmvX3_jq5AnbKY,26707
 worker_automate_hub/tasks/jobs/transferencias.py,sha256=zwCbVTwX15SCeLtvYgAyENeUSIuETx4Z9Cysr1Es8Jo,38013
-worker_automate_hub/tasks/task_definitions.py,sha256=2Jp1H4_qJZqqGyaP6MA87KLt4QNrtWBYWbXu-2gymFo,4459
-worker_automate_hub/tasks/task_executor.py,sha256=3haqLIb5ZPOpr1g20SED0b6sm_pGFinmZMpE4AJRrlo,9582
+worker_automate_hub/tasks/task_definitions.py,sha256=KG84tDaTlCpthdYmdR1193BOudELGcoglmIpQpd2PLQ,4606
+worker_automate_hub/tasks/task_executor.py,sha256=tVmuU-MWDkgbGg3uSK5IBEb_tJqGSnYZpwszdGUofOM,5150
 worker_automate_hub/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/utils/env.py,sha256=TacQjGRO7PUNpttrhTAc5Gnegaiysl2Knsv1P8qfkfs,57
 worker_automate_hub/utils/get_creds_gworkspace.py,sha256=ZJ0IIEjM4IXIV9rwfbOZ1V1wiaMoJAGZeSy0D37sYdU,2212
@@ -69,8 +72,8 @@ worker_automate_hub/utils/toast.py,sha256=xPHc5r5uOxB_cZlCzm13Kt2qSKLLFZALncU6Qg
 worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbVlwDl49Y,7972
 worker_automate_hub/utils/util.py,sha256=3XNBaYVbJ-KgWyHDKY2jIFuQzgBQs-w1d5iEQZv1D4U,125867
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=p5r_L5k7gqCBvV8v6dbLFM6INKiSpGc4Gegid0_YAh4,29017
-worker_automate_hub/worker.py,sha256=tftQpX8liC-_0_bOUf1YYzXSCvloMQBvjmQ6lzfEE-c,4816
-worker_automate_hub-0.5.9.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.9.dist-info/METADATA,sha256=epLf1egEW1wK7rqtYA-5isr6KMSPGRRsu7yb0m0TFls,2893
-worker_automate_hub-0.5.9.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-worker_automate_hub-0.5.9.dist-info/RECORD,,
+worker_automate_hub/worker.py,sha256=ojfBsKdCuUAQS5_F3eZVMLV015GmtrasVvxKsmo2rx8,5942
+worker_automate_hub-0.5.12.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.12.dist-info/METADATA,sha256=VEzm6fiv2ZT3qD3obSxZdLPWw2Hza8yzOnlszMui0JE,2894
+worker_automate_hub-0.5.12.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+worker_automate_hub-0.5.12.dist-info/RECORD,,