worker-automate-hub 0.4.351__tar.gz → 0.4.352__tar.gz

{worker_automate_hub-0.4.351 → worker_automate_hub-0.4.352}/PKG-INFO

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

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

@@ -1,7 +1,12 @@
 import asyncio
+import getpass
+import os
 import warnings
 
 import pyautogui
+from pywinauto.keyboard import send_keys
+from PIL import Image, ImageEnhance
+import pytesseract
 from pywinauto.application import Application
 from pywinauto_recorder.player import set_combobox
 from rich.console import Console
@@ -332,6 +337,83 @@ async def entrada_de_notas_15(task: RpaProcessoEntradaDTO) -> RpaRetornoProcesso
                     status=RpaHistoricoStatusEnum.Falha,
                 )
 
+        try:
+            max_attempts = 60
+            i = 0
+
+            while i < max_attempts:
+                information_pop_up = await is_window_open("Information")
+                if information_pop_up["IsOpened"] == True:
+                    break
+                else:
+                    console.print(f"Aguardando confirmação de nota incluida...\n")
+                    await worker_sleep(5)
+                    i += 1
+
+            information_pop_up = await is_window_open("Information")
+            if information_pop_up["IsOpened"] == True:
+                app = Application().connect(class_name="TFrmNotaFiscalEntrada")
+                main_window = app["Information"]
+
+                main_window.set_focus()
+
+                console.print(f"Obtendo texto do Information...\n")
+                console.print(f"Tirando print da janela do Information 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\\information_popup_{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)
+                if 'nota fiscal inc' in captured_text.lower():
+                    console.print(f"Tentando clicar no Botão OK...\n")
+                    btn_ok = main_window.child_window(class_name="TButton")
+
+                    if btn_ok.exists():
+                        btn_ok.click()
+                        retorno = True
+                else:
+                    return RpaRetornoProcessoDTO(
+                        sucesso=False,
+                        retorno=f"Pop_up Informantion não mapeado para andamento do robô, mensagem {captured_text}",
+                        status=RpaHistoricoStatusEnum.Falha,
+                    )
+            else:
+                console.print(f"Aba Information não encontrada")
+                retorno = await verify_nf_incuded()
+
+        except Exception as e:
+            console.print(f"Erro ao conectar à janela Information: {e}\n")
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno=f"Erro em obter o retorno, Nota inserida com sucesso, erro {e}",
+                status=RpaHistoricoStatusEnum.Falha,
+            )
+
         if retorno:
             console.print("\nNota lançada com sucesso...", style="bold green")
             await worker_sleep(6)

{worker_automate_hub-0.4.351 → worker_automate_hub-0.4.352}/worker_automate_hub/tasks/jobs/entrada_de_notas_16.py

@@ -1,7 +1,12 @@
 import asyncio
+import getpass
+import os
 import warnings
 
 import pyautogui
+from pywinauto.keyboard import send_keys
+from PIL import Image, ImageEnhance
+import pytesseract
 from pywinauto.application import Application
 from pywinauto_recorder.player import set_combobox
 from rich.console import Console
@@ -429,6 +434,82 @@ async def entrada_de_notas_16(task: RpaProcessoEntradaDTO) -> RpaRetornoProcesso
                     retorno=observacao,
                     status=RpaHistoricoStatusEnum.Falha,
                 )
+        try:
+            max_attempts = 60
+            i = 0
+
+            while i < max_attempts:
+                information_pop_up = await is_window_open("Information")
+                if information_pop_up["IsOpened"] == True:
+                    break
+                else:
+                    console.print(f"Aguardando confirmação de nota incluida...\n")
+                    await worker_sleep(5)
+                    i += 1
+
+            information_pop_up = await is_window_open("Information")
+            if information_pop_up["IsOpened"] == True:
+                app = Application().connect(class_name="TFrmNotaFiscalEntrada")
+                main_window = app["Information"]
+
+                main_window.set_focus()
+
+                console.print(f"Obtendo texto do Information...\n")
+                console.print(f"Tirando print da janela do Information 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\\information_popup_{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)
+                if 'nota fiscal inc' in captured_text.lower():
+                    console.print(f"Tentando clicar no Botão OK...\n")
+                    btn_ok = main_window.child_window(class_name="TButton")
+
+                    if btn_ok.exists():
+                        btn_ok.click()
+                        retorno = True
+                else:
+                    return RpaRetornoProcessoDTO(
+                        sucesso=False,
+                        retorno=f"Pop_up Informantion não mapeado para andamento do robô, mensagem {captured_text}",
+                        status=RpaHistoricoStatusEnum.Falha,
+                    )
+            else:
+                console.print(f"Aba Information não encontrada")
+                retorno = await verify_nf_incuded()
+
+        except Exception as e:
+            console.print(f"Erro ao conectar à janela Information: {e}\n")
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno=f"Erro em obter o retorno, Nota inserida com sucesso, erro {e}",
+                status=RpaHistoricoStatusEnum.Falha,
+            )
 
         if retorno:
             console.print("\nNota lançada com sucesso...", style="bold green")

{worker_automate_hub-0.4.351 → worker_automate_hub-0.4.352}/worker_automate_hub/tasks/jobs/sped_fiscal.py

@@ -139,7 +139,7 @@ async def sped_fiscal(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
             await worker_sleep(2)
 
             console.print("Inserindo o período para geração do sped...\n")
-            periodo = panel_TTabSheet.child_window(class_name="TDBIEditDate", found_index=2)
+            periodo = main_window.child_window(class_name="TDBIEditDate", found_index=0)
             periodo.set_edit_text(periodo_dt)
             await worker_sleep(2)
 
@@ -215,6 +215,7 @@ async def sped_fiscal(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                 status=RpaHistoricoStatusEnum.Falha,
             )
         
+
         console.print(f"Verificando sem possui o pop-up de Registro de Entrada... \n")
         registro_entrada_pop_up_class = await is_window_open_by_class("TMsgBox")
         if registro_entrada_pop_up_class == True: