worker-automate-hub 0.5.453__tar.gz → 0.5.454__tar.gz

{worker_automate_hub-0.5.453 → worker_automate_hub-0.5.454}/PKG-INFO

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

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

@@ -7,16 +7,16 @@ from worker_automate_hub.models.dto.rpa_processo_entrada_dto import RpaProcessoE
 from worker_automate_hub.api.client import (
     get_config_by_name
 )
-
 from pywinauto import Application
 from rich.console import Console
-from worker_automate_hub.utils.util import is_window_open_by_class, kill_contabil_processes, kill_process, login_emsys_fiscal, type_text_into_field, worker_sleep, wait_window_close
+from worker_automate_hub.utils.util import is_window_open_by_class, kill_contabil_processes, kill_process, login_emsys_fiscal, type_text_into_field, worker_sleep
 from worker_automate_hub.utils.utils_nfe_entrada import EMSys
 import pyperclip
 import warnings
 import asyncio
 from pytesseract import image_to_string
-from PIL import ImageEnhance, ImageFilter
+from pywinauto import Desktop
+
 pyautogui.PAUSE = 0.5
 pyautogui.FAILSAFE = False
 
@@ -24,42 +24,33 @@ console = Console()
 
 emsys = EMSys()
 
-async def wait_aguarde_window_closed(app):
-
-    app = Application().connect(class_name="TFrmAguardeSys", timeout=60)
-
-    max_attempts = 60
-    i = 0
-
-    while i < max_attempts:
-        try:
-            main_window = app["TMessageForm"]
-            console.print("Janela 'TMessageForm' encontrada!")
-            break
-        except Exception as e:
-            console.print(f"Erro ao tentar acessar TMessageForm: {e}")
-            janela_aguarde = await is_window_open_by_class(
-                "TFrmAguardeSys", "TFrmAguardeSys"
-            )
-            if janela_aguarde["IsOpened"] == True:
-                console.print(
-                    "Aguardando tela de aguarde desaparecer ou conectar...\n"
-                )
-            else:
-                try:
-                    main_window = app["TMessageForm"]
-                    if main_window.exists():
-                        console.print("Janela 'TMessageForm' encontrada!")
-                        break
-                except:
-                    return RpaRetornoProcessoDTO(
-                        sucesso=True,
-                        retorno="Tela de aguardar carregada - Seguindo com andamento do processo (NCM).",
-                        status=RpaHistoricoStatusEnum.Sucesso,
-                    )
+async def wait_aguarde_window_closed():
+    sucesso = False
+    while not sucesso:
+        desktop = Desktop(backend="uia")
+        # Tenta localizar a janela com o título que contém "Aguarde"
+        window = desktop.window(title_re="Aguarde...")
+        # Se a janela existe, continua monitorando
+        if window.exists():
+            console.print(f"Janela 'Aguarde...' ainda aberta", style="bold yellow")
+            logger.info(f"Janela 'Aguarde...' ainda aberta")
+            await worker_sleep(5)
+            continue
+        else:
+            try:
+                desktop_second = Desktop(backend="uia")
+                window_aguarde = desktop_second.window(title_re="Aguarde...")
+                if not window_aguarde.exists():
+                    sucesso = True
+                    break
+                else:
+                    console.print(f"Janela 'Aguarde...' ainda aberta. Seguindo para próxima tentativa", style="bold yellow")
+                    continue
+            except:
+                console.print(f"Janela 'Aguarde...' não existe mais.", style="bold green")
+                await worker_sleep(5)
+                break
 
-        await worker_sleep(5)
-        i += 1
 
 def click_desconfirmar():
     cords = (675, 748)
@@ -107,6 +98,8 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             )
             pyautogui.press("enter")
             await worker_sleep(2)
+            pyautogui.press("down")
+            await worker_sleep(2)
             pyautogui.press("enter")
             console.print(
                 "\nPesquisa: 'Livros Fiscais' realizada com sucesso.",
@@ -166,8 +159,8 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
             # Esperando janela aguarde
             console.print("Aguardando tela de aguarde ser finalizada")
-            await wait_aguarde_window_closed(app)
-            await worker_sleep(3)
+            await wait_aguarde_window_closed()
+            await worker_sleep(10)
             
             console.print("Fechando possivel janela de aviso...")
             await emsys.verify_warning_and_error("Aviso", "&Ok")
@@ -184,8 +177,8 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             
             # Esperando janela aguarde
             console.print("Aguardando tela de aguarde ser finalizada")
-            await wait_aguarde_window_closed(app)
-            await worker_sleep(5)
+            await wait_aguarde_window_closed()
+            await worker_sleep(10)
             
             await worker_sleep(2)
             # Clicando OK em janela de livro incluido