worker-automate-hub 0.5.53__py3-none-any.whl → 0.5.54__py3-none-any.whl

worker_automate_hub/api/webhook_service.py

@@ -8,7 +8,8 @@ console = Console()
 async def send_to_webhook(
     url_retorno: str,
     status: str,
-    observacao: str
+    observacao: str,
+    identificador_webhook: str = None
 ) -> None:    
     """
     Envia uma notificacao para o endpoint do webhook passado como parametro.
@@ -35,6 +36,9 @@ async def send_to_webhook(
         "observacao": observacao,
     }
 
+    if identificador_webhook:
+        data["identificador"] = identificador_webhook
+
     try:
         async with aiohttp.ClientSession(
             connector=aiohttp.TCPConnector(verify_ssl=False)

worker_automate_hub/tasks/jobs/devolucao_prazo_a_faturar.py

@@ -35,6 +35,7 @@ from worker_automate_hub.utils.util import (
     e_ultimo_dia_util,
     delete_xml,
     ocr_warnings,
+    ocr_by_class,
     nf_busca_nf_saida,
     pessoas_ativa_cliente_fornecedor,
     nf_devolucao_liquidar_cupom,
@@ -758,9 +759,7 @@ async def devolucao_prazo_a_faturar(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
                     console.print(f"Imagem preparada com sucesso...\n")
                     console.print(f"Realizando OCR...\n")
                     captured_text = pytesseract.image_to_string(Image.open(path_to_png))
-                    console.print(
-                        f"Texto Full capturado {captured_text}...\n"
-                    )
+                    console.print(f"Texto Full capturado {captured_text}...\n")
                     os.remove(path_to_png)
                     pattern = r"sequ[êe]ncia:\s*(\d+)"
                     match = re.search(pattern, captured_text)
@@ -769,6 +768,9 @@ async def devolucao_prazo_a_faturar(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
                         numero_nota_fiscal = match.group(1)
 
 
+                    app = Application().connect(title="Information", timeout=180)
+                    main_window = app["Information"]
+                    main_window.set_focus()
                     btn_ok = main_window.child_window(class_name="TButton", found_index=0)
                     btn_ok.click()
                     await worker_sleep(3)
@@ -819,17 +821,37 @@ async def devolucao_prazo_a_faturar(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
                 
                 while i < max_attempts:
                     try:
-                        app = Application().connect(class_name="TFrmProcessamentoFEe2", timeout=60)
+                        app = Application().connect(class_name="TFrmProcessamentoFEe2", timeout=10)
                         main_window = app["TFrmProcessamentoFEe2"]
 
-                        await worker_sleep(15)
-
-                        information_operacao_concluida = main_window.child_window(class_name="TMessageForm")
-                        btn_ok = information_operacao_concluida.child_window(class_name="TButton")
-                        btn_ok.click()
-                        await worker_sleep(4)
-                        main_window.close()
-                        break
+                        await worker_sleep(5)
+                        information_pop_up = await is_window_open_by_class("TMessageForm", "TMessageForm")
+                        if information_pop_up["IsOpened"] == True:
+                            msg_pop_up = await ocr_by_class(numero_nota_fiscal, "TMessageForm", "TMessageForm")
+                            if msg_pop_up.sucesso:
+                                if 'operação concl' in msg_pop_up or 'operacao concl' in msg_pop_up:
+                                    information_operacao_concluida = main_window.child_window(class_name="TMessageForm")
+                                    btn_ok = information_operacao_concluida.child_window(class_name="TButton")
+                                    btn_ok.click()
+                                    await worker_sleep(4)
+                                    main_window.close()
+                                    break
+                                else:
+                                    retorno = f"Pop up nao mapeado para seguimento do robo {msg_pop_up} \nEtapas Executadas:\n{steps}"
+                                    return RpaRetornoProcessoDTO(
+                                        sucesso=False,
+                                        retorno=retorno,
+                                        status=RpaHistoricoStatusEnum.Falha,
+                                        tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+                                    )
+                            else:
+                                retorno = f"Não foi possivel realizar a confirmação do msg do OCR \nEtapas Executadas:\n{steps}"
+                                return RpaRetornoProcessoDTO(
+                                    sucesso=False,
+                                    retorno=retorno,
+                                    status=RpaHistoricoStatusEnum.Falha,
+                                    tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+                                )
                     except Exception as e:
                         pass
 

worker_automate_hub/tasks/task_executor.py

@@ -36,6 +36,7 @@ async def perform_task(task: RpaProcessoEntradaDTO):
     logger.info(log_msg)
     task_uuid = task.uuidProcesso
     url_retorno = task.configEntrada.get("urlRetorno", None)
+    identificador_webhook = task.configEntrada.get("identificador", None)
     processo: RpaProcesso = await get_processo(task_uuid)
     if processo is None:
         worker_config = load_worker_config()
@@ -77,7 +78,10 @@ async def perform_task(task: RpaProcessoEntradaDTO):
                 show_toast("Sucesso", f"Processo executado com sucesso: {result}")
 
             if url_retorno is not None and result.sucesso == False:
-                await send_to_webhook(url_retorno, result.status, result.retorno)
+                if identificador_webhook:
+                    await send_to_webhook(url_retorno, result.status, result.retorno, identificador_webhook)
+                else:
+                    await send_to_webhook(url_retorno, result.status, result.retorno)
             return result
         else:
             err_msg = f"Processo não encontrado: {task_uuid}"

worker_automate_hub/utils/util.py

@@ -3437,6 +3437,60 @@ async def ocr_warnings(numero_nota: str)-> RpaRetornoProcessoDTO:
             pass
 
 
+async def ocr_by_class(numero_nota: str, windows_class: str, app_class: str)-> RpaRetornoProcessoDTO:
+    try:
+        app = Application().connect(class_name=windows_class)
+        main_window = app[app_class]
+        main_window.set_focus()
+
+        console.print(f"Obtendo texto da Janela...\n")
+        console.print(f"Tirando print da janela do warning para realização do OCR...\n")
+
+        window_rect = main_window.rectangle()
+        screenshot = pyautogui.screenshot(
+            region=(
+                window_rect.left,
+                window_rect.top,
+                window_rect.width(),
+                window_rect.height(),
+            )
+        )
+        username = getpass.getuser()
+        path_to_png = f"C:\\Users\\{username}\\Downloads\\ocr_bt_class_popup_{numero_nota}.png"
+        screenshot.save(path_to_png)
+        console.print(f"Print salvo em {path_to_png}...\n")
+
+        console.print(
+            f"Preparando a imagem para maior resolução e assertividade no OCR...\n"
+        )
+        image = Image.open(path_to_png)
+        image = image.convert("L")
+        enhancer = ImageEnhance.Contrast(image)
+        image = enhancer.enhance(2.0)
+        image.save(path_to_png)
+        console.print(f"Imagem preparada com sucesso...\n")
+        console.print(f"Realizando OCR...\n")
+        captured_text = pytesseract.image_to_string(Image.open(path_to_png))
+        console.print(f"Texto Full capturado {captured_text}...\n")
+        return RpaRetornoProcessoDTO(
+            sucesso=True,
+            retorno=captured_text,
+            status=RpaHistoricoStatusEnum.Sucesso,
+        )
+    except Exception as e:
+        return RpaRetornoProcessoDTO(
+            sucesso=False,
+            retorno=f"Erro ao realizar a extração do texto vinculado ao warning, erro - {e}",
+            status=RpaHistoricoStatusEnum.Falha,
+            tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+        )
+    finally:
+        try:
+            os.remove(path_to_png)
+        except:
+            pass
+
+
 async def nf_busca_nf_saida(num_nota_fiscal: str) -> RpaRetornoProcessoDTO:
     try:
         app = Application().connect(class_name="TFrmNotaFiscalSaida", timeout=60)
@@ -3592,30 +3646,54 @@ async def nf_devolucao_liquidar_cupom(num_nota_fiscal: str, data_atual: str) ->
         await worker_sleep(3)
         confirm_pop_up = await is_window_open_by_class("TMessageForm","TMessageForm")
         if confirm_pop_up["IsOpened"] == True:
-            app_confirm = Application().connect(class_name="TMessageForm")
-            main_window_confirm = app_confirm["TMessageForm"]
-
-            btn_yes = main_window_confirm["&Yes"]
-            try:
-                btn_yes.click()
-                await worker_sleep(3)
-                console.print("O botão Yes foi clicado com sucesso.", style="green")
-                #main_window.close()
-            except:
-                return RpaRetornoProcessoDTO(
-                    sucesso=False,
-                    retorno=f"Não foi possivel clicar em Yes durante para a confirmação da liquidação do titulo",
-                    status=RpaHistoricoStatusEnum.Falha,
-                    tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
-                )
+            capture_text_ocr = await ocr_by_class(num_nota_fiscal,"TMessageForm","TMessageForm")
+            if capture_text_ocr.sucesso == True:
+                msg_result = capture_text_ocr.retorno
+                if "deseja realmente" in msg_result.lower():
+                    app_confirm = Application().connect(class_name="TMessageForm")
+                    main_window_confirm = app_confirm["TMessageForm"]
+
+                    btn_yes = main_window_confirm["&Yes"]
+                    try:
+                        btn_yes.click()
+                        await worker_sleep(3)
+                        console.print("O botão Yes foi clicado com sucesso.", style="green")
+                        #main_window.close()
+                    except:
+                        return RpaRetornoProcessoDTO(
+                            sucesso=False,
+                            retorno=f"Não foi possivel clicar em Yes durante para a confirmação da liquidação do titulo",
+                            status=RpaHistoricoStatusEnum.Falha,
+                            tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+                        )
+                else:
+                    return RpaRetornoProcessoDTO(
+                        sucesso=False,
+                        retorno=f"POP não mapeado: {msg_result}",
+                        status=RpaHistoricoStatusEnum.Falha,
+                        tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+                    )
         
         console.print('Aguardando a confirmação de Liquidação realizado com sucesso')
         try:
             app = Application().connect(title="Informação", timeout=60)
             main_window = app["Informação"]
             main_window.set_focus()
-            btn_ok = main_window.child_window(class_name="Button")
-            btn_ok.click()
+            msg_pop_up = await ocr_by_class(num_nota_fiscal,"TMessageForm","TMessageForm")
+            if msg_pop_up.sucesso == True:
+                msg_result = msg_pop_up.retorno
+                if "sucesso" in msg_result.lower():
+                    btn_ok = main_window.child_window(class_name="Button")
+                    btn_ok.click()
+                    pyautogui.press('enter')
+                    await worker_sleep(1)
+                else:
+                    return RpaRetornoProcessoDTO(
+                        sucesso=False,
+                        retorno=f"Não foi possivel obter a confirmação de titulo liquidado com sucesso, pop_up nao mepeado: {msg_result}",
+                        status=RpaHistoricoStatusEnum.Falha,
+                        tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
+                    )
         except Exception as e:
             return RpaRetornoProcessoDTO(
                 sucesso=False,
@@ -3624,6 +3702,10 @@ async def nf_devolucao_liquidar_cupom(num_nota_fiscal: str, data_atual: str) ->
                 tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)]
             )
         
+
+        app = Application().connect(class_name="TFrmTituloReceber", timeout=60)
+        main_window = app["TFrmTituloReceber"]
+        main_window.set_focus()
         console.print('Confirmando se o titulo alterou para Liquidado')
         panel_TPage = main_window.child_window(class_name="TcxTreeView")
         panel_TTabSheet = panel_TPage.child_window(class_name="TcxCustomInnerTreeView")
@@ -3638,6 +3720,9 @@ async def nf_devolucao_liquidar_cupom(num_nota_fiscal: str, data_atual: str) ->
 
         if radio_liquidado.is_checked():
             console.print("Botão 'Liquidado' está selecionado, seguindo com o processo...\n")
+            app = Application().connect(class_name="TFrmTituloReceber", timeout=60)
+            main_window = app["TFrmTituloReceber"]
+            main_window.set_focus()
             main_window.close()
         else:
             return RpaRetornoProcessoDTO(

{worker_automate_hub-0.5.53.dist-info → worker_automate_hub-0.5.54.dist-info}/METADATA

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

@@ -7,7 +7,7 @@ worker_automate_hub/api/helpers/api_helpers.py,sha256=SkheO2fXexeh-a4shr8Qzsz_kZ
 worker_automate_hub/api/rdp_service.py,sha256=L7orr60FkJr6zjETtA4me8tRYCW9m1g-i5pq6AILUFo,1647
 worker_automate_hub/api/rpa_fila_service.py,sha256=qa9JLuX6kL9Bv_PyPE_CzIikKwy0WdOf2FrIpFdMcYI,2125
 worker_automate_hub/api/rpa_historico_service.py,sha256=wQw8_k8i0S-r7qrrn4BSoztkgtiuH_bHRSEytbZebDg,9690
-worker_automate_hub/api/webhook_service.py,sha256=1VgkcOF0vY3cpV7-UOgrUj0qRuK9joDCvLhJkBBT0sQ,1751
+worker_automate_hub/api/webhook_service.py,sha256=S1a8IIS5LYY16I5E2pPPczaUnUz-_CwmD9_1QBdQD2g,1879
 worker_automate_hub/cli.py,sha256=JB45pjPJ8_E-4xw0OjqDMcAw-tpDV0mjmvwJRTnTzY0,6862
 worker_automate_hub/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/config/settings.py,sha256=4BkIYV0C9C_tl68WNfw3aoywr1-_bRJsKysSnSRALT0,2450
@@ -37,7 +37,7 @@ worker_automate_hub/tasks/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 worker_automate_hub/tasks/jobs/coleta_dje_process.py,sha256=UkLWTC5Ub2qBb0yY-8IZ0DLLOVJOfNTq_z9krx_t25Q,29476
 worker_automate_hub/tasks/jobs/conexao_rdp.py,sha256=lNjD-JxpPTXdvFEJAkajC_Ltyw81rIEJeu7FUfRf2qI,9586
 worker_automate_hub/tasks/jobs/descartes.py,sha256=0bMCJ0xIcK8TLSgjF6taAbGFLMU-sGDVFIhUYSRtrbg,41092
-worker_automate_hub/tasks/jobs/devolucao_prazo_a_faturar.py,sha256=H2pD9pkigmdcV94QUdYeAeIc_mENJb6t8hw0NCtBms0,77482
+worker_automate_hub/tasks/jobs/devolucao_prazo_a_faturar.py,sha256=PErDGphh75kc5SGJKI90L26qPJKl0BAkk8eRgjwmRvA,79240
 worker_automate_hub/tasks/jobs/ecac_estadual_go.py,sha256=dKkf22nH5gp3RErq5u0UzRsKyJ81fc6ZZ4vLtUuMwHA,21002
 worker_automate_hub/tasks/jobs/ecac_estadual_main.py,sha256=8WmKe4-MRtzHobXz2S4YBDNN8alfawkC-BBlRY-mn1g,1726
 worker_automate_hub/tasks/jobs/ecac_estadual_mt.py,sha256=C26zmpGQGUq6sP9lU9nanM3Fje-rkyx5tjwmRy4lyL8,25300
@@ -66,17 +66,17 @@ worker_automate_hub/tasks/jobs/playground.py,sha256=7vWDg9DwToHwGJ6_XOa8TQ6LmfRV
 worker_automate_hub/tasks/jobs/sped_fiscal.py,sha256=uHA5zQmAqqb-L_jdbBKiqINV4vXI58CL3wAI-2Xt_oQ,28938
 worker_automate_hub/tasks/jobs/transferencias.py,sha256=i3fCanHZ_pSwPW03THpQPqlzrDjbV4QbKkXpUqV3dxM,39272
 worker_automate_hub/tasks/task_definitions.py,sha256=yNu2GSC1PxjblcMgT7UOFhEUq4CMaQqDfN3hAF4yz_U,4954
-worker_automate_hub/tasks/task_executor.py,sha256=Hup_C2LeOkn7GEHQaEDqo8MIPdSPuaJ4hzHOOK7nqZU,5311
+worker_automate_hub/tasks/task_executor.py,sha256=1ZBVW2QS0io-gVyR4VNnbpaLJnSpTxtHIbRlnuQuj1Y,5566
 worker_automate_hub/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 worker_automate_hub/utils/env.py,sha256=TacQjGRO7PUNpttrhTAc5Gnegaiysl2Knsv1P8qfkfs,57
 worker_automate_hub/utils/get_creds_gworkspace.py,sha256=ZJ0IIEjM4IXIV9rwfbOZ1V1wiaMoJAGZeSy0D37sYdU,2212
 worker_automate_hub/utils/logger.py,sha256=FYV9fg0_RAYJF_ZOCJEbqQAiCXlXk2gMpvUU1rzT_xs,671
 worker_automate_hub/utils/toast.py,sha256=xPHc5r5uOxB_cZlCzm13Kt2qSKLLFZALncU6Qg3Ft68,1162
 worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbVlwDl49Y,7972
-worker_automate_hub/utils/util.py,sha256=xBvFZet8PBxLEm97SFGHXjG3evWuUPpm1Q7gQxqeeUQ,155619
+worker_automate_hub/utils/util.py,sha256=ch8qDZfKBHCYKKjblVSQ4X07m9AsYpOPD5RflaipDm0,159514
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=iYpOs7fb7C3bY61QHySr00xlDNzrrCP0zaexOpBPuaQ,31930
 worker_automate_hub/worker.py,sha256=CT-poyP1ZYvubArYsnnNd9oJ53SPaDwgr6p6keS3nI4,6248
-worker_automate_hub-0.5.53.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.5.53.dist-info/METADATA,sha256=bFF74h-BGAk3vZP_sHvwOBTB69WgkzQTd1UrhOe6leY,2894
-worker_automate_hub-0.5.53.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-worker_automate_hub-0.5.53.dist-info/RECORD,,
+worker_automate_hub-0.5.54.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.5.54.dist-info/METADATA,sha256=MpWKhIzCV_6rKtJD-Vfn7LYwGeqXpIfixLPCPDN-vjo,2894
+worker_automate_hub-0.5.54.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+worker_automate_hub-0.5.54.dist-info/RECORD,,