worker-automate-hub 0.5.798__tar.gz → 0.5.800__tar.gz

{worker_automate_hub-0.5.798 → worker_automate_hub-0.5.800}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: worker-automate-hub
-Version: 0.5.798
+Version: 0.5.800
 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.798 → worker_automate_hub-0.5.800}/pyproject.toml

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

@@ -7,10 +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 (
     RpaHistoricoStatusEnum,
@@ -87,7 +84,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
         # Fecha a instancia do emsys - caso esteja aberta
         await kill_all_emsys()
 
-        app = Application(backend="win32").start("C:\\Rezende\\EMSys3\\EMSysFiscal.exe")
+        app = Application(backend="win32").start("C:\\Rezende\\EMSys3\\EMSysFiscal_39.exe")
         warnings.filterwarnings(
             "ignore",
             category=UserWarning,
@@ -205,6 +202,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
@@ -242,7 +240,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             # Clicando em incluir livro
             try:
                 console.print("Clicando em incluir livro")
-                cords = (695, 729)
+                cords = (698, 728)
                 pyautogui.click(x=cords[0], y=cords[1])
                 await worker_sleep(5)
             except:
@@ -341,6 +339,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
@@ -502,181 +501,8 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
             main_window.close()
 
-            await worker_sleep(5)
-
-            console.print("Selecionar Livro Saída aberto")
-
-            # Selecionar linha livro de saída aberto
-            # imagem = "assets\\abertura_livros\\livro_saida_aberto.png"
-            imagem = "assets\\abertura_livros\\livro_saida_aberto.png"
-
-            # Tenta localizar a imagem na tela
-            localizacao = pyautogui.locateCenterOnScreen(imagem, confidence=0.9)
-
-            if localizacao:
-                print(f"Imagem livro de saída aberto encontrado em: {localizacao}")
-                pyautogui.moveTo(localizacao)
-                pyautogui.click()
-            else:
-                console.print("Imagem livro de saída aberto não encontrado na tela.")
-
-            # Clicar em alterar livro
-            # imagem = "assets\\abertura_livros\\alterar_livro.png"
-            imagem = "assets\\abertura_livros\\alterar_livro.png"
-            # Tenta localizar a imagem na tela
-            localizacao = pyautogui.locateCenterOnScreen(
-                imagem, confidence=0.9
-            )  # você pode ajustar o confidence
-
-            if localizacao:
-                print(f"Imagem alterar livro encontrado em: {localizacao}")
-                pyautogui.moveTo(localizacao)
-                pyautogui.click()
-            else:
-                console.print("Imagem alterar livro não encontrada na tela.")
-
-            await worker_sleep(8)
-
-            # Clicar em Livro fiscal
-            # imagem = "assets\\abertura_livros\\livro_fiscal.png"
-            imagem = "assets\\abertura_livros\\livro_fiscal.png"
-            # Tenta localizar a imagem na tela
-            localizacao = pyautogui.locateCenterOnScreen(
-                imagem, confidence=0.9
-            )  # você pode ajustar o confidence
-
-            if localizacao:
-                print(f"Imagem Livro fiscal encontrado em: {localizacao}")
-                pyautogui.moveTo(localizacao)
-                pyautogui.click()
-            else:
-                console.print("Imagem Livro fiscal não encontrada na tela.")
-
-            await worker_sleep(4)
-
-            # Clicar em Gerar Relatório
-            # imagem = "assets\\abertura_livros\\gerar_registros.png"
-            imagem = "assets\\abertura_livros\\gerar_registros.png"
-
-            # Tenta localizar a imagem na tela
-            localizacao = pyautogui.locateCenterOnScreen(
-                imagem, confidence=0.9
-            )  # você pode ajustar o confidence
-
-            if localizacao:
-                print(f"Imagem gerar relatório encontrado em: {localizacao}")
-                pyautogui.moveTo(localizacao)
-                pyautogui.click()
-            else:
-                console.print("Imagem gerar relatório não encontrada na tela.")
-
-            await worker_sleep(4)
-
-            ##### Janela Gerar Registro ####
-            console.print("Confirmar Registro")
-            app = Application().connect(class_name="TMsgBox", timeout=60)
-            main_window = app["TMsgBox"]
-            main_window.set_focus()
-
-            # Clicar em Sim
-            main_window.child_window(class_name="TBitBtn", found_index=1).click_input()
-
-            await worker_sleep(4)
-
-            console.print("Clicar em confirmar")
-            app = Application().connect(
-                class_name="TPerguntasLivrosFiscaisForm", timeout=60
-            )
-            main_window = app["TPerguntasLivrosFiscaisForm"]
-            main_window.set_focus()
-            main_window.child_window(class_name="TButton", found_index=1).click_input()
-
-            await worker_sleep(60)
-
-            # Clicar no aviso
-            try:
-                app_msgbox = Application().connect(class_name="TMsgBox", timeout=10)
-                box = app_msgbox["TMsgBox"]
-                print("Janela 'TMsgBox' encontrada.")
-                box.set_focus()
-                box.child_window(class_name="TBitBtn", found_index=0).click_input()
-                print("1Clicou no botão 'TBitBtn'.")
-                await worker_sleep(5)
-            except:
-                pass
-            await worker_sleep(5)
-            try:
-                app_msg = Application().connect(class_name="TMessageForm", timeout=10)
-                form = app_msg["TMessageForm"]
-                console.print("Janela de confirmação 'TMessageForm' encontrada.")
-                form.set_focus()
-                form.child_window(class_name="TButton", found_index=1).click_input()
-                print("2Clicou no botão de confirmação.")
-                await worker_sleep(5)
-            except:
-                pass
-            await worker_sleep(5)
-            try:
-                app_msg = Application().connect(class_name="TMessageForm", timeout=10)
-                form = app_msg["TMessageForm"]
-                console.print("Janela de confirmação 'TMessageForm' encontrada.")
-                form.set_focus()
-                form.child_window(class_name="TButton", found_index=0).click_input()
-                print("3Clicou no botão de confirmação.")
-                await worker_sleep(5)
-            except:
-                pass
-            await worker_sleep(5)
-            try:
-                app_msg = Application().connect(class_name="TMessageForm", timeout=10)
-                form = app_msg["TMessageForm"]
-                console.print("Janela de confirmação 'TMessageForm' encontrada.")
-                form.set_focus()
-                form.child_window(class_name="TButton", found_index=1).click_input()
-                print("4Clicou no botão de confirmação.")
-                await worker_sleep(5)
-            except:
-                pass
-            await worker_sleep(5)
-            try:
-                app_msg = Application().connect(class_name="TMessageForm", timeout=10)
-                form = app_msg["TMessageForm"]
-                console.print("Janela de confirmação 'TMessageForm' encontrada.")
-                form.set_focus()
-                form.child_window(class_name="TButton", found_index=0).click_input()
-                print("5Clicou no botão de confirmação.")
-                await worker_sleep(5)
-            except:
-                pass
-            await worker_sleep(5)
-            try:
-                # Caminho da imagem que deve desaparecer
-                console.print("Aguardar carregar dados")
-                # imagem = "assets\\abertura_livros\\janela_carregada.png"
-                imagem = "assets\\abertura_livros\\janela_carregada.png"
-
-                # Tempo máximo de espera (em segundos)
-                tempo_limite = 600  # 10 minutos
-                intervalo = 2  # segundos entre as verificações
-
-                inicio = time.time()
-
-                while True:
-                    localizacao = pyautogui.locateOnScreen(imagem, confidence=0.9)
-
-                    if localizacao:
-                        print("Imagem ainda presente... aguardando")
-                    else:
-                        print("Imagem desapareceu da tela.")
-                        break  # A imagem sumiu, podemos seguir
-
-                    if time.time() - inicio > tempo_limite:
-                        print("Tempo esgotado. A imagem não desapareceu.")
-                        break
+            await worker_sleep(3)
 
-                    time.sleep(intervalo)
-            except:
-                pass
             # Conecta na janela principal
             console.print("Janela Livro de Apuração ICMS 2ª Etapa")
             app = Application().connect(class_name="TFrmPrincipalFiscal", timeout=60)
@@ -712,7 +538,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             await worker_sleep(2)
             send_keys("{ENTER}")
 
-            await worker_sleep(5)
+            # await worker_sleep(5)
             console.print("Inserindo competência...")
 
             # Conecta na janela
@@ -734,27 +560,126 @@ 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
-            )  # você pode ajustar o confidence
-
+            )  
+  
             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("Apuração incluida com sucesso")
-                retorno = "Apuração incluida com sucesso"
+                console.print("Botão incluir clicado com sucesso")
+
+                await worker_sleep(2)
+
+                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...")
+
+                    # === 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}")
+
+                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:
+
         console.print(f"Erro ao executar abertura de livros fiscais, erro : {erro}")
         return RpaRetornoProcessoDTO(
             sucesso=False,
@@ -762,3 +687,4 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             status=RpaHistoricoStatusEnum.Falha,
             tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
         )
+