worker-automate-hub 0.5.758__tar.gz → 0.5.760__tar.gz

{worker_automate_hub-0.5.758 → worker_automate_hub-0.5.760}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.758"
+version = "0.5.760"
 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.760/worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py

@@ -0,0 +1,224 @@
+import asyncio
+import os
+from datetime import datetime
+from pywinauto import Application, timings, findwindows
+import sys
+import io
+# 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 rich.console import Console
+import re
+from pywinauto.keyboard import send_keys
+import warnings
+from pywinauto.application import Application
+from worker_automate_hub.api.client import get_config_by_name, send_file
+from worker_automate_hub.utils.util import (
+    kill_all_emsys,
+    login_emsys,
+    set_variable,
+    type_text_into_field,
+    worker_sleep,
+)
+from pywinauto_recorder.player import set_combobox
+
+from datetime import timedelta
+import pyautogui
+from worker_automate_hub.utils.logger import logger
+from worker_automate_hub.utils.utils_nfe_entrada import EMSys
+
+emsys = EMSys()
+
+console = Console()
+pyautogui.PAUSE = 0.5
+pyautogui.FAILSAFE = False
+
+
+async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
+    try:
+        config = await get_config_by_name("login_emsys")
+        periodo = task.configEntrada["periodo"]
+        periodo_format = periodo.replace("/", "")
+        filial = task.configEntrada["filialEmpresaOrigem"]
+        historico_id = task.historico_id
+
+        console.print("Finalizando processos antigos do EMSys...", style="bold yellow")
+        await kill_all_emsys()
+
+        console.print("Iniciando EMSys...", style="bold green")
+        app = Application(backend="win32").start("C:\\Rezende\\EMSys3\\EMSys3_35.exe")
+        warnings.filterwarnings(
+            "ignore",
+            category=UserWarning,
+            message="32-bit application should be automated using 32-bit Python",
+        )
+
+        console.print("Fazendo login no EMSys...", style="bold cyan")
+        return_login = await login_emsys(
+            config.conConfiguracao, app, task, filial_origem=filial
+        )
+
+        if return_login.sucesso:
+            console.print("Login realizado com sucesso", style="bold green")
+            type_text_into_field(
+                "Rel. Saldo Estoque ", app["TFrmMenuPrincipal"]["Edit"], True, "50"
+            )
+            pyautogui.press("enter")
+            await worker_sleep(2)
+            pyautogui.press("enter")
+        else:
+            logger.info(f"\nError Message: {return_login.retorno}")
+            console.print(f"Erro no login: {return_login.retorno}", style="bold red")
+            return return_login
+
+        await worker_sleep(2)
+
+        console.print("Abrindo janela Relatório de Saldo de Estoque...", style="bold cyan")
+        app = Application().connect(class_name="TFrmRelSaldoEstoque", timeout=60)
+        main_window = app["TFrmRelSaldoEstoque"]
+        main_window.set_focus()
+
+        console.print("Marcando campo de data...", style="bold cyan")
+        main_window.child_window(class_name="TCheckBox", found_index=3).click_input()
+
+        await worker_sleep(2)
+
+        console.print(f"Inserindo período: {periodo}", style="bold cyan")
+        data_input = main_window.child_window(class_name="TDBIEditDate", found_index=0) 
+        data_input.set_edit_text(periodo)
+
+        console.print("Gerando relatório...", style="bold cyan")
+        main_window.child_window(class_name="TBitBtn", found_index=0).click_input()  
+
+        timings.wait_until_passes(
+            timeout=1800,
+            retry_interval=1,
+            func=lambda: Application().connect(class_name="TFrmPreviewRelatorio"),
+        )
+        await worker_sleep(2)
+
+        console.print("Abrindo Preview Relatório...", style="bold cyan")
+        app = Application().connect(class_name="TFrmPreviewRelatorio")
+        main_window = app["TFrmPreviewRelatorio"]
+        main_window.set_focus()
+
+        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")
+
+            console.print("Procurando botão de salvar (imagem)...", style="bold cyan")
+            caminho = r'assets\\extracao_relatorios\\btn_salvar.png'
+            if os.path.isfile(caminho):
+                pos = pyautogui.locateCenterOnScreen(caminho, 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:
+                console.print("Imagem do botão salvar NÃO existe", style="bold red")
+
+            await worker_sleep(8)
+
+            console.print("Selecionando formato Excel...", style="bold cyan")
+            app = Application().connect(class_name="TFrmRelatorioFormato")
+            main_window = app["TFrmRelatorioFormato"]
+            combo = main_window.ComboBox
+            console.print(f"Itens do ComboBox: {combo.texts()}", style="bold yellow")
+            combo.select(8)
+
+            await worker_sleep(5)
+
+            console.print("Confirmando salvar...", style="bold cyan")
+            main_window.child_window(class_name="TBitBtn", found_index=1).click_input()  
+
+            await worker_sleep(5)
+
+            console.print("Abrindo janela de salvar arquivo...", style="bold cyan")
+            app = Application().connect(title_re="Salvar para arquivo", timeout=30)
+            main_window = app.window(title_re="Salvar para arquivo")
+
+            caminho_arquivo = rf"C:\Users\automatehub\Downloads\saldo_estoque_{periodo_format}_{filial}.xlsx"
+            campo_nome = main_window.child_window(class_name="Edit", control_id=1148).wrapper_object()
+            campo_nome.set_edit_text(caminho_arquivo)
+            console.print(f"Arquivo configurado para: {caminho_arquivo}", style="bold green")
+
+            await worker_sleep(2)
+            
+            main_window.child_window(class_name="Button", found_index=0).click_input()  
+
+            await worker_sleep(4)
+
+            console.print("Aguardando finalização do processo de impressão/salvamento...", style="bold cyan")
+            try:
+                app = Application().connect(title_re="Printing")
+                janela = app.window(title_re="Printing")
+                janela.wait_not("visible", timeout=60)
+                console.print("Janela 'Printing' fechada.", style="bold green")
+            except findwindows.ElementNotFoundError:
+                console.print("Janela 'Printing' não apareceu.", style="bold yellow")
+
+            # Verifica se o arquivo foi salvo
+            if os.path.exists(caminho_arquivo):
+                console.print(f"Arquivo encontrado: {caminho_arquivo}", style="bold green")
+                with open(caminho_arquivo, "rb") as file:
+                    file_bytes = io.BytesIO(file.read())
+                sucesso = True
+            else:
+                console.print("Arquivo não encontrado, tentando novamente...", style="bold red")
+                tentativa += 1
+                await worker_sleep(3)
+
+        if not sucesso:
+            console.print("Falha após 5 tentativas. Arquivo não foi gerado.", style="bold red")
+
+        nome_com_extensao = f'saldo_estoque_{periodo_format}_{filial}.xlsx'
+        console.print("Enviando arquivo XLS para o BOF...", style="bold cyan")
+        try:
+            await send_file(
+                historico_id,
+                nome_com_extensao,
+                "xlsx",
+                file_bytes,
+                file_extension="xlsx",
+            )
+            console.print("Removendo arquivo da pasta downloads", style="bold yellow")
+            os.remove(caminho_arquivo)
+            return RpaRetornoProcessoDTO(
+                sucesso=True,
+                retorno="Relatório enviado com sucesso!",
+                status=RpaHistoricoStatusEnum.Sucesso,
+            )
+
+        except Exception as e:
+            console.print(f"Erro ao enviar o arquivo: {e}", style="bold red")
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno=f"Erro ao enviar o arquivo: {e}",
+                status=RpaHistoricoStatusEnum.Falha,
+                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+            )
+
+        console.print("Processo concluído com sucesso!", style="bold green")
+
+    except Exception as ex:
+        retorno = f"Erro Processo Fechamento Balancete: {str(ex)}"
+        logger.error(retorno)
+        console.print(retorno, style="bold red")
+        return RpaRetornoProcessoDTO(
+            sucesso=False,
+            retorno=retorno,
+            status=RpaHistoricoStatusEnum.Falha,
+            tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+        )

{worker_automate_hub-0.5.758 → worker_automate_hub-0.5.760}/worker_automate_hub/tasks/jobs/extracao_saldo_estoque_fiscal.py

@@ -340,28 +340,51 @@ async def extracao_saldo_estoque_fiscal(
         print("✅ Texto inserido no campo Nome")
 
         await worker_sleep(3)
+       
+        # Clicar em 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")
+
+        # 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 = f"C:\\Users\\automatehub\\Downloads\\saldo_estoque_fiscal_{periodo_format}_{filial}.xlsx"
+        campo_nome.set_focus()
+        campo_nome.set_edit_text(caminho_arquivo)
+
+        print("✅ Texto inserido no campo Nome")
+
+        await worker_sleep(2)
 
         # Clicar em ok para salvar
-        keyboard.send_keys("{TAB}{TAB}{ENTER}", pause=0.3)
+        main_window.child_window(class_name="Button", found_index=0).click_input()
 
-        await worker_sleep(5)
+        await worker_sleep(20)
 
-        caminho_img = r"assets\\extracao_realtorios\\janela_printing.png"
+        # caminho_img = r"assets\\extracao_relatorios\\janela_printing.png"
 
         # 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")
 
-            print("Aguardando a janela 'Printing' sumir...")
+            print("⏳ Aguardando a janela 'Printing' sumir...")
             janela.wait_not("visible", timeout=60)  # espera até 60 segundos
-            print("Janela 'Printing' fechada.")
+            print("✅ Janela 'Printing' fechada.")
 
         except findwindows.ElementNotFoundError:
-            print("Janela 'Printing' não estava aberta.")
+            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 +401,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 +414,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)

worker_automate_hub-0.5.758/worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py

@@ -1,233 +0,0 @@
-import asyncio
-import os
-from datetime import datetime
-from pywinauto import Application, timings, findwindows
-import sys
-import io
-# 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 rich.console import Console
-import re
-from pywinauto.keyboard import send_keys
-import warnings
-from pywinauto.application import Application
-from worker_automate_hub.api.client import get_config_by_name, send_file
-from worker_automate_hub.utils.util import (
-    kill_all_emsys,
-    login_emsys,
-    set_variable,
-    type_text_into_field,
-    worker_sleep,
-)
-from pywinauto_recorder.player import set_combobox
-
-from datetime import timedelta
-import pyautogui
-from worker_automate_hub.utils.logger import logger
-from worker_automate_hub.utils.utils_nfe_entrada import EMSys
-
-emsys = EMSys()
-
-console = Console()
-pyautogui.PAUSE = 0.5
-pyautogui.FAILSAFE = False
-
-
-async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
-    try:
-        config = await get_config_by_name("login_emsys")
-        periodo = task.configEntrada["periodo"]
-        periodo_format = periodo.replace("/", "")
-        filial = task.configEntrada["filialEmpresaOrigem"]
-        historico_id = task.historico_id
-        await kill_all_emsys()
-
-        app = Application(backend="win32").start("C:\\Rezende\\EMSys3\\EMSys3_35.exe")
-        warnings.filterwarnings(
-            "ignore",
-            category=UserWarning,
-            message="32-bit application should be automated using 32-bit Python",
-        )
-        console.print("\nEMSys iniciando...", style="bold green")
-        return_login = await login_emsys(
-            config.conConfiguracao, app, task, filial_origem=filial
-        )
-
-        if return_login.sucesso == True:
-            type_text_into_field(
-                "Rel. Saldo Estoque ", app["TFrmMenuPrincipal"]["Edit"], True, "50"
-            )
-            pyautogui.press("enter")
-            await worker_sleep(2)
-            pyautogui.press("enter")
-
-        else:
-            logger.info(f"\nError Message: {return_login.retorno}")
-            console.print(f"\nError Message: {return_login.retorno}", style="bold red")
-            return return_login
-
-        await worker_sleep(2)
-
-        ##### Janela Relatório Saldos do Estoque #####
-        # Marcar check box data
-        app = Application().connect(class_name="TFrmRelSaldoEstoque", timeout=60)
-        main_window = app["TFrmRelSaldoEstoque"]
-        main_window.set_focus()
-
-        # Captura o campo de data
-        data_chk = main_window.child_window(
-            class_name="TCheckBox", found_index=3
-        ).click_input()
-
-        await worker_sleep(2)
-        # Insere a data
-        data_input = main_window.child_window(class_name="TDBIEditDate", found_index=0)
-
-        data_input.set_edit_text(periodo)
-
-        # Clicar em gerar relatório
-        btn_gerar = main_window.child_window(
-            class_name="TBitBtn", found_index=0
-        ).click_input()
-
-        # Aguarda até 60 segundos para a janela aparecer
-        timings.wait_until_passes(
-            timeout=1800,
-            retry_interval=1,
-            func=lambda: Application().connect(class_name="TFrmPreviewRelatorio"),
-        )
-
-        await worker_sleep(10)
-
-        # Conecta à janela Preview Relatorio
-        app = Application().connect(class_name="TFrmPreviewRelatorio")
-        main_window = app["TFrmPreviewRelatorio"]
-        main_window.set_focus()
-
-        # 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.")
-            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)
-
-        # Conecta na janela Configuração para Salvar Arquivo
-        app = Application().connect(class_name="TFrmRelatorioFormato")
-        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)
-
-        # Clicar em 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")
-
-        # 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 = f"C:\\Users\\automatehub\\Downloads\\saldo_estoque_{periodo_format}_{filial}.xlsx"
-        campo_nome.set_focus()
-        campo_nome.set_edit_text(caminho_arquivo)
-
-        print("✅ Texto inserido no campo Nome")
-
-        await worker_sleep(2)
-
-        # Clicar em ok para salvar
-        main_window.child_window(class_name="Button", found_index=0).click_input()
-
-        await worker_sleep(20)
-
-        # caminho_img = r"assets\\extracao_relatorios\\janela_printing.png"
-
-        # 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")
-
-            print("⏳ Aguardando a janela 'Printing' sumir...")
-            janela.wait_not("visible", timeout=60)  # espera até 60 segundos
-            print("✅ Janela 'Printing' fechada.")
-
-        except findwindows.ElementNotFoundError:
-            print("⚠️ Janela 'Printing' não estava aberta.")
-
-        nome_com_extensao = f"saldo_estoque_{periodo_format}_{filial}.xlsx"
-        # lê o arquivo
-        print(caminho_arquivo)
-        with open(f"{caminho_arquivo}", "rb") as file:
-            file_bytes = io.BytesIO(file.read())
-
-        console.print("Enviar Excel para o BOF")
-        try:
-            await send_file(
-                historico_id,
-                nome_com_extensao,
-                "xlsx",
-                file_bytes,
-                file_extension="xlsx",
-            )
-            console.print("Removendo arquivo XLS da pasta downloads")
-            os.remove(f"{caminho_arquivo}")
-            return RpaRetornoProcessoDTO(
-                sucesso=True,
-                retorno="Relatório gerado com sucesso",
-                status=RpaHistoricoStatusEnum.Sucesso,
-            )
-
-        except Exception as e:
-            console.print(f"Erro ao enviar o arquivo: {e}", style="bold red")
-            return RpaRetornoProcessoDTO(
-                sucesso=False,
-                retorno=f"Erro ao enviar o arquivo: {e}",
-                status=RpaHistoricoStatusEnum.Falha,
-                tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
-            )
-    except Exception as ex:
-        retorno = f"Erro Processo Fechamento Balancete: {str(ex)}"
-        logger.error(retorno)
-        console.print(retorno, style="bold red")
-        return RpaRetornoProcessoDTO(
-            sucesso=False,
-            retorno=retorno,
-            status=RpaHistoricoStatusEnum.Falha,
-            tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
-        )