worker-automate-hub 0.5.806__tar.gz → 0.5.808__tar.gz

{worker_automate_hub-0.5.806 → worker_automate_hub-0.5.808}/PKG-INFO

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

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

@@ -67,17 +67,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")
@@ -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,49 @@ 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 = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\notas_rejeitadas.png"
+                    img_dialog = "assets\\abertura_livros\\notas_rejeitadas.png"
+                    if os.path.exists(img_dialog):
+                        caixa = pyautogui.locateOnScreen(img_dialog, confidence=0.86, 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 confirmação.")
+
+                            await worker_sleep(5)
+
+                            # img_dialog = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\gerar_rel_notas_rejeitadas.png"
+                            img_dialog = "assets\\abertura_livros\\gerar_rel_notas_rejeitadas.png"
+                            if os.path.exists(img_dialog):
+                                caixa = pyautogui.locateOnScreen(img_dialog, confidence=0.86, 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=0).click_input()
+                                    print("Clicou no botão de confirmaçã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 +419,9 @@ 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 +456,27 @@ 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 = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\confirmado_livros.png"
+            imagem = "assets\\abertura_livros\\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
@@ -577,86 +607,6 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
                 await worker_sleep(2)
 
-                # flag para controlar se é a primeira vez
-                primeira_confirmacao = True
-
-                while True:
-                    janela_aberta = False
-
-                    # 1. Verifica se a imagem apareceu e clica em 'Sim' na janela "Informação"
-                    try:
-                        if pyautogui.locateOnScreen(imagem, confidence=0.9):
-                            print("Imagem 'livros_incluidos' apareceu na tela.")
-                            try:
-                                app_info = Application().connect(title="Informação", timeout=5)
-                                info_window = app_info["Informação"]
-                                info_window.set_focus()
-                                console.print("Clicando em 'Sim' na janela Informação...")
-                                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}")
-
-                    # Verifica se a janela TMsgBox de aviso está aberta
-                    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("Clicou no botão 'TBitBtn'.")
-                    except ElementNotFoundError:
-                        pass
-                    except Exception as e:
-                        print(f"[Erro ao procurar/clicar na TMsgBox]: {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()
-
-                        idx = 1 if primeira_confirmacao else 0
-                        try:
-                            form.child_window(class_name="TButton", found_index=idx).click_input()
-                            print(f"Clicou no botão de confirmação (found_index={idx}).")
-                            primeira_confirmacao = False  # nas próximas vezes será 0
-                        except Exception as e:
-                            print(f"[Falha ao clicar no botão found_index={idx}]: {e}")
-
-                    except ElementNotFoundError:
-                        pass
-                    except Exception as e:
-                        print(f"[Erro ao procurar/clicar na TMessageForm]: {e}")
-
-                    await worker_sleep(5)
-
-                    # 3. Verifica se a janela do relatório está aberta
-                    try:
-                        app_report = Application().connect(class_name="TFrmPreviewRelatorio", timeout=2)
-                        janela = app_report["TFrmPreviewRelatorio"]
-                        print("Janela 'TFrmPreviewRelatorio' encontrada.")
-                        janela_aberta = True
-                    except ElementNotFoundError:
-                        pass
-                    except Exception as e:
-                        print(f"[Erro ao procurar TFrmPreviewRelatorio]: {e}")
-
-                    # Se encontrou a janela de relatório, sai
-                    if janela_aberta:
-                        print("Relatório carregado. Saindo do loop.")
-                        break
-
-                    # Verifica tempo limite
-                    if time.time() - inicio > tempo_limite:
-                        print("Tempo esgotado. Relatório não carregado.")
-                        break
-
-                    print("Aguardando janela de relatório... (verificando novas confirmações se houver)")
-                    time.sleep(intervalo)
-
 
                 # === 5) Se não houve 'Aviso' com erro mapeado, considera sucesso ===
                 console.print("Nenhum erro confirmado. Fluxo OK.")
@@ -682,3 +632,4 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             status=RpaHistoricoStatusEnum.Falha,
             tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
         )
+

{worker_automate_hub-0.5.806 → worker_automate_hub-0.5.808}/worker_automate_hub/tasks/jobs/opex_capex.py

@@ -12,7 +12,7 @@ import win32clipboard
 from PIL import Image, ImageEnhance
 from pywinauto.application import Application
 from pywinauto.keyboard import send_keys
-from pywinauto.timings import wait_until
+from pywinauto.timings import TimeoutError as PywTimeout, wait_until
 from pywinauto_recorder.player import set_combobox
 from rich.console import Console
 import sys
@@ -471,6 +471,7 @@ async def opex_capex(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                 
         # Clicar em itens da nota
         imagem = "assets\\entrada_notas\\itens_nota.png"
+        # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\entrada_notas\itens_nota.png"
 
         # Tenta localizar a imagem na tela
         while True:
@@ -547,6 +548,11 @@ async def opex_capex(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                         send_keys("{ENTER}")
                     except Exception as e:
                         print(f"Erro ao selecionar opção: {e}")
+                    # Localize o combobox pelo class_name
+                    combo = main_window.child_window(class_name="TDBIComboBox", found_index=4)
+                    
+                    # Seleciona diretamente o texto
+                    combo.select("IPI 0%")
                             
                     # Clicar em alterar
                     main_window.child_window(class_name="TDBIBitBtn", found_index=3).click()
@@ -585,27 +591,99 @@ async def opex_capex(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
 
         panel_TabPagamento = panel_TTabSheet.child_window(title="Pagamento")
 
-        tipo_cobranca = panel_TTabSheet.child_window(
-            class_name="TDBIComboBox", found_index=0
-        )
-        tipo_cobranca.click()
-        # Lista de opções em ordem de prioridade
+        # Combo alvo (ajuste found_index se precisar)
+        tipo_cobranca = panel_TTabSheet.child_window(class_name="TDBIComboBox", found_index=0)
+
+        # Ordem de preferência
         opcoes = [
             "BANCO DO BRASIL BOLETO FIDC",
             "BANCO DO BRASIL BOLETO",
-            "BOLETO"
+            "BOLETO",
         ]
 
-        # Vamos percorrer o combo com setas
-        max_tentativas = 20  # evita loop infinito
-        for _ in range(max_tentativas):
-            texto_atual = tipo_cobranca.window_text().strip()
-            if any(opcao.lower() == texto_atual.lower() for opcao in opcoes):
-                send_keys("{ENTER}")
-                break
-            send_keys("{DOWN}")
-                
-        await worker_sleep(3)
+        # 1) Tenta .select() direto (não digita nada)
+        selecionado = None
+        for alvo in opcoes:
+            try:
+                tipo_cobranca.select(alvo)
+                if tipo_cobranca.window_text().strip().lower() == alvo.lower():
+                    selecionado = alvo
+                    break
+            except Exception:
+                pass
+
+        # 2) Abre a LISTA e seleciona o item exato (sem digitar)
+        if not selecionado:
+            tipo_cobranca.set_focus()
+            tipo_cobranca.click_input()
+            send_keys('%{DOWN}')  # ALT+DOWN para abrir o dropdown
+            # tenta achar a janela da lista (Delphi/Win32)
+            lista = None
+            app = tipo_cobranca.app
+            for crit in (dict(title="||List"), dict(class_name="ComboLBox"), dict(class_name_re=".*(List|Combo).*")):
+                try:
+                    cand = app.window(**crit)
+                    if cand.exists(timeout=0.5):
+                        lista = cand
+                        break
+                except Exception:
+                    pass
+
+            if lista:
+                # tenta selecionar por índice (texto exato)
+                try:
+                    itens = [t.strip() for t in lista.texts() if str(t).strip()]
+                except Exception:
+                    itens = []
+
+                idx_alvo = -1
+                alvo_escolhido = None
+                for alvo in opcoes:
+                    for i, t in enumerate(itens):
+                        if t.lower() == alvo.lower():
+                            idx_alvo = i
+                            alvo_escolhido = alvo
+                            break
+                    if idx_alvo >= 0:
+                        break
+
+                if idx_alvo >= 0:
+                    try:
+                        lista.select(idx_alvo)
+                    except Exception:
+                        # fallback por teclas sem digitar texto do item
+                        send_keys('{HOME}')
+                        for _ in range(idx_alvo):
+                            send_keys('{DOWN}')
+                    send_keys('{ENTER}')
+                    try:
+                        wait_until(2, 0.2, lambda: tipo_cobranca.window_text().strip() != "")
+                    except PywTimeout:
+                        pass
+                    if tipo_cobranca.window_text().strip().lower() == alvo_escolhido.lower():
+                        selecionado = alvo_escolhido
+                else:
+                    # fallback só com setas (sem digitar): vai ao topo e desce checando
+                    send_keys('{HOME}')
+                    vistos = set()
+                    for _ in range(60):
+                        atual = tipo_cobranca.window_text().strip()
+                        if atual.lower() in (o.lower() for o in opcoes):
+                            send_keys('{ENTER}')
+                            selecionado = atual
+                            break
+                        if atual.lower() in vistos:
+                            # deu a volta
+                            send_keys('{ESC}')
+                            break
+                        vistos.add(atual.lower())
+                        send_keys('{DOWN}')
+
+        # (opcional) validação dura
+        if not selecionado or selecionado.lower() not in (o.lower() for o in opcoes):
+            raise RuntimeError(f"Não consegui selecionar uma opção válida. Ficou: '{tipo_cobranca.window_text().strip()}'")
+
+        print("Selecionado:", selecionado)
         
         dt_vencimento_nota = nota.get("dataVencimento")  # vem como '2025-09-26'
         data_atual = datetime.now().date()
@@ -634,10 +712,8 @@ async def opex_capex(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                     
         console.print(f"Incluindo registro...\n")
         try:
-            ASSETS_PATH = "assets"
-            inserir_registro = pyautogui.locateOnScreen(
-                "assets\\entrada_notas\\IncluirRegistro.png", confidence=0.8
-            )
+            inserir_registro = pyautogui.locateOnScreen("assets\entrada_notas\IncluirRegistro.png", confidence=0.8)
+            # inserir_registro = pyautogui.locateOnScreen(r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\entrada_notas\IncluirRegistro.png", confidence=0.8)
             pyautogui.click(inserir_registro)
         except Exception as e:
             console.print(
@@ -830,4 +906,3 @@ async def opex_capex(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
         # Deleta o xml
         await delete_xml(numero_nota)
 
-

{worker_automate_hub-0.5.806 → worker_automate_hub-0.5.808}/worker_automate_hub/utils/util.py

@@ -339,9 +339,27 @@ async def login_emsys_fiscal(
         message="32-bit application should be automated using 32-bit Python",
     )
     await worker_sleep(2)
-    filial_cod = task.configEntrada.get("empresa")
-    if filial_cod == None:
-        filial_cod = task.configEntrada.get("filialEmpresaOrigem")
+    filial_cod = (
+    task.configEntrada.get("empresa")
+    or task.configEntrada.get("filialEmpresaOrigem")
+    or task.configEntrada.get("descricaoFilial")
+    )
+
+    # Extrai apenas os dígitos iniciais da string
+    num = None
+    if filial_cod is not None:
+        s = str(filial_cod).strip()
+        m = re.match(r'^(\d+)', s)          # pega o número do INÍCIO
+        if not m:
+            m = re.search(r'\d+', s)        # fallback: primeiro número que aparecer
+        if m:
+            num = m.group(1)
+
+    if num is None:
+        raise ValueError(f"Não foi possível extrair número de: {filial_cod!r}")
+
+    filial_cod = num
+        
     console.print(f"Empresa a ser processada: {filial_cod}")
 
     try: