worker-automate-hub 0.5.9__tar.gz → 0.5.10__tar.gz

{worker_automate_hub-0.5.9 → worker_automate_hub-0.5.10}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.9"
+version = "0.5.10"
 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.10/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-0.5.10/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("uuid_robo")
+
+    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-0.5.9 → worker_automate_hub-0.5.10}/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,