worker-automate-hub 0.5.749__py3-none-any.whl → 0.5.912__py3-none-any.whl

worker_automate_hub/tasks/jobs/abertura_livros_fiscais.py

@@ -8,9 +8,7 @@ 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__), '..', '..', '..')))
-
-
+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,
@@ -47,6 +45,8 @@ console = Console()
 
 emsys = EMSys()
 
+# ASSETS_PATH = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros"
+ASSETS_PATH  = r"assets\abertura_livros"
 
 @repeat(times=10, delay=5)
 async def wait_aguarde_window_closed(app, timeout=60):
@@ -69,17 +69,6 @@ async def wait_aguarde_window_closed(app, timeout=60):
     console.log("Timeout esperando a janela Aguarde...")
 
 
-def click_desconfirmar():
-    cords = (675, 748)
-    pyautogui.click(x=cords[0], y=cords[1])
-
-
-def ctrl_c():
-    pyautogui.press("tab", presses=12)  # verificar
-    pyautogui.hotkey("ctrl", "c")
-    return pyperclip.paste()
-
-
 async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
     try:
         config = await get_config_by_name("login_emsys_fiscal")
@@ -94,11 +83,11 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             message="32-bit application should be automated using 32-bit Python",
         )
 
-        await worker_sleep(4)
+        await worker_sleep(8)
 
         try:
             app = Application(backend="win32").connect(
-                class_name="TFrmLoginModulo", timeout=50
+                class_name="TFrmLoginModulo", timeout=120
             )
         except:
             return RpaRetornoProcessoDTO(
@@ -149,7 +138,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             console.print("Aguardando janela 'Movimento de Livro Fiscal' aparecer...")
 
             # Tempo limite de espera (em segundos)
-            timeout = 60
+            timeout = 3600
             inicio = time.time()
 
             # Espera até a janela aparecer
@@ -205,7 +194,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             saida_checkbox.click_input()
 
             console.print("Aguardar marcar caixa de saida")
-            imagem = "assets\\abertura_livros\\saida_marcada.png"
+            imagem = fr"{ASSETS_PATH}\saida_marcada.png"
 
             tempo_limite = 600  # 10 minutos
             intervalo = 2  # segundos entre verificações
@@ -242,7 +231,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,9 +330,9 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             await worker_sleep(5)
 
             console.print("Aguardar o término de carregar")
-            imagem = "assets\\abertura_livros\\livros_incluidos.png"
+            imagem = fr"{ASSETS_PATH}\livros_incluidos.png"
 
-            tempo_limite = 1200  # 20 minutos
+            tempo_limite = 6600  # 1h
             intervalo = 2  # segundos entre as verificações
 
             inicio = time.time()
@@ -365,10 +354,10 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
                             info_window.child_window(
                                 class_name="Button", found_index=0
                             ).click_input()
-                        except Exception as e:
-                            print(f"Erro ao clicar em 'Sim' na janela Informação: {e}")
-                except Exception as e:
-                    print(f"[Erro ao procurar imagem]: {e}")
+                        except:
+                            pass
+                except:
+                    pass
 
                 # Verifica se a janela TMsgBox de aviso está aberta
                 try:
@@ -378,25 +367,65 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
                     box.set_focus()
                     box.child_window(class_name="TBitBtn", found_index=0).click_input()
                     print("Clicou no botão 'TBitBtn'.")
-                except ElementNotFoundError:
+                except:
                     pass
-                except Exception as e:
-                    print(f"[Erro ao procurar/clicar na TMessageForm]: {e}")
 
                 # 2. Verifica e trata janela de confirmação TMessageForm
                 try:
-                    app_msg = Application().connect(
-                        class_name="TMessageForm", timeout=2
-                    )
-                    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("Clicou no botão de confirmação.")
-                except ElementNotFoundError:
+                    img_dialog = fr"{ASSETS_PATH}\notas_rejeitadas.png"
+                    if os.path.exists(img_dialog):
+                        caixa = pyautogui.locateOnScreen(
+                            img_dialog, confidence=0.9, grayscale=True
+                        )
+                        if caixa:
+                            print("Janela 'notas rejeitadas' detectada por imagem.")
+                            region = (caixa.left, caixa.top, caixa.width, caixa.height)
+
+                            app_msg = Application().connect(
+                                class_name="TMessageForm", timeout=2
+                            )
+                            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("Clicou no botão de sim.")
+
+                            await worker_sleep(5)
+
+                            img_dialog = fr"{ASSETS_PATH}\gerar_rel_notas_rejeitadas.png"
+                            if os.path.exists(img_dialog):
+                                caixa = pyautogui.locateOnScreen(
+                                    img_dialog, confidence=0.9, grayscale=True
+                                )
+                                if caixa:
+                                    print(
+                                        "Janela 'notas relatorios rejeitadas' detectada por imagem."
+                                    )
+                                    region = (
+                                        caixa.left,
+                                        caixa.top,
+                                        caixa.width,
+                                        caixa.height,
+                                    )
+
+                                    app_msg = Application().connect(
+                                        class_name="TMessageForm", timeout=2
+                                    )
+                                    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("Clicou no botão de não.")
+                except:
                     pass
-                except Exception as e:
-                    print(f"[Erro ao procurar/clicar na TMessageForm]: {e}")
 
                 # 3. Verifica se a janela do relatório está aberta
                 try:
@@ -406,10 +435,8 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
                     janela = app_report["TFrmPreviewRelatorio"]
                     print("Janela 'TFrmPreviewRelatorio' encontrada.")
                     janela_aberta = True
-                except ElementNotFoundError:
+                except:
                     pass
-                except Exception as e:
-                    print(f"[Erro ao procurar TFrmPreviewRelatorio]: {e}")
 
                 # Se encontrou a janela de relatório, sai
                 if janela_aberta:
@@ -444,9 +471,26 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             main_window = app["TFrmPreviewRelatorio"]
             main_window.set_focus()
 
+            # Verificando se a foi confirmado os livres
+            console.print("Verificando se os livros foram confirmados")
+
+            imagem = fr"{ASSETS_PATH}\confirmado_livros.png"
+            if os.path.exists(imagem):
+                caixa = pyautogui.locateOnScreen(imagem, confidence=0.9, grayscale=True)
+            else:
+                console.print("Imagem confirmada não encontrada")
+                return RpaRetornoProcessoDTO(
+                    sucesso=False,
+                    retorno=f"Erro na Abertura de Livro Fiscal: Imagem confirmada não encontrada",
+                    status=RpaHistoricoStatusEnum.Falha,
+                    tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+                )
+
             # Clicar em fechar
             main_window.close()
 
+            console.print("Livros confirmados")
+
             await worker_sleep(5)
 
             # Conecta na janela principal
@@ -502,181 +546,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 +583,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
@@ -725,7 +596,9 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
                 class_name="TDBIEditDate", found_index=0
             )
             data_input.click_input()
-            await worker_sleep(1)
+
+            await worker_sleep(5)
+
             data_input.set_edit_text("")  # Limpa o campo
 
             # Define a competência
@@ -734,27 +607,36 @@ 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 = "assets\\abertura_livros\\btn_incluir_apuracao.png"
+            imagem = fr"{ASSETS_PATH}\btn_incluir_apuracao.png"
 
             # Tenta localizar a imagem na tela
-            localizacao = pyautogui.locateCenterOnScreen(
-                imagem, confidence=0.9
-            )  # você pode ajustar o confidence
+            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("Apuração incluida com sucesso")
-                retorno = "Apuração incluida com sucesso"
+                console.print("Botão incluir clicado com sucesso")
+
+                await worker_sleep(2)
+
+                # === 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 +644,4 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             status=RpaHistoricoStatusEnum.Falha,
             tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
         )
+