worker-automate-hub 0.5.759__py3-none-any.whl → 0.5.761__py3-none-any.whl

worker_automate_hub/tasks/jobs/extracao_saldo_estoque_fiscal.py

@@ -282,86 +282,173 @@ async def extracao_saldo_estoque_fiscal(
 
         await worker_sleep(2)
 
-        # Clicar em salvar
-        caminho = r"assets\\extracao_relatorios\\btn_salvar.png"
-        # Verifica se o arquivo existe
-        if os.path.isfile(caminho):
-            print("A imagem existe:", caminho)
-
-            # Procura a imagem na tela
-            pos = pyautogui.locateCenterOnScreen(
-                caminho, confidence=0.9
-            )  # ajuste o confidence se necessário
-            if pos:
-                pyautogui.click(pos)  # clica no centro da imagem
-                print("Clique realizado na imagem.")
+        max_tentativas = 5
+        tentativa = 1
+        sucesso = False
+
+        while tentativa <= max_tentativas and not sucesso:
+            console.print(f"Tentativa {tentativa} de {max_tentativas}", style="bold cyan")
+
+            # 1) Abrir o picker de formatos pelo botão (imagem)
+            console.print("Procurando botão de salvar (imagem)...", style="bold cyan")
+            caminho_img = r'assets\\extracao_relatorios\\btn_salvar.png'
+            if os.path.isfile(caminho_img):
+                pos = pyautogui.locateCenterOnScreen(caminho_img, confidence=0.9)
+                if pos:
+                    pyautogui.click(pos)
+                    console.print("Clique realizado no botão salvar", style="bold green")
+                else:
+                    console.print("Imagem encontrada mas não está visível na tela", style="bold yellow")
             else:
-                print("Imagem encontrada no disco, mas não está visível na tela.")
-        else:
-            print("A imagem NÃO existe:", caminho)
-
-        await worker_sleep(2)
+                console.print("Imagem do botão salvar NÃO existe", style="bold red")
 
-        # Conecta na janela Configuração para Salvar Arquivo
-        app = Application().connect(class_name="TFrmRelatorioFormato", found_index=0)
-        main_window = app["TFrmRelatorioFormato"]
-        main_window.set_focus()
-        # Acessa o ComboBox pelo identificador conhecido
-        combo = main_window.ComboBox
-
-        # Garante que existe "Excel" na lista
-        itens = combo.texts()
-        print("Itens do ComboBox:", itens)
-
-        # Seleciona o Excel correto (o segundo da lista, índice 8)
-        combo.select(8)
-
-        await worker_sleep(2)
+            await worker_sleep(8)
 
-        # Clicar em Salvar
-        btn_salvar = main_window.child_window(
-            class_name="TBitBtn", found_index=1
-        ).click_input()
-
-        await worker_sleep(5)
-
-        # Conecta na janela "Salvar para arquivo"
-        app = Application().connect(title_re="Salvar para arquivo", timeout=30)
-        main_window = app.window(title_re="Salvar para arquivo")
+            # 2) Selecionar formato "Excel" na janela TFrmRelatorioFormato
+            console.print("Selecionando formato Excel...", style="bold cyan")
+            try:
+                app_fmt = Application().connect(class_name="TFrmRelatorioFormato", timeout=10)
+                win_fmt = app_fmt["TFrmRelatorioFormato"]
+                win_fmt.wait("visible", timeout=10)
+
+                combo = win_fmt.ComboBox
+                textos = combo.texts()
+                console.print(f"Itens do ComboBox: {textos}", style="bold yellow")
+
+                # Se souber o índice correto, mantenha. Caso contrário, tente por texto contendo 'Excel'
+                try:
+                    combo.select(8)
+                except Exception:
+                    alvo = None
+                    for i, t in enumerate(textos):
+                        if "EXCEL" in str(t).upper() or "XLSX" in str(t).upper():
+                            alvo = i
+                            break
+                    if alvo is not None:
+                        combo.select(alvo)
+                    else:
+                        console.print("Não foi possível localizar a opção de Excel no ComboBox.", style="bold red")
+                        tentativa += 1
+                        await worker_sleep(2)
+                        continue
+
+                await worker_sleep(1)
+
+                # Botão OK/Confirmar na janela de formato
+                win_fmt.child_window(class_name="TBitBtn", found_index=1).wait("enabled", timeout=5)
+                win_fmt.child_window(class_name="TBitBtn", found_index=1).click_input()
+            except Exception as e:
+                console.print(f"Falha ao selecionar formato: {e}", style="bold red")
+                tentativa += 1
+                await worker_sleep(3)
+                continue
 
-        # Campo Nome (Edit) - use set_edit_text para evitar problemas de escape
-        campo_nome = main_window.child_window(
-            class_name="Edit", control_id=1148
-        ).wrapper_object()
-        caminho_arquivo = rf"C:\Users\automatehub\Downloads\saldo_estoque_fiscal_{periodo_format}_{filial}.xlsx"
-        campo_nome.set_focus()
-        campo_nome.set_edit_text(caminho_arquivo)
+            await worker_sleep(5)
 
-        print("✅ Texto inserido no campo Nome")
+            # 3) Janela "Salvar para arquivo"
+            console.print("Abrindo janela de salvar arquivo...", style="bold cyan")
+            try:
+                app_save = Application().connect(title_re="Salvar para arquivo", timeout=30)
+                win_save = app_save.window(title_re="Salvar para arquivo")
+                win_save.wait("visible", timeout=30)
+            except Exception as e:
+                console.print(f"Não achou a janela 'Salvar para arquivo': {e}", style="bold red")
+                tentativa += 1
+                await worker_sleep(3)
+                continue
+
+            # Caminho do arquivo a salvar
+            caminho_arquivo = rf"C:\Users\automatehub\Downloads\saldo_estoque_fiscal_{periodo_format}_{filial}.xlsx"
+
+            # Se já existe, removemos para evitar pop-up de confirmação
+            if os.path.exists(caminho_arquivo):
+                try:
+                    os.remove(caminho_arquivo)
+                    console.print("Arquivo existente removido para evitar prompt de sobrescrita.", style="bold yellow")
+                except Exception as e:
+                    console.print(f"Não foi possível remover o arquivo existente: {e}", style="bold red")
 
-        await worker_sleep(3)
+            try:
+                # Campo "Nome" (Edit, control_id=1148)
+                campo_nome = win_save.child_window(class_name="Edit", control_id=1148).wrapper_object()
+                campo_nome.set_focus()
+                # limpa conteúdo
+                try:
+                    campo_nome.set_edit_text("")
+                except Exception:
+                    # fallback limpando com Ctrl+A + Delete
+                    campo_nome.type_keys("^a{DELETE}", pause=0.02)
+
+                # digita caminho
+                campo_nome.type_keys(caminho_arquivo, with_spaces=True, pause=0.01)
+                console.print(f"Arquivo configurado para: {caminho_arquivo}", style="bold green")
+
+                await worker_sleep(1)
+
+                # Botão Salvar (primeiro Button)
+                btn_salvar = win_save.child_window(class_name="Button", found_index=0)
+                btn_salvar.wait("enabled", timeout=10)
+                btn_salvar.click_input()
+            except Exception as e:
+                console.print(f"Erro ao confirmar salvar: {e}", style="bold red")
+                tentativa += 1
+                await worker_sleep(3)
+                continue
 
-        # Clicar em ok para salvar
-        keyboard.send_keys("{TAB}{TAB}{ENTER}", pause=0.3)
+            await worker_sleep(2)
 
-        await worker_sleep(5)
+            # 3.1) Tratar confirmação de sobrescrita, se aparecer
+            try:
+                # Pode vir em PT/EN dependendo do SO
+                # Título comum: "Confirm Save As" (EN) ou "Confirmar Salvar Como" (PT)
+                try:
+                    app_conf = Application().connect(title_re="Confirm(ar)?( )?Salvar( )?Como|Confirm Save As", timeout=3)
+                    win_conf = app_conf.window(title_re="Confirm(ar)?( )?Salvar( )?Como|Confirm Save As")
+                    win_conf.wait("visible", timeout=3)
+                    # Botões costumam ser "Sim"/"Yes" como class_name="Button"
+                    # Tente o primeiro botão (Yes/Sim)
+                    win_conf.child_window(class_name="Button", found_index=0).click_input()
+                    console.print("Confirmação de sobrescrita respondida.", style="bold yellow")
+                except Exception:
+                    pass
+            except Exception:
+                pass
 
-        caminho_img = r"assets\\extracao_realtorios\\janela_printing.png"
+            await worker_sleep(2)
 
-        # Aguarda até a janela com título "Printing" (ou "Salvando...") fechar
-        try:
-            app = Application().connect(title_re="Printing")  # conecta se existir
-            janela = app.window(title_re="Printing")
+            # 4) Aguardar o processamento/Printing encerrar
+            console.print("Aguardando finalização do processo de impressão/salvamento...", style="bold cyan")
+            try:
+                app_print = Application().connect(title_re="Printing", timeout=5)
+                win_print = app_print.window(title_re="Printing")
+                try:
+                    win_print.wait_not("visible", timeout=60)
+                    console.print("Janela 'Printing' fechada.", style="bold green")
+                except Exception:
+                    console.print("Janela 'Printing' não fechou no tempo esperado. Seguindo.", style="bold yellow")
+            except findwindows.ElementNotFoundError:
+                console.print("Janela 'Printing' não apareceu.", style="bold yellow")
+            except Exception as e:
+                console.print(f"Erro ao aguardar 'Printing': {e}", style="bold yellow")
+
+            # 5) Validar arquivo salvo
+            if os.path.exists(caminho_arquivo):
+                console.print(f"Arquivo encontrado: {caminho_arquivo}", style="bold green")
+                with open(caminho_arquivo, "rb") as f:
+                    file_bytes = io.BytesIO(f.read())
+                sucesso = True
+            else:
+                console.print("Arquivo não encontrado, tentando novamente...", style="bold red")
+                tentativa += 1
+                await worker_sleep(3)
 
-            print("Aguardando a janela 'Printing' sumir...")
-            janela.wait_not("visible", timeout=60)  # espera até 60 segundos
-            print("Janela 'Printing' fechada.")
+        if not sucesso:
+            console.print("Falha após 5 tentativas. Arquivo não foi gerado.", style="bold red")
 
-        except findwindows.ElementNotFoundError:
-            print("Janela 'Printing' não estava aberta.")
 
         nome_com_extensao = f"saldo_estoque_fiscal_{periodo_format}_{filial}.xlsx"
         # lê o arquivo
+        print(caminho_arquivo)
         with open(f"{caminho_arquivo}", "rb") as file:
             file_bytes = io.BytesIO(file.read())
 
@@ -378,7 +465,7 @@ async def extracao_saldo_estoque_fiscal(
             os.remove(f"{caminho_arquivo}")
             return RpaRetornoProcessoDTO(
                 sucesso=True,
-                retorno="Relatório enviado com sucesso",
+                retorno="Relatório gerado com sucesso",
                 status=RpaHistoricoStatusEnum.Sucesso,
             )
 
@@ -391,8 +478,6 @@ async def extracao_saldo_estoque_fiscal(
                 tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
             )
 
-        print("")
-
     except Exception as ex:
         retorno = f"Erro Processo Saldo Estoque Fiscal: {str(ex)}"
         logger.error(retorno)
@@ -403,3 +488,4 @@ async def extracao_saldo_estoque_fiscal(
             status=RpaHistoricoStatusEnum.Falha,
             tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
         )
+

{worker_automate_hub-0.5.759.dist-info → worker_automate_hub-0.5.761.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.759
+Version: 0.5.761
 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.759.dist-info → worker_automate_hub-0.5.761.dist-info}/RECORD

@@ -71,7 +71,7 @@ worker_automate_hub/tasks/jobs/exemplo_processo.py,sha256=nV0iLoip2FH2-FhLmhX3nP
 worker_automate_hub/tasks/jobs/extracao_fechamento_contabil.py,sha256=6Kr5DKjKLqtFvGzyiXtt7xrQsuU898l8pQXDq9C6AX8,19567
 worker_automate_hub/tasks/jobs/extracao_fechamento_emsys.py,sha256=-T2nZUDiFrUGm_KLxJd_4qcrageDxVpWW3KAAniLFC4,21448
 worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py,sha256=nFAv5Bvwwo2Mhy69q5BqR-XsKtg4KIqAunrUmGeW1S0,9173
-worker_automate_hub/tasks/jobs/extracao_saldo_estoque_fiscal.py,sha256=OJQAkE6mqm7lNodleYRWwJq3syxrTqcPGt27_u6NPPs,14950
+worker_automate_hub/tasks/jobs/extracao_saldo_estoque_fiscal.py,sha256=d4ckXWJ2sZmE8femwVsGpy6YRIqcleEUOyhLDpjoidM,20010
 worker_automate_hub/tasks/jobs/fechar_conexao_rdp.py,sha256=UWAKCS2dbfgDlSQOBdjmVJXfD1MMuUrOi3weDgB0CAc,5718
 worker_automate_hub/tasks/jobs/fidc_exportacao_docs_portal_b2b.py,sha256=tWUmYy3Zhi3JEt8AoqTsWpU-wbf5-OxhCrTOooh1WH4,15616
 worker_automate_hub/tasks/jobs/fidc_gerar_nosso_numero.py,sha256=FAmcCqKVjedf7wIped8XRLIZ9S3oWc6fakF-r1Zm0kg,12637
@@ -101,7 +101,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 worker_automate_hub/utils/util.py,sha256=V2WtWoETdTrAtGA8UgeqAAVphUj9KkGSZFzYsHJFATA,210055
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=TOXKSHOPxy8N3-ROpTGjNIHstX0i2b8qekcj1tRvjG8,38174
 worker_automate_hub/worker.py,sha256=uhZ3f-iaQ1i8cANbljp50vkYl-Xm0_sHtjwwF_2y72o,7191
-worker_automate_hub-0.5.759.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.759.dist-info/METADATA,sha256=tMWbnY_CoFvxTSMsKPGVRbkAsl_HM5XxtbKuLCk0Ajw,3049
-worker_automate_hub-0.5.759.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-worker_automate_hub-0.5.759.dist-info/RECORD,,
+worker_automate_hub-0.5.761.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.761.dist-info/METADATA,sha256=QkyzB7A1SY7-XUnJpdV0SyGi78QauZQ6N1-rjyb2T_U,3049
+worker_automate_hub-0.5.761.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+worker_automate_hub-0.5.761.dist-info/RECORD,,