worker-automate-hub 0.4.415__tar.gz → 0.4.416__tar.gz

{worker_automate_hub-0.4.415 → worker_automate_hub-0.4.416}/PKG-INFO

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

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

@@ -7,7 +7,7 @@ import pyautogui
 import pytesseract
 from datetime import datetime, timedelta
 from pywinauto.application import Application
-from PIL import Image, ImageEnhance
+from PIL import Image, ImageEnhance, ImageFilter
 from pywinauto.keyboard import send_keys
 import win32clipboard
 from pywinauto_recorder.player import set_combobox
@@ -461,39 +461,50 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                     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(),
+                        max_attempts = 3
+                        attempts = 0
+                        resultado = None
+                        while attempts < max_attempts:
+                            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")
+                            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)
+                            console.print(
+                                f"Preparando a imagem para maior resolução e assertividade no OCR...\n"
+                            )
+                            image = Image.open(path_to_png)
+                            image = image.filter(ImageFilter.SHARPEN)
+                            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")
+                            await worker_sleep(1)
+                            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)
+                            pattern_qtd_restante = r"Quantidade restante:\s([\d,]+)"
+                            resultado = re.search(pattern_qtd_restante, captured_text)
+                            if resultado:
+                                break
+                            else:
+                                await worker_sleep(2)
+                                attempts += 1
 
                         try:
                             if resultado: