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

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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.4.404"
+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.404 → 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
                             )