worker-automate-hub 0.4.403__tar.gz → 0.4.405__tar.gz

{worker_automate_hub-0.4.403 → worker_automate_hub-0.4.405}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: worker-automate-hub
-Version: 0.4.403
+Version: 0.4.405
 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.4.403 → worker_automate_hub-0.4.405}/pyproject.toml

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

@@ -400,7 +400,7 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                     send_keys("{LEFT 3}")
 
                     distribuiu_algo = False
-
+                    distribuicao_atual = []
                     last_line_almoxarifado_emsys = 'x'
                     max_distribuicao = 0
 
@@ -448,6 +448,7 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                                     pyautogui.write(quantidade_combustivel)
                                     pyautogui.press('enter')
                                     list_tanques_distribuidos.append(second_info_distribuicao_obs)
+                                    distribuicao_atual.append(f"Valor do tipo de combustivel:{codigo_almoxarifado_emsys} nas observações é {quantidade_combustivel}")
                                     distribuiu_algo = True
 
                         max_distribuicao = max_distribuicao + 1
@@ -458,7 +459,70 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                     console.print(f"Index Tanque: {index_tanque}")
 
                     if distribuiu_algo:
+                        console.print(f"Extraindo informação da Janela de Distribuir Itens para realização do OCR...\n")
+
+                        main_window.set_focus()
+                        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\\distribuir_iten_{nota.get("nfe")}.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"
+                        )
+                        os.remove(path_to_png)
+
+                        pattern_qtd_restante = r"Quantidade restante:\s([\d,]+)"
+                        resultado = re.search(pattern_qtd_restante, captured_text)
+
+                        try:
+                            if resultado:
+                                quantidade_restante = int(resultado.group(1).replace(",", ""))
+                            
+                                if quantidade_restante > 0:
+                                    return RpaRetornoProcessoDTO(
+                                        sucesso=False,
+                                        retorno=f"A distribuição informada diverge do que está na nota, {', '.join(distribuicao_atual)} e o valor da quantidade restante é de {str(quantidade_restante)}.",
+                                        status=RpaHistoricoStatusEnum.Falha,
+                                    )
+                                else:
+                                    console.print(f"A quantidade restante é igual ou menor que 0, seguindo... \n")
+                            else:
+                                return RpaRetornoProcessoDTO(
+                                    sucesso=False,
+                                    retorno=f"Não foi possivel obter o resultado da quantidade restante, texto extraido: {resultado}",
+                                    status=RpaHistoricoStatusEnum.Falha,
+                                )
+                        except Exception as e:
+                            return RpaRetornoProcessoDTO(
+                                sucesso=False,
+                                retorno=f"Erro {e}, Não foi possivel obter o resultado da quantidade restante, texto extraido: {resultado}",
+                                status=RpaHistoricoStatusEnum.Falha,
+                            )
+
                         console.print(f"Algum Item foi distribuido, clicando em OK para salvar")
+                        main_window.set_focus()
+                        await worker_sleep(1)
                         btn_ok = main_window.child_window(
                                 class_name="TBitBtn", found_index=1
                             )

{worker_automate_hub-0.4.403 → worker_automate_hub-0.4.405}/worker_automate_hub/tasks/jobs/transferencias.py

@@ -613,10 +613,10 @@ async def transferencias(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
             app = Application().connect(title="Informação")
             dialog = app.window(title="Informação")
 
-            btn_ok = dialog.child_window(title="OK", class_name="Button")
-            if btn_ok.exists():
+            btn_yes = dialog.child_window(title="OK", class_name="Button")
+            if btn_yes.exists():
                 try:
-                    btn_ok.click()
+                    btn_yes.click()
                     await worker_sleep(3)
                     console.print(
                         "O botão OK de pré-venda incluída foi clicado com sucesso.",
@@ -632,7 +632,6 @@ async def transferencias(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                         retorno=f"Falha ao clicar no botão OK de pré-venda incluída: {e}",
                         status=RpaHistoricoStatusEnum.Falha,
                     )
-
             else:
 
                 console.print(
@@ -656,22 +655,66 @@ async def transferencias(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
 
         await worker_sleep(3)
 
-        screenshot_path = take_screenshot()
+        # screenshot_path = take_screenshot()
         # Message 'Deseja pesquisar pré-venda?'
         console.print(
             "Verificando a existencia da mensagem: 'Deseja pesquisar pré-venda?'...\n"
         )
-        message_prevenda = take_target_position(screenshot_path, "Deseja")
-        if message_prevenda is not None:
-            button_yes = find_target_position(screenshot_path, "Yes", attempts=5)
-            if button_yes is not None:
-                pyautogui.click(button_yes)
+        try:
+            app = Application().connect(title="Confirm")
+            dialog_prevenda = app.window(title="Confirm")
+
+            btn_yes = dialog_prevenda.child_window(title="&Yes", class_name="TButton")
+            if btn_yes.exists():
+                try:
+                    btn_yes.click()
+                    await worker_sleep(3)
+                    console.print(
+                        "O botão YES de pesquisa pré-venda incluída foi clicado com sucesso.",
+                        style="green",
+                    )
+                except Exception as e:
+                    console.print(
+                        f"Falha ao clicar no botão YES de pesquisa pré-venda incluída: {e}",
+                        style="red",
+                    )
+                    return RpaRetornoProcessoDTO(
+                        sucesso=False,
+                        retorno=f"Falha ao clicar no botão YES de pesquisa pré-venda incluída: {e}",
+                        status=RpaHistoricoStatusEnum.Falha,
+                    )
             else:
-                pyautogui.click(914,560)
-                console.log('Clicou em "Yes" para confirmar pré-venda', style="bold green")
-        else:
-            log_msg = f"Mensagem 'Deseja pesquisar pré-venda?' não encontrada."
-            console.print(log_msg, style="bold yellow")
+
+                console.print(
+                    "O Botão OK de pré-venda incluída não foi encontrado.", style="red"
+                )
+                return RpaRetornoProcessoDTO(
+                    sucesso=False,
+                    retorno="O Botão YES de pesquisa pré-venda incluída não foi encontrado.",
+                    status=RpaHistoricoStatusEnum.Falha,
+                )
+
+        except Exception as e:
+            console.print(
+                f"O Botão YES de pesquisa pré-venda incluída não foi encontrado: {e}", style="red"
+            )
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno=f"O Botão YES de pesquisa pré-venda incluída não foi encontrado: {e}",
+                status=RpaHistoricoStatusEnum.Falha,
+            )
+
+        # message_prevenda = take_target_position(screenshot_path, "Deseja")
+        # if message_prevenda is not None:
+        #     button_yes = find_target_position(screenshot_path, "Yes", attempts=5)
+        #     if button_yes is not None:
+        #         pyautogui.click(button_yes)
+        #     else:
+        #         pyautogui.click(914,560)
+        #         console.log('Clicou em "Yes" para confirmar pré-venda', style="bold green")
+        # else:
+        #     log_msg = f"Mensagem 'Deseja pesquisar pré-venda?' não encontrada."
+        #     console.print(log_msg, style="bold yellow")
         await worker_sleep(5)
 
         # screenshot_path = take_screenshot()
@@ -767,7 +810,7 @@ async def transferencias(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
         pyautogui.click(pre_venda_sucesso)
         console.log('Clicou em "OK" na mensagem de sucesso', style="bold green")
 
-        await worker_sleep(5)
+        await worker_sleep(10)
 
         # Faturando Pre-venda
         retorno = await faturar_pre_venda(task)
@@ -856,5 +899,4 @@ async def transferencias(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
             sucesso=False,
             retorno=log_msg,
             status=RpaHistoricoStatusEnum.Falha,
-        )
-
+        )