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

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

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

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

@@ -4,6 +4,7 @@ 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,
@@ -21,15 +22,14 @@ from worker_automate_hub.utils.util import (
 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.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
@@ -69,7 +69,9 @@ async def open_contabil_processes():
         console.print(f"Error: {error}")
 
 
-async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
+async def geracao_balancetes_filial(
+    task: RpaProcessoEntradaDTO,
+) -> RpaRetornoProcessoDTO:
     try:
         await kill_all_emsys()
         await open_contabil_processes()
@@ -112,7 +114,7 @@ async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
         await worker_sleep(4)
         pyautogui.write(config.conConfiguracao.get("pass"))
         pyautogui.press("enter")
-        
+
         await worker_sleep(10)
 
         main_window = None
@@ -151,17 +153,23 @@ async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
         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()
+        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
+            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)
+            send_keys("^a{DELETE}")
+            send_keys(
+                periodo_inicial.replace("/", "")
+            )  # fallback: digita só dígitos (máscara)
 
         # ---- Final ----
         try:
@@ -169,58 +177,76 @@ async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
         except Exception:
             ctrl_final.set_focus()
             ctrl_final.click_input()
-            send_keys('^a{DELETE}')
-            send_keys(periodo_final.replace('/', ''))
-        
-        await worker_sleep(2)
-        
+            send_keys("^a{DELETE}")
+            send_keys(periodo_final.replace("/", ""))
+
+        await worker_sleep(4)
+
         console.print("Selecionar detalhada por centro de custo..")
-        detalhada = win.child_window(
-            class_name="TRzComboBox", found_index=0
-        )
+        detalhada = win.child_window(class_name="TRzComboBox", found_index=0)
         detalhada.select("Centro de Custo")
-        
-        await worker_sleep(2)
+
+        await worker_sleep(6)
 
         console.print("Selecionar considerar contas analíticas zerada")
-        contas_analit_zeradas = win.child_window(class_name="TRzCheckBox", found_index=2).click()
+        contas_analit_zeradas = win.child_window(
+            class_name="TRzCheckBox", found_index=2
+        ).click()
+
+        await worker_sleep(4)
 
         console.print("Selecionar considerar contas sintéticas zerada")
-        contas_sint_zeradas = win.child_window(class_name="TRzCheckBox", found_index=0).click()
+        contas_sint_zeradas = win.child_window(
+            class_name="TRzCheckBox", found_index=0
+        ).click()
+
+        await worker_sleep(4)
 
         console.print("Selecionar por filiais")
-        selec_filiais = win.child_window(class_name="TRzCheckBox", found_index=3).click()
+        selec_filiais = win.child_window(
+            class_name="TRzCheckBox", found_index=3
+        ).click()
+
+        await worker_sleep(4)
 
         console.print("Selecionar CSV")
         selec_csv = win.child_window(class_name="TRzComboBox", found_index=1)
         selec_csv.select("Arquivo CSV")
 
+        await worker_sleep(4)
+
         console.print("Clicar em gerar relatório")
-        btn_gerar_relatorio = win.child_window(class_name="TBitBtn", found_index=0).click()
+        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")
+        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
+            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("^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
+        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)
 
@@ -228,53 +254,62 @@ async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
 
         # 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)
+        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("/","")
+            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}")    
+            edit.set_edit_text("")  # limpa
+            edit.set_edit_text(
+                rf"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)
+            send_keys("^a{DELETE}")
+            send_keys(
+                rf"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)")
+        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")
+        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}"
+        path_to_txt = rf"C:\Users\automatehub\Downloads\balancete_{periodo_inicial}_{periodo_final}_{filial}"
 
-        with open(f"{path_to_txt}.csv", 'rb') as file:
+        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")
+            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
+                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}!"
@@ -283,9 +318,9 @@ async def geracao_balancetes_filial(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
                 sucesso=False,
                 retorno=result,
                 status=RpaHistoricoStatusEnum.Falha,
-                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
             )
-        
+
     except Exception as erro:
         return RpaRetornoProcessoDTO(
             sucesso=False,