worker-automate-hub 0.5.66__tar.gz → 0.5.67__tar.gz

{worker_automate_hub-0.5.66 → worker_automate_hub-0.5.67}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.66
+Version: 0.5.67
 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.66 → worker_automate_hub-0.5.67}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.66"
+version = "0.5.67"
 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.66 → worker_automate_hub-0.5.67}/worker_automate_hub/tasks/jobs/conexao_rdp.py

@@ -1,5 +1,6 @@
 import asyncio
 import platform
+import re
 import subprocess
 import socket
 import pyautogui
@@ -7,7 +8,7 @@ import os
 import pygetwindow as gw
 from rich.console import Console
 import pygetwindow as gw
-from pywinauto import Application
+from pywinauto import Application, Desktop
 from worker_automate_hub.api.client import change_robot_status_true 
 from worker_automate_hub.models.dto.rpa_historico_request_dto import RpaHistoricoStatusEnum, RpaRetornoProcessoDTO, RpaTagDTO, RpaTagEnum
 from worker_automate_hub.models.dto.rpa_processo_rdp_dto import RpaProcessoRdpDTO
@@ -127,6 +128,23 @@ class RDPConnection:
         except Exception as error:
             console.print(f"Erro ao abrir a conexão RDP: {error}")
 
+    async def verificar_conexao_estabelecida(self):
+        janelas_rdp = [
+            win
+            for win in gw.getAllTitles()
+            if self.ip in win
+        ]
+
+        janela = None
+
+        for titulo in janelas_rdp:
+            janela = gw.getWindowsWithTitle(titulo)[0]
+            if not janela:
+                console.print(f"Erro ao localizar a janela: {titulo}")
+                continue
+            else:
+                return True
+
     async def conectar(self):
         console.print(f"Iniciando cliente RDP para {self.ip}")
         try:
@@ -163,8 +181,41 @@ class RDPConnection:
             except Exception as e:
                 console.print(f"Erro ao ativar a janela RDP: {e}", style="bold red")
 
+            app_window = None
+            possible_titles_patterns = [
+                re.compile(r"Conexão de Área de Trabalho Remota", re.IGNORECASE),
+                re.compile(r"Ligação ao ambiente de trabalho remoto", re.IGNORECASE),
+                re.compile(r"Segurança do Windows", re.IGNORECASE)
+            ]
+
+            janelas = Desktop(backend="uia").windows()
+            for janela in janelas:
+                titulo = janela.window_text()
+                for pattern in possible_titles_patterns:
+                    if pattern.search(titulo):
+                        try:
+                            app = Application(backend="uia").connect(title_re=pattern)
+                            app_window = app.window(title_re=pattern)
+                            console.print(f"Janela encontrada: {titulo}")
+                            break
+                        except Exception as e:
+                            console.print(f"Erro ao conectar a janela '{titulo}': {e}")
+                if app_window:
+                    break
+
+            await worker_sleep(2)
+
+            if app_window:
+                try:
+                    app_window.restore()
+                    app_window.set_focus()
+                    console.print("Foco definido com sucesso.")
+                except Exception as e:
+                    console.print(f"Erro ao definir foco na janela: {e}")
+            else:
+                console.print("Nenhuma janela correspondente foi encontrada.")
+
             await worker_sleep(5)
-           
             console.print("Inserindo senha.")
             pyautogui.write(self.password, interval=0.1)
             pyautogui.press("enter")
@@ -175,8 +226,6 @@ class RDPConnection:
             console.print("Credenciais inseridas.")
             await worker_sleep(5)
             
-            console.print("Conexão RDP ativa. Mantendo o script em execução.")
-            
             try:
                 base_path = os.path.join("worker_automate_hub", "tasks", "jobs")
                 rdp_file_path = os.path.join(base_path, "connection.rdp")
@@ -184,8 +233,13 @@ class RDPConnection:
                     os.remove(rdp_file_path)
             except Exception as error:
                 console.print(f"Erro ao deletar arquivo 'connection.rdp'. Error: {error}")
+
+            conexao_rdp_aberta = await self.verificar_conexao_estabelecida()
+            if conexao_rdp_aberta is not True:
+                raise ConnectionError("Não foi possível estabelecer conexão RDP.")
             
             return True
+
         except Exception as e:
             logger.error(f"Erro ao tentar conectar via RDP: {e}")
             return False
@@ -202,6 +256,7 @@ async def conexao_rdp(task: RpaProcessoRdpDTO) -> RpaRetornoProcessoDTO:
                 retorno="Não foi possível estabelecer conexão RDP. Verifique o IP e a disponibilidade da porta.",
                 status=RpaHistoricoStatusEnum.Falha, tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
             )
+
         sucesso_conexao = await rdp.conectar()
         if sucesso_conexao:
             console.print("Processo de conexão RDP executado com sucesso.")
@@ -224,7 +279,7 @@ async def conexao_rdp(task: RpaProcessoRdpDTO) -> RpaRetornoProcessoDTO:
                 status=RpaHistoricoStatusEnum.Falha, tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
             )
     except Exception as ex:
-        err_msg = f"Erro ao executar conexao_rdp: {ex}"
+        err_msg = f"Erro ao executar conexão RDP: {ex}"
         logger.error(err_msg)
         console.print(err_msg, style="bold red")
         return RpaRetornoProcessoDTO(