worker-automate-hub 0.5.762__tar.gz → 0.5.763__tar.gz

{worker_automate_hub-0.5.762 → worker_automate_hub-0.5.763}/PKG-INFO

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

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

@@ -1,10 +1,10 @@
 import asyncio
 import os
 from datetime import datetime
-from pywinauto import Application, timings, findwindows, keyboard
+from pywinauto import Application, timings, findwindows, keyboard, Desktop
 import sys
 import io
-
+import win32gui
 # sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..')))
 
 from worker_automate_hub.models.dto.rpa_historico_request_dto import (
@@ -282,14 +282,18 @@ async def extracao_saldo_estoque_fiscal(
 
         await worker_sleep(2)
 
+
         max_tentativas = 5
         tentativa = 1
         sucesso = False
 
+        # defina caminho_arquivo ANTES para não ficar indefinido
+        caminho_arquivo = rf"C:\Users\automatehub\Downloads\saldo_estoque_fiscal_{periodo_format}_{filial}.xlsx"
+
         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)
+            # 1) Abrir o picker 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):
@@ -304,39 +308,92 @@ async def extracao_saldo_estoque_fiscal(
 
             await worker_sleep(8)
 
-            # 2) Selecionar formato "Excel" na janela TFrmRelatorioFormato
+            # 2) Selecionar formato Excel (desambiguando múltiplas 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)
+                desktop = Desktop(backend="win32")
+
+                # Liste todas as visíveis
+                wins_visiveis = desktop.windows(class_name="TFrmRelatorioFormato", visible_only=True)
+                if not wins_visiveis:
+                    raise RuntimeError("Janela de formato não apareceu.")
+
+                # 2.1) Tente a janela em foco (foreground)
+                h_fore = win32gui.GetForegroundWindow()
+                alvo = None
+                for w in wins_visiveis:
+                    if w.handle == h_fore:
+                        alvo = w
+                        break
+
+                # 2.2) Se não estiver em foco, pegue a que contém um TComboBox (a 'Configuração para Salvar arq...')
+                if alvo is None:
+                    candidatos = []
+                    for w in wins_visiveis:
+                        try:
+                            if w.child_window(class_name="TComboBox").exists(timeout=0.8):
+                                candidatos.append(w)
+                        except Exception:
+                            pass
+                    if candidatos:
+                        alvo = candidatos[-1]     # a mais recente
+                    else:
+                        alvo = wins_visiveis[-1]  # fallback
+
+                # Trabalhe via WindowSpecification
+                spec_fmt = desktop.window(handle=alvo.handle)
+                spec_fmt.wait("visible", timeout=10)
+                win_fmt = spec_fmt.wrapper_object()
+                win_fmt.set_focus()
+
+                # Acessar o ComboBox
+                try:
+                    combo_spec = spec_fmt.child_window(class_name="TComboBox")
+                except Exception:
+                    combo_spec = spec_fmt.child_window(control_type="ComboBox")
+                combo_spec.wait("exists enabled", timeout=10)
+                combo = combo_spec.wrapper_object()
 
-                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'
+                # Seleção por índice conhecido; fallback por texto
                 try:
                     combo.select(8)
                 except Exception:
-                    alvo = None
+                    alvo_idx = None
                     for i, t in enumerate(textos):
                         if "EXCEL" in str(t).upper() or "XLSX" in str(t).upper():
-                            alvo = i
+                            alvo_idx = i
                             break
-                    if alvo is not None:
-                        combo.select(alvo)
-                    else:
+                    if alvo_idx is None:
                         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
+                    combo.select(alvo_idx)
 
                 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()
+                # Clique em OK
+                btn_ok_spec = spec_fmt.child_window(class_name="TBitBtn", found_index=1)
+                btn_ok_spec.wait("enabled", timeout=5)
+                btn_ok_spec.click_input()
+
+                # Aguarde a janela de formato desaparecer
+                try:
+                    spec_fmt.wait_not("visible", timeout=10)
+                except Exception:
+                    pass
+
+                # Feche possíveis duplicatas remanescentes (defensivo)
+                for w in desktop.windows(class_name="TFrmRelatorioFormato", visible_only=True):
+                    if w.handle != alvo.handle:
+                        try:
+                            w.close()
+                        except Exception:
+                            pass
+
             except Exception as e:
                 console.print(f"Falha ao selecionar formato: {e}", style="bold red")
                 tentativa += 1
@@ -348,19 +405,17 @@ async def extracao_saldo_estoque_fiscal(
             # 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)
+                app_save = Application(backend="win32").connect(title_re="Salvar para arquivo|Salvar como|Save As", timeout=30)
+                spec_save = app_save.window(title_re="Salvar para arquivo|Salvar como|Save As")
+                spec_save.wait("visible", timeout=30)
+                win_save = spec_save.wrapper_object()
             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
+            # 3.1) Remover arquivo pré-existente
             if os.path.exists(caminho_arquivo):
                 try:
                     os.remove(caminho_arquivo)
@@ -368,27 +423,32 @@ async def extracao_saldo_estoque_fiscal(
                 except Exception as e:
                     console.print(f"Não foi possível remover o arquivo existente: {e}", style="bold red")
 
+            # 3.2) Preencher nome e salvar
             try:
-                # Campo "Nome" (Edit, control_id=1148)
-                campo_nome = win_save.child_window(class_name="Edit", control_id=1148).wrapper_object()
+                campo_spec = spec_save.child_window(class_name="Edit", control_id=1148)
+                campo_spec.wait("exists enabled visible", timeout=10)
+                campo_nome = campo_spec.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()
+                btn_salvar_spec = spec_save.child_window(class_name="Button", found_index=0)
+                btn_salvar_spec.wait("enabled", timeout=10)
+                btn_salvar_spec.click_input()
+
+                # Esperar a janela sumir
+                try:
+                    spec_save.wait_not("visible", timeout=15)
+                except Exception:
+                    pass
+
             except Exception as e:
                 console.print(f"Erro ao confirmar salvar: {e}", style="bold red")
                 tentativa += 1
@@ -397,32 +457,27 @@ async def extracao_saldo_estoque_fiscal(
 
             await worker_sleep(2)
 
-            # 3.1) Tratar confirmação de sobrescrita, se aparecer
+            # 3.3) Confirmar sobrescrita (se houver)
             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
+                app_conf = Application(backend="win32").connect(
+                    title_re="Confirm(ar)?( )?Salvar( )?Como|Confirm Save As", timeout=3
+                )
+                spec_conf = app_conf.window(title_re="Confirm(ar)?( )?Salvar( )?Como|Confirm Save As")
+                spec_conf.wait("visible", timeout=3)
+                spec_conf.child_window(class_name="Button", found_index=0).click_input()
+                console.print("Confirmação de sobrescrita respondida.", style="bold yellow")
             except Exception:
                 pass
 
             await worker_sleep(2)
 
-            # 4) Aguardar o processamento/Printing encerrar
+            # 4) Aguardar 'Printing' (se existir)
             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")
+                app_print = Application(backend="win32").connect(title_re="Printing", timeout=5)
+                spec_print = app_print.window(title_re="Printing")
                 try:
-                    win_print.wait_not("visible", timeout=60)
+                    spec_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")
@@ -445,7 +500,6 @@ async def extracao_saldo_estoque_fiscal(
         if not sucesso:
             console.print("Falha após 5 tentativas. Arquivo não foi gerado.", style="bold red")
 
-
         nome_com_extensao = f"saldo_estoque_fiscal_{periodo_format}_{filial}.xlsx"
         # lê o arquivo
         print(caminho_arquivo)