worker-automate-hub 0.4.448__py3-none-any.whl → 0.4.449__py3-none-any.whl

worker_automate_hub/tasks/jobs/conexao_rdp.py

@@ -3,6 +3,7 @@ import platform
 import subprocess
 import socket
 import pyautogui
+import os
 import pygetwindow as gw
 from rich.console import Console
 import pygetwindow as gw
@@ -55,16 +56,10 @@ class RDPConnection:
         return ping_alcancado and porta_aberta
     
     async def clicar_no_titulo(self):
-        """
-        Função para clicar no título das janelas de Conexão de Área de Trabalho Remota
-        com o botão direito do mouse e executar comandos.
-        """
         janelas_rdp = [
             win
             for win in gw.getAllTitles()
-            if "Ligação ao Ambiente de Trabalho Remoto" in win
-            or "Remote Desktop Connection" in win
-            or "Conexão de Área de Trabalho Remota" in win
+            if self.ip in win
         ]
 
         for titulo in janelas_rdp:
@@ -74,88 +69,98 @@ class RDPConnection:
                 continue
 
             console.print(f"Processando janela: {titulo}")
-
-            # Obtém as coordenadas da janela
             x, y = janela.left, janela.top
 
             try:
-                # Move o mouse para o título da janela e clica com o botão direito
-                pyautogui.moveTo(x + 10, y + 10)  # Ajuste para alinhar ao título da janela
-                pyautogui.click(button="right")
-                
-                await worker_sleep(5)
-                pyautogui.press("down", presses=7, interval=0.1)
-                await worker_sleep(5)
+                pyautogui.moveTo(x + 2, y + 2)
+                pyautogui.hotkey("alt", "space")
+                await worker_sleep(2)
+                pyautogui.press("down", presses=6, interval=0.1)
+                await worker_sleep(2)
                 pyautogui.hotkey("enter")
-                await worker_sleep(5)
+                await worker_sleep(2)
                 pyautogui.hotkey("enter")
                 break
 
             except Exception as e:
                 console.print(f"Erro ao interagir com a janela {titulo}: {e}")
-    
+
+    async def open_rdp(self):
+        base_path = os.path.join("worker_automate_hub", "tasks", "jobs")
+        rdp_file_path = os.path.join(base_path, "connection.rdp")
+        os.makedirs(base_path, exist_ok=True)
+        
+        rdp_content = f"""screen mode id:i:1
+            desktopwidth:i:1920
+            desktopheight:i:1080
+            use multimon:i:0
+            session bpp:i:32
+            compression:i:1
+            keyboardhook:i:2
+            displayconnectionbar:i:1
+            disable wallpaper:i:0
+            allow font smoothing:i:1
+            allow desktop composition:i:1
+            disable full window drag:i:0
+            disable menu anims:i:0
+            disable themes:i:0
+            disable cursor setting:i:0
+            bitmapcachepersistenable:i:1
+            full address:s:{self.ip}
+            username:s:{self.user}
+            winposstr:s:0,1,{(1920 // 4)},{(1080 // 4)},{(1920 // 4) + (1920 // 2)},{(1080 // 4) + (1080 // 2)}
+        """
+
+        try:
+            with open(rdp_file_path, "w") as file:
+                file.write(rdp_content)
+            print(f"Arquivo RDP criado: {rdp_file_path}")
+        except Exception as error:
+            print(f"Erro ao montar o arquivo RDP: {error}")
+            return
+
+        try:
+            subprocess.Popen(["mstsc", rdp_file_path], close_fds=True, start_new_session=True)
+            console.print("Conexão RDP iniciada.")
+        except Exception as error:
+            console.print(f"Erro ao abrir a conexão RDP: {error}")
+
     async def conectar(self):
         console.print(f"Iniciando cliente RDP para {self.ip}")
         try:
-            pyautogui.hotkey("win", "d")
-            await worker_sleep(5)  # Tempo para garantir que todas as janelas sejam minimizadas
-            console.print("Todas as janelas minimizadas com sucesso.")
-        
-            subprocess.Popen(["mstsc", f"/v:{self.ip}"], close_fds=True, start_new_session=True)
-            await worker_sleep(10)  # Tempo aumentado para garantir abertura
-            
-            console.print("Procurando janela 'Ligação ao ambiente de trabalho remoto'")
-            try:
-                windows = gw.getWindowsWithTitle("Ligação ao ambiente de trabalho remoto")
-            except:
-                windows = gw.getWindowsWithTitle("Conexão de Área de Trabalho Remota")
-                
-            if not windows:
-                logger.warning("Tentando encontrar janela com título em inglês 'Remote Desktop Connection'")
-                windows = gw.getWindowsWithTitle("Remote Desktop Connection")
-            
-            if not windows:
-                logger.error("Janela de RDP não encontrada.")
-                return False
-            
-            rdp_window = windows[0]
-            console.print(f"Janela '{rdp_window.title}' encontrada. Focando na janela.")
-            
-            # Restaurar janela se estiver minimizada
-            if rdp_window.isMinimized:
-                rdp_window.restore()
-                await worker_sleep(5)
-            
-            # Redimensionar para 50% da tela
-            screen_width, screen_height = pyautogui.size()
-            new_width = screen_width // 2
-            new_height = screen_height // 2
-            rdp_window.resizeTo(new_width, new_height)
-            rdp_window.moveTo(screen_width // 4, screen_height // 4)
-            console.print(f"Janela redimensionada para {new_width}x{new_height}.")
-            
-            rdp_window.activate()
-            await worker_sleep(5)  # Tempo extra para garantir que a janela está ativa
-            
-            # Clique para garantir o foco
-            pyautogui.click(rdp_window.left + 50, rdp_window.top + 50)
             await worker_sleep(3)
+            await self.open_rdp()
+            await worker_sleep(5)
             
-            # Inserir credenciais
-            
+            console.print("Procurando janela RDP")
             try:
-                app = Application(backend="uia").connect(title="Segurança do Windows")
-                dialog = app.window(title="Segurança do Windows")
-
-                edit_field = dialog.child_window(auto_id="EditField_1", control_type="Edit")
-
-                if edit_field.exists():
-                    console.print("Inserindo usuário.")
-                    pyautogui.write(self.user, interval=0.1)
-                    pyautogui.press("tab")
-                    await worker_sleep(5)
-            except:
-                pass
+                possible_titles = ["Conexão de Área de Trabalho Remota", "Ligação ao ambiente de trabalho remoto"]
+                app = None
+                app_window = None
+
+                for title in possible_titles:
+                    try:
+                        app = Application(backend="uia").connect(title=title)
+                        app_window = app.window(title=title)
+                        console.print(f"Janela encontrada: {title}")
+                        break
+                    except Exception:
+                        continue
+
+                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.")
+
+                pyautogui.press("left")
+                await worker_sleep(1)
+                pyautogui.hotkey("enter")
+                await worker_sleep(3)
+
+            except Exception as e:
+                console.print(f"Erro ao ativar a janela RDP: {e}", style="bold red")
+
            
             console.print("Inserindo senha.")
             pyautogui.write(self.password, interval=0.1)
@@ -165,9 +170,18 @@ class RDPConnection:
             await worker_sleep(5)
             pyautogui.press("enter")
             console.print("Credenciais inseridas.")
-            await worker_sleep(5)  # Tempo para conexão ser concluída
+            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")
+                if os.path.exists(rdp_file_path):
+                    os.remove(rdp_file_path)
+            except Exception as error:
+                console.print(f"Erro ao deletar arquivo 'connection.rdp'. Error: {error}")
+            
             return True
         except Exception as e:
             logger.error(f"Erro ao tentar conectar via RDP: {e}")
@@ -191,7 +205,6 @@ async def conexao_rdp(task: RpaProcessoRdpDTO) -> RpaRetornoProcessoDTO:
             
             await rdp.clicar_no_titulo()
             
-            # Mantém o script ativo para manter a conexão RDP aberta
             return RpaRetornoProcessoDTO(
                 sucesso=True,
                 retorno="Conexão RDP estabelecida com sucesso.",

{worker_automate_hub-0.4.448.dist-info → worker_automate_hub-0.4.449.dist-info}/METADATA

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

@@ -32,7 +32,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=4eEf3Hsm9TCGESn5cK-uSpHwbCYeCOBG44Q4cBYwmvc,9293
+worker_automate_hub/tasks/jobs/conexao_rdp.py,sha256=vbbEsGdpgISBBVnJo8FI81zqzponqEPIAT1kNLsxizQ,9183
 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
@@ -70,7 +70,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 worker_automate_hub/utils/util.py,sha256=JkSKFgnI6geDEIlpaJJ5Btd7XdPllfUSFcjvPAaG-bo,125954
 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.4.448.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.4.448.dist-info/METADATA,sha256=xolmgP4uWa1s6in1Ou0z7VzqrwKBwt5Ru_XiQqCzkZw,2895
-worker_automate_hub-0.4.448.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-worker_automate_hub-0.4.448.dist-info/RECORD,,
+worker_automate_hub-0.4.449.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.4.449.dist-info/METADATA,sha256=VrWE59j1n-zIdItKYymXrHmkzwmCADS3IXdY8LfuM7w,2895
+worker_automate_hub-0.4.449.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+worker_automate_hub-0.4.449.dist-info/RECORD,,