worker-automate-hub 0.5.758__py3-none-any.whl → 0.5.759__py3-none-any.whl

worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py

@@ -49,154 +49,142 @@ async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
         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("\nEMSys iniciando...", style="bold green")
+
+        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 == True:
+        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"\nError Message: {return_login.retorno}", style="bold red")
+            console.print(f"Erro no login: {return_login.retorno}", style="bold red")
             return return_login
 
         await worker_sleep(2)
 
-        ##### Janela Relatório Saldos do Estoque #####
-        # Marcar check box data
+        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()
 
-        # Captura o campo de data
-        data_chk = main_window.child_window(
-            class_name="TCheckBox", found_index=3
-        ).click_input()
+        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)
-        # Insere a data
-        data_input = main_window.child_window(class_name="TDBIEditDate", found_index=0)
 
+        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)
 
-        # Clicar em gerar relatório
-        btn_gerar = main_window.child_window(
-            class_name="TBitBtn", found_index=0
-        ).click_input()
+        console.print("Gerando relatório...", style="bold cyan")
+        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(2)
 
-        await worker_sleep(10)
-
-        # Conecta à janela Preview Relatorio
+        console.print("Abrindo Preview Relatório...", style="bold cyan")
         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.")
+        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:
-                print("Imagem encontrada no disco, mas não está visível na tela.")
-        else:
-            print("A imagem NÃO existe:", caminho)
+                console.print("Imagem do botão salvar NÃO existe", style="bold red")
 
-        await worker_sleep(2)
+            await worker_sleep(8)
 
-        # 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)
+            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(2)
-
-        # Clicar em Salvar
-        main_window.child_window(class_name="TBitBtn", found_index=1).click_input()
+            await worker_sleep(5)
 
-        await worker_sleep(5)
+            console.print("Confirmando salvar...", style="bold cyan")
+            main_window.child_window(class_name="TBitBtn", found_index=1).click_input()  
 
-        # 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")
+            await worker_sleep(5)
 
-        # 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)
+            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")
 
-        print("✅ Texto inserido no campo Nome")
+            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)
-
-        # 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.")
+            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)
 
-        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())
+        if not sucesso:
+            console.print("Falha após 5 tentativas. Arquivo não foi gerado.", style="bold red")
 
-        console.print("Enviar Excel para o BOF")
+        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,
@@ -205,11 +193,11 @@ async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
                 file_bytes,
                 file_extension="xlsx",
             )
-            console.print("Removendo arquivo XLS da pasta downloads")
-            os.remove(f"{caminho_arquivo}")
+            console.print("Removendo arquivo da pasta downloads", style="bold yellow")
+            os.remove(caminho_arquivo)
             return RpaRetornoProcessoDTO(
                 sucesso=True,
-                retorno="Relatório gerado com sucesso",
+                retorno="Relatório enviado com sucesso!",
                 status=RpaHistoricoStatusEnum.Sucesso,
             )
 
@@ -221,6 +209,9 @@ async def extracao_saldo_estoque(task: RpaProcessoEntradaDTO):
                 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)

{worker_automate_hub-0.5.758.dist-info → worker_automate_hub-0.5.759.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.758
+Version: 0.5.759
 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.dist-info → worker_automate_hub-0.5.759.dist-info}/RECORD

@@ -70,7 +70,7 @@ worker_automate_hub/tasks/jobs/entrada_de_notas_9000.py,sha256=0mOmS28tQKF5m7vMz
 worker_automate_hub/tasks/jobs/exemplo_processo.py,sha256=nV0iLoip2FH2-FhLmhX3nPqsfl_MPufZ3E5Q5krJvdc,3544
 worker_automate_hub/tasks/jobs/extracao_fechamento_contabil.py,sha256=6Kr5DKjKLqtFvGzyiXtt7xrQsuU898l8pQXDq9C6AX8,19567
 worker_automate_hub/tasks/jobs/extracao_fechamento_emsys.py,sha256=-T2nZUDiFrUGm_KLxJd_4qcrageDxVpWW3KAAniLFC4,21448
-worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py,sha256=LocziY74XAKdFyyPJPRt9_zs3J59-kysggjmpASqdxU,8384
+worker_automate_hub/tasks/jobs/extracao_saldo_estoque.py,sha256=nFAv5Bvwwo2Mhy69q5BqR-XsKtg4KIqAunrUmGeW1S0,9173
 worker_automate_hub/tasks/jobs/extracao_saldo_estoque_fiscal.py,sha256=OJQAkE6mqm7lNodleYRWwJq3syxrTqcPGt27_u6NPPs,14950
 worker_automate_hub/tasks/jobs/fechar_conexao_rdp.py,sha256=UWAKCS2dbfgDlSQOBdjmVJXfD1MMuUrOi3weDgB0CAc,5718
 worker_automate_hub/tasks/jobs/fidc_exportacao_docs_portal_b2b.py,sha256=tWUmYy3Zhi3JEt8AoqTsWpU-wbf5-OxhCrTOooh1WH4,15616
@@ -101,7 +101,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 worker_automate_hub/utils/util.py,sha256=V2WtWoETdTrAtGA8UgeqAAVphUj9KkGSZFzYsHJFATA,210055
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=TOXKSHOPxy8N3-ROpTGjNIHstX0i2b8qekcj1tRvjG8,38174
 worker_automate_hub/worker.py,sha256=uhZ3f-iaQ1i8cANbljp50vkYl-Xm0_sHtjwwF_2y72o,7191
-worker_automate_hub-0.5.758.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.758.dist-info/METADATA,sha256=1xkBHjCMiZUphzKk7_pcpyhiQxsr6bHAO9-4ihxnt8M,3049
-worker_automate_hub-0.5.758.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-worker_automate_hub-0.5.758.dist-info/RECORD,,
+worker_automate_hub-0.5.759.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.759.dist-info/METADATA,sha256=tMWbnY_CoFvxTSMsKPGVRbkAsl_HM5XxtbKuLCk0Ajw,3049
+worker_automate_hub-0.5.759.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+worker_automate_hub-0.5.759.dist-info/RECORD,,