worker-automate-hub 0.4.403__py3-none-any.whl → 0.4.405__py3-none-any.whl

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/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,
-        )
-
+        )

{worker_automate_hub-0.4.403.dist-info → worker_automate_hub-0.4.405.dist-info}/METADATA

@@ -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.dist-info → worker_automate_hub-0.4.405.dist-info}/RECORD

@@ -48,14 +48,14 @@ worker_automate_hub/tasks/jobs/entrada_de_notas_39.py,sha256=YMXbcQ-IXimJzH8npsG
 worker_automate_hub/tasks/jobs/entrada_de_notas_500.py,sha256=ryl5AZ2617aROSVcS-N5wQIBlFQ1Bp0khUH9fP4DOzU,26542
 worker_automate_hub/tasks/jobs/entrada_de_notas_505.py,sha256=jIml8gjXPdI6_x7S9VVV8IrKZRF7_PTNOMnhNmYMDTU,14490
 worker_automate_hub/tasks/jobs/entrada_de_notas_7139.py,sha256=8XP9G3n0PUeshbRWkWKOMnyUGRWspIolPZVqQTR3SMI,14184
-worker_automate_hub/tasks/jobs/entrada_de_notas_9.py,sha256=eAFBeEkhf7dLnj-vW7sEnxcBr1BJ6FkvIofNKaJcnyg,43877
+worker_automate_hub/tasks/jobs/entrada_de_notas_9.py,sha256=OPOE43iMTUg3uNf2ZxqENV_NWn-g7HXBavHHgPN7RoA,47796
 worker_automate_hub/tasks/jobs/exemplo_processo.py,sha256=3-zxbb-9YHPmSA_K1Qgxp_FwSqg2QDjGBRCLxDZ8QoQ,3451
 worker_automate_hub/tasks/jobs/fidc_gerar_nosso_numero.py,sha256=Y-mKYcgX1kwcy_sOHvctn7clXnY3O3fohvOs7tKPMkM,9484
 worker_automate_hub/tasks/jobs/fidc_remessa_cobranca_cnab240.py,sha256=QBGm6eS5JghgNWNqZlk1g2a2iV8LnBLiOTBBL3Giet0,4181
 worker_automate_hub/tasks/jobs/login_emsys.py,sha256=IoGCIvO4UwmuxOZEn3cvYJlKyhsWvtHvbFk8vwjTroQ,5620
 worker_automate_hub/tasks/jobs/playground.py,sha256=bdnXv3C7WLQUxt4edGZDfAbRJJ2-q4zuIQaK3GLnaUc,1765
 worker_automate_hub/tasks/jobs/sped_fiscal.py,sha256=KquAdk26kvIKqT5hkHvzKMpa1UzJphx6s7qDv9pVuH4,26243
-worker_automate_hub/tasks/jobs/transferencias.py,sha256=keCUlvbPhLrTiZQ9Co_r6IqL5wjYqVGB_-mgvdDe9pM,36080
+worker_automate_hub/tasks/jobs/transferencias.py,sha256=GM8OjjIgxu9biXYyQVnW3c-af06tdIPqy3s9QH9UXSs,38023
 worker_automate_hub/tasks/task_definitions.py,sha256=2Jp1H4_qJZqqGyaP6MA87KLt4QNrtWBYWbXu-2gymFo,4459
 worker_automate_hub/tasks/task_executor.py,sha256=eXFgWbcM8aMPwENvNix8KyFv7BqvIwhXUDkBSI1ul7M,8873
 worker_automate_hub/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -67,7 +67,7 @@ worker_automate_hub/utils/updater.py,sha256=en2FCGhI8aZ-JNP3LQm64NJDc4awCNW7UhbV
 worker_automate_hub/utils/util.py,sha256=YCbgMMBwhp9LD6NUmE5S2K6S12TNEpGJm25FqfsIKPE,119544
 worker_automate_hub/utils/utils_nfe_entrada.py,sha256=yQFyDzvXPSO-UYV1m38WLJ3AWQgZYcyUkHZdCKE1jG4,28360
 worker_automate_hub/worker.py,sha256=KDBU3L2kVobndrnN5coRZFTwVmBLKmPJjRv20sCo5Hc,4697
-worker_automate_hub-0.4.403.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
-worker_automate_hub-0.4.403.dist-info/METADATA,sha256=IN1TuqG_rz_vigeyb-kIbf32zEusW-lXw_U3D7lFKwk,2895
-worker_automate_hub-0.4.403.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-worker_automate_hub-0.4.403.dist-info/RECORD,,
+worker_automate_hub-0.4.405.dist-info/entry_points.txt,sha256=sddyhjx57I08RY8X7UxcTpdoOsWULAWNKN9Xr6pp_Kw,54
+worker_automate_hub-0.4.405.dist-info/METADATA,sha256=JxHUFknJopcBtPIJogmyAzwxlLNoOtbmoI4kr1RZkNs,2895
+worker_automate_hub-0.4.405.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+worker_automate_hub-0.4.405.dist-info/RECORD,,