worker-automate-hub 0.5.799__py3-none-any.whl → 0.5.801__py3-none-any.whl

worker_automate_hub/tasks/jobs/abertura_livros_fiscais.py

@@ -7,6 +7,7 @@ import sys
 import os
 from pywinauto.findwindows import ElementNotFoundError
 from pywinauto.keyboard import send_keys
+
 # sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..')))
 from worker_automate_hub.utils.logger import logger
 from worker_automate_hub.models.dto.rpa_historico_request_dto import (
@@ -202,6 +203,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             saida_checkbox.click_input()
 
             console.print("Aguardar marcar caixa de saida")
+            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\saida_marcada.png"
             imagem = "assets\\abertura_livros\\saida_marcada.png"
 
             tempo_limite = 600  # 10 minutos
@@ -338,6 +340,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             await worker_sleep(5)
 
             console.print("Aguardar o término de carregar")
+            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\livros_incluidos.png"
             imagem = "assets\\abertura_livros\\livros_incluidos.png"
 
             tempo_limite = 1200  # 20 minutos
@@ -558,45 +561,132 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
             console.print("Clicando no botão incluir apuração")
             # Clicar em incluir apuração
-            # imagem = "assets\\abertura_livros\\btn_incluir_apuracao.png"
+            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\btn_incluir_apuracao.png"
             imagem = "assets\\abertura_livros\\btn_incluir_apuracao.png"
 
             # Tenta localizar a imagem na tela
-            localizacao = pyautogui.locateCenterOnScreen(
-                imagem, confidence=0.9
-            )  
-  
+            localizacao = pyautogui.locateCenterOnScreen(imagem, confidence=0.9)
+
             if localizacao:
-                print(f"Imagem incluir apuração encontrado em: {localizacao}")
+                console.print(f"Imagem incluir apuração encontrado em: {localizacao}")
                 pyautogui.moveTo(localizacao)
                 pyautogui.click()
+                console.print("Botão incluir clicado com sucesso")
 
-                await worker_sleep(10)
+                await worker_sleep(2)
 
-                imagem = "assets\\abertura_livros\\aviso_erro.png"
+                desk = Desktop(backend="win32")
+                dlg = None
+                t_end = time.time() + 8.0  # até 8s procurando a janela
+                while time.time() < t_end:
+                    try:
+                        # variações: Aviso | Atenção | Erro
+                        wins = desk.windows(
+                            title_re=".*(Aviso|Aten[cç][aã]o|Erro).*", visible_only=True
+                        )
+                        if wins:
+                            dlg = wins[0]
+                            break
+                    except Exception:
+                        pass
+                    time.sleep(0.25)
+
+                if dlg:
+                    console.print(
+                        "Janela 'Aviso' detectada. Capturando texto e screenshot..."
+                    )
 
-                # Tenta localizar a imagem na tela
-                localizacao_erro = pyautogui.locateCenterOnScreen(
-                    imagem, confidence=0.9
-                )  
-    
-                if localizacao_erro:
-                    return RpaRetornoProcessoDTO(
-                    sucesso=False,
-                    retorno=f"Erro ao Incluir apuração, situação diferente de confirmado.",
-                    status=RpaHistoricoStatusEnum.Falha,
-                    tags=[RpaTagDTO(descricao=RpaTagEnum.Negocio)],
-                )
+                    # === 2) Captura de screenshot da janela ===
+                    try:
+                        # pasta de prints (ajuste se quiser outra)
+                        prints_dir = os.path.join(os.getcwd(), "prints")
+                        os.makedirs(prints_dir, exist_ok=True)
+                        ts = datetime.now().strftime("%Y%m%d_%H%M%S")
+                        screenshot_path = os.path.join(prints_dir, f"aviso_{ts}.png")
+                        dlg.set_focus()
+                        img = dlg.capture_as_image()
+                        img.save(screenshot_path)
+                        console.print(f"Screenshot salvo: {screenshot_path}")
+                    except Exception as e:
+                        screenshot_path = None
+                        console.print(f"Falha ao salvar screenshot do 'Aviso': {e}")
+
+                    # === 3) Lê todo o texto da janela ===
+                    texto_aviso = ""
+                    try:
+                        partes = []
+                        for ctrl in dlg.descendants():
+                            try:
+                                t = ctrl.window_text()
+                                if t and t.strip():
+                                    partes.append(t.strip())
+                            except Exception:
+                                pass
+                        texto_aviso = " ".join(
+                            dict.fromkeys(partes)
+                        )  # remove duplicados mantendo ordem
+                        console.print(f"Texto do Aviso: {texto_aviso}")
+                    except Exception as e:
+                        console.print(f"Falha ao coletar texto do 'Aviso': {e}")
+
+                    # === 4) Validação do conteúdo (edite/adicione frases-chave conforme seu caso) ===
+                    FRASES_ERRO = [
+                        "O período para apuração informado",
+                        "situação diferente de confirmado",
+                    ]
+
+                    contem_msg_erro = any(
+                        frase.lower() in texto_aviso.lower() for frase in FRASES_ERRO
+                    )
+
+                    if contem_msg_erro:
+                        console.print(
+                            "Confirmação de erro pelo texto da janela 'Aviso'. Retornando falha."
+                        )
+                        retorno_msg = "Erro ao Incluir apuração."
+                        if texto_aviso:
+                            retorno_msg += f" Detalhe: {texto_aviso}"
+                        if screenshot_path:
+                            retorno_msg += f" | Print: {screenshot_path}"
+
+                        return RpaRetornoProcessoDTO(
+                            sucesso=False,
+                            retorno=retorno_msg,
+                            status=RpaHistoricoStatusEnum.Falha,
+                            tags=[RpaTagDTO(descricao=RpaTagEnum.Negocio)],
+                        )
+                    else:
+                        console.print(
+                            "Janela 'Aviso' não contém mensagem de erro mapeada. Seguindo fluxo."
+                        )
+                        # (Opcional) você pode clicar no 'OK' e continuar:
+                        try:
+                            # tenta clicar em botão OK se existir
+                            for c in dlg.descendants():
+                                try:
+                                    if c.window_text().strip().lower() in ("ok", "&ok"):
+                                        c.click_input()
+                                        break
+                                except Exception:
+                                    pass
+                        except Exception:
+                            pass
+                        # não retorna aqui; deixa seguir para sucesso
+
+                # === 5) Se não houve 'Aviso' com erro mapeado, considera sucesso ===
+                console.print("Nenhum erro confirmado. Fluxo OK.")
+                try:
+                    main_window.close()
+                    console.print("Janela principal fechada.")
+                except Exception as e:
+                    console.print(f"Falha ao fechar janela principal (ignorado): {e}")
 
-                main_window.close()
-                
-                console.print("Apuração incluida com sucesso")
-                retorno = "Apuração incluida com sucesso"
+                retorno = "Apuração incluída com sucesso"
                 return RpaRetornoProcessoDTO(
-                    sucesso=True, retorno=retorno, status=RpaHistoricoStatusEnum.Sucesso
+                    sucesso=True,
+                    retorno=retorno,
+                    status=RpaHistoricoStatusEnum.Sucesso,
                 )
-            else:
-                console.print("Imagem incluir apuração não encontrada na tela.")
 
     except Exception as erro:
 

{worker_automate_hub-0.5.799.dist-info → worker_automate_hub-0.5.801.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: worker-automate-hub
-Version: 0.5.799
+Version: 0.5.801
 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.799.dist-info → worker_automate_hub-0.5.801.dist-info}/RECORD

@@ -36,7 +36,7 @@ worker_automate_hub/models/dto/rpa_sap_dto.py,sha256=eovdvKCtVIhRct3PL98KjsdlngL
 worker_automate_hub/models/dto/rpa_sistema_dto.py,sha256=sLkmJei8x6sl-1-IXUKDmYQuKx0sotYQREPyhQqPmRg,161
 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/abertura_livros_fiscais.py,sha256=zDp4kxspMYmsq1OnSyZr6kYqJt6ajGRiucz1f62KHz4,23081
+worker_automate_hub/tasks/jobs/abertura_livros_fiscais.py,sha256=Ldh1IuxrdDizkTvQRj91IC-I3ubhJ7dDNaTaAso0q8Y,27800
 worker_automate_hub/tasks/jobs/coleta_dje_process.py,sha256=UkLWTC5Ub2qBb0yY-8IZ0DLLOVJOfNTq_z9krx_t25Q,29476
 worker_automate_hub/tasks/jobs/conexao_rdp.py,sha256=S6QC4xhuo0pB5FjaUJZNMm1LIgEjpjifReCTBDqxH-U,11719
 worker_automate_hub/tasks/jobs/cte_manual.py,sha256=JucHpRMjMiy-QEJ0wtFnytLpN53tKXgCDI05nGLGclU,603
@@ -103,7 +103,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 worker_automate_hub/utils/util.py,sha256=noQRUSAjRnoDb1c4iMZ5eoyrNp59a8T9K78MHhalASw,210255
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=F7jk95LpDwl5WfaQXahCA5yDdnySnWdctDqczHXwGqE,38195
 worker_automate_hub/worker.py,sha256=zEnYUrm5kY2cHbbee15QJkwkx4euD2SB2zRvUIbjS90,6850
-worker_automate_hub-0.5.799.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.799.dist-info/METADATA,sha256=WwW5bb159GaBzTtHvsu3mHUdjUFXI1_gagjyWcl4Xaw,3100
-worker_automate_hub-0.5.799.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-worker_automate_hub-0.5.799.dist-info/RECORD,,
+worker_automate_hub-0.5.801.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.801.dist-info/METADATA,sha256=eEB2-yI8bOkLhHqeeDvla_2kq0gPKjv7xaAsfUhbKDs,3100
+worker_automate_hub-0.5.801.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+worker_automate_hub-0.5.801.dist-info/RECORD,,