worker-automate-hub 0.5.814__tar.gz → 0.5.816__tar.gz

{worker_automate_hub-0.5.814 → worker_automate_hub-0.5.816}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.814"
+version = "0.5.816"
 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.814 → worker_automate_hub-0.5.816}/worker_automate_hub/api/client.py

@@ -744,6 +744,8 @@ async def send_file(
         elif file_extension == "xlsx":
             filename = desArquivo
             content_type = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+        elif file_extension == "csv": 
+            content_type = "text/csv"
         else:
             raise ValueError(f"Extensão de arquivo não suportada: {file_extension}")
 

worker_automate_hub-0.5.816/worker_automate_hub/tasks/jobs/geracao_balancetes_filial.py

@@ -0,0 +1,295 @@
+import numpy
+import asyncio
+import sys
+import os
+import io
+from pywinauto.keyboard import send_keys
+# sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..')))
+from worker_automate_hub.models.dto.rpa_historico_request_dto import (
+    RpaHistoricoStatusEnum,
+    RpaRetornoProcessoDTO,
+    RpaTagDTO,
+    RpaTagEnum,
+)
+from worker_automate_hub.models.dto.rpa_processo_entrada_dto import (
+    RpaProcessoEntradaDTO,
+)
+from worker_automate_hub.utils.util import (
+    kill_all_emsys,
+    worker_sleep,
+)
+from rich.console import Console
+import pyautogui
+from datetime import datetime
+# from dateutil.relativedelta import relativedelta
+from PIL import ImageFilter, ImageEnhance
+from pytesseract import image_to_string
+from pywinauto import Application, Desktop
+import subprocess
+import os
+from worker_automate_hub.api.client import (
+    get_config_by_name, send_file
+)
+from worker_automate_hub.utils.utils_nfe_entrada import EMSys
+import psutil
+from time import sleep
+
+pyautogui.PAUSE = 0.5
+console = Console()
+emsys = EMSys()
+
+
+def get_text_from_window(window, relative_coords, value=None):
+    try:
+        screenshot = window.capture_as_image()
+        imagem = screenshot.convert("L")
+        imagem = imagem.filter(ImageFilter.SHARPEN)
+        imagem = ImageEnhance.Contrast(imagem).enhance(2)
+        cropped_screenshot = imagem.crop(relative_coords)
+        texto = image_to_string(cropped_screenshot, lang="por")
+        return (value.upper() in texto.upper()) if value != None else texto.lower()
+    except Exception as error:
+        console.print(f"Error: {error}")
+
+
+async def open_contabil_processes():
+    try:
+        console.print("Abrindo EMSys Contabil...")
+        os.startfile("C:\\Users\\automatehub\\Desktop\\Contabil\\contabil1.lnk")
+        await worker_sleep(3)
+        os.startfile("C:\\Users\\automatehub\\Desktop\\Contabil\\contabil2.lnk")
+        await worker_sleep(30)
+        os.startfile("C:\\Users\\automatehub\\Desktop\\Contabil\\contabil3.lnk")
+        await worker_sleep(20)
+        pyautogui.hotkey("win", "d")
+        await worker_sleep(4)
+        os.startfile("C:\\Users\\automatehub\\Desktop\\Contabil\\contabil4.lnk")
+        await worker_sleep(2)
+    except Exception as error:
+        console.print(f"Error: {error}")
+
+
+async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
+    try:
+        await kill_all_emsys()
+        await open_contabil_processes()
+        config = await get_config_by_name("login_emsys_contabil")
+        filial = task.configEntrada.get("filialEmpresaOrigem")
+        periodo_inicial = task.configEntrada.get("periodoInicial")
+        periodo_final = task.configEntrada.get("periodoFinal")
+        historico_id = task.historico_id
+        if historico_id:
+            console.print("Historico ID recuperado com sucesso...\n")
+        app = None
+        max_attempts = 30
+        console.print("Tentando encontrar janela de login...")
+        for attempt in range(max_attempts):
+            try:
+                app = Application(backend="win32").connect(
+                    title="Selecione o Usuário para autenticação"
+                )
+                console.print("Janela encontrada!")
+                break
+            except:
+                console.print("Janela ainda nao encontrada...")
+                await worker_sleep(1)
+        if not app:
+            console.print("Nao foi possivel encontrar a janela de login...")
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno="Erro durante tentativa localizacao de janelas...",
+                status=RpaHistoricoStatusEnum.Falha,
+                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+            )
+        await emsys.verify_warning_and_error("Erro", "&Ok")
+        await worker_sleep(4)
+        pyautogui.click(x=1021, y=127)
+        console.print("Logando...")
+        await emsys.verify_warning_and_error("Erro", "&Ok")
+        pyautogui.write(config.conConfiguracao.get("user"))
+        pyautogui.press("enter")
+
+        await worker_sleep(4)
+        pyautogui.write(config.conConfiguracao.get("pass"))
+        pyautogui.press("enter")
+        
+        await worker_sleep(10)
+
+        main_window = None
+        for attempt in range(max_attempts):
+            main_window = Application().connect(title="EMSys [Contabil]")
+            main_window = main_window.top_window()
+            if main_window.exists():
+                console.print("Janela encontrada!")
+                break
+            console.print("Janela ainda nao encontrada...")
+            await worker_sleep(1)
+
+        if not main_window:
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno="Erro durante tentativa localizacao de janelas....",
+                status=RpaHistoricoStatusEnum.Falha,
+                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+            )
+
+        # Adicionando foco
+        try:
+            main_window.set_focus()
+            console.print(f"Ativando janela: {main_window}")
+        except Exception as error:
+            console.print(f"Erro ao setar foco na janela: {main_window}")
+
+        await worker_sleep(4)
+        console.print("Cicar em BAL")
+        pyautogui.click(x=453, y=96)
+        await worker_sleep(4)
+
+        console.print("Preenchendo campo periodo...")
+
+        app = Application(backend="win32").connect(class_name="TFrmBalancete")
+        win = app.window(class_name="TFrmBalancete")
+
+        # Seus índices: inicial = found_index=1, final = found_index=0
+        ctrl_inicial = win.child_window(class_name="TRzEditDate", found_index=1).wrapper_object()
+        ctrl_final   = win.child_window(class_name="TRzEditDate", found_index=0).wrapper_object()
+
+        # ---- Inicial ----
+        try:
+            ctrl_inicial.set_edit_text(periodo_inicial)   # tenta via WM_SETTEXT
+        except Exception:
+            ctrl_inicial.set_focus()
+            ctrl_inicial.click_input()
+            send_keys('^a{DELETE}')
+            send_keys(periodo_inicial.replace('/', ''))   # fallback: digita só dígitos (máscara)
+
+        # ---- Final ----
+        try:
+            ctrl_final.set_edit_text(periodo_final)
+        except Exception:
+            ctrl_final.set_focus()
+            ctrl_final.click_input()
+            send_keys('^a{DELETE}')
+            send_keys(periodo_final.replace('/', ''))
+        
+        await worker_sleep(2)
+        
+        console.print("Selecionar detalhada por centro de custo..")
+        detalhada = win.child_window(
+            class_name="TRzComboBox", found_index=0
+        )
+        detalhada.select("Centro de Custo")
+        
+        await worker_sleep(2)
+
+        console.print("Selecionar considerar contas analíticas zerada")
+        contas_analit_zeradas = win.child_window(class_name="TRzCheckBox", found_index=2).click()
+
+        console.print("Selecionar considerar contas sintéticas zerada")
+        contas_sint_zeradas = win.child_window(class_name="TRzCheckBox", found_index=0).click()
+
+        console.print("Selecionar por filiais")
+        selec_filiais = win.child_window(class_name="TRzCheckBox", found_index=3).click()
+
+        console.print("Selecionar CSV")
+        selec_csv = win.child_window(class_name="TRzComboBox", found_index=1)
+        selec_csv.select("Arquivo CSV")
+
+        console.print("Clicar em gerar relatório")
+        btn_gerar_relatorio = win.child_window(class_name="TBitBtn", found_index=0).click()
+
+        # Selecionar filial
+        app = Application(backend="win32").connect(title="Seleção de Empresas", timeout=10)
+        dlg = app.window(title ="Seleção de Empresas")
+        edit = dlg.child_window(class_name="TEdit", found_index=0).wrapper_object()
+
+        # Tenta via WM_SETTEXT
+        try:
+            edit.set_focus()
+            edit.set_edit_text("")      # limpa
+            edit.set_edit_text("3")     # escreve
+        except Exception:
+            # Fallback: digita como teclado
+            edit.set_focus()
+            edit.click_input()
+            send_keys('^a{DELETE}')
+            send_keys("3", with_spaces=True)
+
+        await worker_sleep(3)
+       
+        # Marcar filial
+        imagem_alvo = "assets\\geracao_bal_filial\\btn_selec_uma_filial.png"
+
+        btn_sect_uma = pyautogui.locateCenterOnScreen(imagem_alvo, confidence=0.9)  # requer opencv-python
+        if btn_sect_uma:  # se achou, clica
+            pyautogui.click(btn_sect_uma)
+
+        btn_ok = dlg.child_window(class_name="TBitBtn", title="&OK").click()
+
+        # aguarda até a janela "Gera Arquivo CSV (Excel)" existir (ou ficar visível)
+        csv_win = Desktop(backend="win32").window(title="Gera Arquivo CSV (Excel)")
+        csv_win.wait('exists', timeout=3600)
+
+        app_csv = Application(backend="win32").connect(title="Gera Arquivo CSV (Excel)")
+        dlg_csv = app_csv.window(title="Gera Arquivo CSV (Excel)")
+        edit = dlg_csv.child_window(class_name="Edit", found_index=0)
+        # Tenta via WM_SETTEXT (mais estável)
+        try:
+            periodo_inicial = periodo_inicial.replace("/","")
+            periodo_final = periodo_final.replace("/","")
+            edit.set_focus()
+            edit.set_edit_text("")              # limpa
+            edit.set_edit_text(fr"C:\Users\automatehub\Downloads\balancete_{periodo_inicial}_{periodo_final}_{filial}")    
+        except Exception:
+            # Fallback: digita como teclado
+            edit.set_focus()
+            edit.click_input()
+            send_keys('^a{DELETE}')
+            send_keys(fr"C:\Users\automatehub\Downloads\balancete_{periodo_inicial}_{periodo_final}_{filial}", with_spaces=True)
+
+        # Clicar em salvar
+        app = Application(backend="win32").connect(title="Gera Arquivo CSV (Excel)", timeout=10)
+        dlg = app.window(title ="Gera Arquivo CSV (Excel)")
+        btn_salvar = dlg.child_window(class_name="Button", found_index=0).click()
+
+        await worker_sleep(3)
+
+        # Janela confirmação clicar em OK
+        app = Application(backend="win32").connect(title="Informação", timeout=10)
+        dlg = app.window(title ="Informação")
+        btn_ok = dlg.child_window(class_name="Button", found_index=0).click()
+
+        console.print("Arquivo salvo com sucesso...\n")
+        await worker_sleep(3)
+        path_to_txt = fr"C:\Users\automatehub\Downloads\balancete_{periodo_inicial}_{periodo_final}_{filial}"
+
+        with open(f"{path_to_txt}.csv", 'rb') as file:
+            file_bytes = io.BytesIO(file.read())
+
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        desArquivo = f"balancete_{periodo_inicial}_{periodo_final}_{filial}.csv"
+        try:
+            await send_file(historico_id, desArquivo, "csv", file_bytes, file_extension="csv")
+            os.remove(path_to_txt+".csv")
+            return RpaRetornoProcessoDTO(
+                sucesso=True,
+                retorno="Balancete gerado com sucesso",
+                status=RpaHistoricoStatusEnum.Sucesso
+            )
+        except Exception as e:
+            result = f"Arquivo balancete gerado com sucesso, porém gerou erro ao realizar o envio para o backoffice {e} - Arquivo ainda salvo na dispositivo utilizado no diretório {path_to_txt}!"
+            console.print(result, style="bold red")
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno=result,
+                status=RpaHistoricoStatusEnum.Falha,
+                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+            )
+        
+    except Exception as erro:
+        return RpaRetornoProcessoDTO(
+            sucesso=False,
+            retorno=f"Erro durante o processo integração contabil, erro : {erro}",
+            status=RpaHistoricoStatusEnum.Falha,
+            tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+        )

{worker_automate_hub-0.5.814 → worker_automate_hub-0.5.816}/worker_automate_hub/tasks/jobs/integracao_contabil_generica.py

@@ -202,7 +202,7 @@ async def integracao_contabil_generica(
 
         # console.print("Selecionando item do campo origem...")
         uuid_processo = task.uuidProcesso
-        caminho_imagem = os.path.join(ASSETS_PATH_BASE, "integracao_contabil", f"{uuid_processo}.png")
+        caminho_imagem = f"assets\\integracao_contabil\\{uuid_processo}.png"
 
         await metodo_selecao_origem_especial()
         await localizar_e_clicar(caminho_imagem)  # main_window.set_focus()
@@ -258,7 +258,7 @@ async def integracao_contabil_generica(
         console.print("Verificar se existem lotes")
         try:
             # Verifica mensagem sem lote pra integrar
-            imagem_alvo = os.path.join(ASSETS_PATH_BASE, "integracao_contabil", "sem_lote.png")
+            imagem_alvo = "assets\\integracao_contabil\\sem_lote.png"
 
             localizacao = pyautogui.locateOnScreen(imagem_alvo, confidence=0.9)
 
@@ -297,7 +297,7 @@ async def integracao_contabil_generica(
         await worker_sleep(5)
 
         try:
-            imagem_finalizada = os.path.join(ASSETS_PATH_BASE, "integracao_contabil", "pesquisa_finalizada.png")
+            imagem_finalizada = "assets\\integracao_contabil\\pesquisa_finalizada.png"
 
             if not os.path.exists(imagem_finalizada):
                 raise FileNotFoundError(
@@ -399,9 +399,9 @@ async def integracao_contabil_generica(
                 total_debito = numeros[1].window_text()
                 total_credito = numeros[2].window_text()
 
-                console.print("Diferença:", diferenca)
-                console.print("Total Débito:", total_debito)
-                console.print("Total Crédito:", total_credito)
+                print("Diferença:", diferenca)
+                print("Total Débito:", total_debito)
+                print("Total Crédito:", total_credito)
 
                 if diferenca > "0,00":
                     clicou = True
@@ -433,16 +433,19 @@ async def integracao_contabil_generica(
 
             await worker_sleep(5)
 
-            # --- AJUSTE: retorno de SUCESSO quando detectar integracao_sucesso.png
-            assets_int_cont = os.path.join(ASSETS_PATH_BASE, "integracao_contabil")
+            assets_int_cont = "assets\\integracao_contabil\\"
             err_dict = {
-                os.path.join(assets_int_cont, "erro_duplicidade.png"): "Integração não realizada. Erro de Duplicidade localizado enquanto finalizava a integração, contate o suporte do Emsys.",
-                os.path.join(assets_int_cont, "conta_indefinida_error.png"): "Integração não realizada. Conta contábil indefinida no sistema.",
-                os.path.join(assets_int_cont, "lote_sem_complemento_error.png"): "Integração não realizada. Lote encontrado sem complemento obrigatório.",
-                os.path.join(assets_int_cont, "diferenca_cred_deb.png"): "Integração não realizada. Existem diferença em lotes consistentes, por favor verificar.",
-                os.path.join(assets_int_cont, "integracao_sucesso.png"): "Integração Finalizada com Sucesso.",
+                assets_int_cont
+                + "erro_duplicidade.png": "Integração não realizada. Erro de Duplicidade localizado enquanto finalizava a integração, contate o suporte do Emsys.",
+                assets_int_cont
+                + "conta_indefinida_error.png": "Integração não realizada. Conta contábil indefinida no sistema.",
+                assets_int_cont
+                + "lote_sem_complemento_error.png": "Integração não realizada. Lote encontrado sem complemento obrigatório.",
+                assets_int_cont
+                + "diferenca_cred_deb.png": "Integração não realizada. Existem diferença em lotes consistentes, por favor verificar.",
+                assets_int_cont
+                + "integracao_sucesso.png": "Integração Finalizada com Sucesso.",
             }
-
             # Aguardar finalizar
             while True:
                 try:
@@ -451,56 +454,23 @@ async def integracao_contabil_generica(
                     )
                     msg_box = app["TMsgBox"]
 
-                    # Verifica por imagens de SUCESSO/ERRO
+                    # Antes de qualquer coisa, verifica por imagem de erro
                     for img_path, mensagem in err_dict.items():
                         try:
-                            found = pyautogui.locateOnScreen(img_path, confidence=0.90)
-                        except Exception:
-                            found = None
+                            err = pyautogui.locateOnScreen(img_path, confidence=0.90)
+                        except:
+                            continue
 
-                        if found:
-                            # Caso SUCESSO
-                            if img_path.endswith("integracao_sucesso.png"):
+                        if err:
+                            if "integracao_sucesso.png" in img_path:
                                 console.print(f"[green]{mensagem}[/green]")
-                                # fecha a TMsgBox se estiver aberta (best-effort)
-                                try:
-                                    msg_box.child_window(class_name="TBitBtn", found_index=0).click_input()
-                                except:
-                                    pass
-
-                                # Se integrou com 'Lotes Consistentes' marcados, é FALHA
-                                if lotesMarcados:
-                                    try:
-                                        # busca novamente o checkbox para garantir escopo
-                                        cb = dlg.child_window(title="Lotes Consistentes", class_name="TCheckBox")
-                                        if cb.exists() and cb.is_enabled() and cb.is_visible():
-                                            cb.click_input()
-                                            print("Checkbox 'Lotes Consistentes' desmarcado com sucesso.")
-                                    except Exception as e:
-                                        print(f"Não foi possível desmarcar checkbox: {e}")
-
-                                    time.sleep(5)
-                                    return RpaRetornoProcessoDTO(
-                                        sucesso=False,
-                                        retorno="Integração realizada, porém, existem LOTES INCONSISTENTES.",
-                                        status=RpaHistoricoStatusEnum.Falha,
-                                        tags=[RpaTagDTO(descricao=RpaTagEnum.Negocio)],
-                                    )
-
-                                # Se não estava marcado, aí sim é sucesso
-                                return RpaRetornoProcessoDTO(
-                                    sucesso=True,
-                                    retorno=mensagem,
-                                    status=RpaHistoricoStatusEnum.Sucesso,
-                                    tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
-                                )
-
-                            # Demais imagens tratadas como ERRO
+                                msg_box.child_window(
+                                    class_name="TBitBtn", found_index=0
+                                ).click_input()
+                                break
+
                             console.print(f"[red]Erro encontrado:[/red] {mensagem}")
-                            try:
-                                msg_box.child_window(class_name="TBitBtn", found_index=0).click_input()
-                            except:
-                                pass
+
                             return RpaRetornoProcessoDTO(
                                 sucesso=False,
                                 retorno=mensagem,
@@ -510,16 +480,19 @@ async def integracao_contabil_generica(
 
                     await worker_sleep(1)
 
-                    # Se nenhuma imagem foi encontrada, tenta só fechar a TMsgBox (se existir) e sai
                     try:
                         app = Application(backend="win32").connect(
                             class_name="TMsgBox", found_index=0
                         )
                         main_window = app["TMsgBox"]
-                        main_window.child_window(class_name="TBitBtn", found_index=0).click_input()
+                        main_window.child_window(
+                            class_name="TBitBtn", found_index=0
+                        ).click_input()
                         break
                     except ElementNotFoundError:
-                        console.print("[yellow]Janela TMsgBox ainda não visível.[/yellow]")
+                        console.print(
+                            "[yellow]Janela TMsgBox ainda não visível.[/yellow]"
+                        )
                         break
 
                 except ElementNotFoundError:
@@ -530,23 +503,6 @@ async def integracao_contabil_generica(
 
                 await worker_sleep(1)
 
-                try:
-                    app = Application(backend="win32").connect(
-                        class_name="TMsgBox", found_index=0
-                    )
-                    main_window = app["TMsgBox"]
-                    main_window.child_window(class_name="TBitBtn", found_index=0).click_input()
-                    time.sleep(5)
-                    break
-                except ElementNotFoundError:
-                    console.print("[yellow]Janela TMsgBox ainda não visível.[/yellow]")
-
-                except Exception as e:
-                    print(f"Erro inesperado ao verificar janela de confirmação: {e}")
-                    break
-
-                await worker_sleep(1)
-
                 try:
                     app = Application(backend="win32").connect(
                         class_name="TMsgBox", found_index=0
@@ -560,7 +516,28 @@ async def integracao_contabil_generica(
                 except ElementNotFoundError:
                     console.print("[yellow]Janela TMsgBox ainda não visível.[/yellow]")
 
-            
+            if lotesMarcados:
+                if (
+                    checkbox.exists()
+                    and checkbox.is_enabled()
+                    and checkbox.is_visible()
+                ):
+                    checkbox.click_input()
+                    print("Checkbox 'Lotes Consistentes' desmarcado com sucesso.")
+                time.sleep(5)
+                return RpaRetornoProcessoDTO(
+                    sucesso=False,
+                    retorno=f"Integração realizada, porém, existem LOTES INCONSISTENTES.",
+                    status=RpaHistoricoStatusEnum.Falha,
+                    tags=[RpaTagDTO(descricao=RpaTagEnum.Negocio)],
+                )
+            else:
+                return RpaRetornoProcessoDTO(
+                    sucesso=True,
+                    retorno=f"Sucesso ao executar processo de integracao contabil",
+                    status=RpaHistoricoStatusEnum.Sucesso,
+                    tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+                )
         except Exception as erro:
             return RpaRetornoProcessoDTO(
                 sucesso=False,
@@ -576,4 +553,3 @@ async def integracao_contabil_generica(
             status=RpaHistoricoStatusEnum.Falha,
             tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
         )
-

{worker_automate_hub-0.5.814 → worker_automate_hub-0.5.816}/worker_automate_hub/tasks/task_definitions.py

@@ -105,6 +105,9 @@ from worker_automate_hub.tasks.jobs.opex_capex import (
 from worker_automate_hub.tasks.jobs.entrada_de_notas_22 import (
     entrada_de_notas_22,
 )
+from worker_automate_hub.tasks.jobs.geracao_balancetes_filial import (
+    geracao_balancetes_filial,
+)
 
 
 task_definitions = {
@@ -212,7 +215,8 @@ task_definitions = {
     "16debe45-3520-4f63-acfe-ef0e8784fcab": extracao_saldo_estoque,
     "9cbc6016-7c0e-4a3a-8ee9-fb9dc4b35e33": extracao_saldo_estoque_fiscal,
     "07072711-c9d0-49e4-b180-530cecbe0728": opex_capex,
-    "98bc6679-2e6b-4757-9fdc-b27eebd98f54": entrada_de_notas_22
+    "98bc6679-2e6b-4757-9fdc-b27eebd98f54": entrada_de_notas_22,
+    "d7794924-0330-453c-b79b-74f3c8991562": geracao_balancetes_filial
 }