worker-automate-hub 0.5.52__tar.gz → 0.5.54__tar.gz

{worker_automate_hub-0.5.52 → worker_automate_hub-0.5.54}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.52"
+version = "0.5.54"
 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.52 → worker_automate_hub-0.5.54}/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-0.5.52 → worker_automate_hub-0.5.54}/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,
@@ -210,6 +211,10 @@ async def devolucao_prazo_a_faturar(task: RpaProcessoEntradaDTO) -> RpaRetornoPr
                 btn_ir_para_nota = pyautogui.locateOnScreen(ASSETS_PATH + "\\notas_saida\\ir_para_nota_a_fatura_esmaecido.png", confidence=0.8)
                 if btn_ir_para_nota:
                     console.print("Botão 'Ir para nota de faturar' inativo, seguindo com o processo...\n")
+                    app = Application().connect(class_name="TFrmNotaFiscalSaida", timeout=60)
+                    main_window = app["TFrmNotaFiscalSaida"]
+                    main_window.set_focus()
+                    main_window.close()
                 else:
                     retorno = f"Botão [Ir para nota de faturar] está ativo, impossibilitando realizar a devolução \nEtapas Executadas:\n{steps}"
                     return RpaRetornoProcessoDTO(
@@ -754,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)
@@ -765,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)
@@ -815,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-0.5.52 → worker_automate_hub-0.5.54}/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-0.5.52 → worker_automate_hub-0.5.54}/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)
@@ -3561,7 +3615,7 @@ async def nf_devolucao_liquidar_cupom(num_nota_fiscal: str, data_atual: str) ->
 
         main_window.set_focus()
 
-        page_control = main_window.child_window(class_name="TPageControl")
+        page_control = main_window.child_window(class_name="TPageControl", found_index=0)
         # Navegar até a aba TabSheetLiquidacao
         tab_sheet_liquidacao = page_control.child_window(title="TabSheetLiquidacao")
         panel_liquidacao = tab_sheet_liquidacao.child_window(class_name="TPanel", found_index=0)
@@ -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(