worker-automate-hub 0.5.8__py3-none-any.whl → 0.5.10__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/tasks/jobs/conexao_rdp.py

@@ -24,7 +24,7 @@ class RDPConnection:
         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")
+        self.uuid_robo = task.configEntrada.get("uuidRobo")
     
     async def verificar_conexao(self) -> bool:
         sistema_operacional = platform.system().lower()
@@ -206,7 +206,7 @@ async def conexao_rdp(task: RpaProcessoRdpDTO) -> RpaRetornoProcessoDTO:
         if sucesso_conexao:
             console.print("Processo de conexão RDP executado com sucesso.")
             
-            await rdp.clicar_no_titulo()
+            #await rdp.clicar_no_titulo()
 
             #Altera coluna ativo para true or false 
             await change_robot_status_true(rdp.uuid_robo)

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/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-0.5.8.dist-info → worker_automate_hub-0.5.10.dist-info}/METADATA

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

@@ -4,6 +4,7 @@ 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/rdp_service.py,sha256=cDG3RP59U1m2rRX51IMqqKwkQ2-Rgl0PRJ8e4cXyhjs,1547
 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/webhook_service.py,sha256=9KvTv1U02CSYpf_vLnqs4RZqZzR7Mn0v1OVOI451yOM,1762
@@ -32,7 +33,7 @@ worker_automate_hub/models/dto/rpa_sistema_dto.py,sha256=sLkmJei8x6sl-1-IXUKDmYQ
 worker_automate_hub/tasks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/tasks/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/tasks/jobs/coleta_dje_process.py,sha256=rf4fW-FaHUl1MS7b03z4cwI3zHfNw8FWxvjvNY3Xn20,28773
-worker_automate_hub/tasks/jobs/conexao_rdp.py,sha256=l0lF8v1xMQSHX3506TFQJd5eZX2dLEo2jDzZG8jyeDc,9421
+worker_automate_hub/tasks/jobs/conexao_rdp.py,sha256=fUjTQ1mcU-whoN23tCdny4CVF2SNN_g1vk2WpIxg5e8,9421
 worker_automate_hub/tasks/jobs/descartes.py,sha256=RIjrZIzkW77NiKeXbxFN9k872cz3fl9cG1m5TJpjmmY,40134
 worker_automate_hub/tasks/jobs/ecac_estadual_go.py,sha256=aPckQRlRozFS_OK3C9wNdMCmqO6AM4djwqY2uSSaPmo,20687
 worker_automate_hub/tasks/jobs/ecac_estadual_main.py,sha256=FFpAdtZLO4uelWZooCVpm4JePv_iDt5nwVKrk1ipZJQ,1599
@@ -53,13 +54,14 @@ 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=zdr988qpz0RSY88yjxLOCwhSIvAUW83qGAgvqK8jZN0,5446
 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_definitions.py,sha256=KG84tDaTlCpthdYmdR1193BOudELGcoglmIpQpd2PLQ,4606
 worker_automate_hub/tasks/task_executor.py,sha256=3haqLIb5ZPOpr1g20SED0b6sm_pGFinmZMpE4AJRrlo,9582
 worker_automate_hub/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/utils/env.py,sha256=TacQjGRO7PUNpttrhTAc5Gnegaiysl2Knsv1P8qfkfs,57
@@ -70,7 +72,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 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.8.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.8.dist-info/METADATA,sha256=J08lr7fCd3ifeDcKcXZDLSQcADVFUIvJsQfz5guQr1I,2893
-worker_automate_hub-0.5.8.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-worker_automate_hub-0.5.8.dist-info/RECORD,,
+worker_automate_hub-0.5.10.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.10.dist-info/METADATA,sha256=wuiX3o9vZLo-k3Gz67TSZL3DpBjM9H2zhOUyGHsKKFI,2894
+worker_automate_hub-0.5.10.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+worker_automate_hub-0.5.10.dist-info/RECORD,,