worker-automate-hub 0.4.395__tar.gz → 0.4.397__tar.gz

{worker_automate_hub-0.4.395 → worker_automate_hub-0.4.397}/PKG-INFO

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

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

@@ -3,18 +3,22 @@ import warnings
 import os
 import io
 import uuid
-from datetime import datetime
+from datetime import datetime,date
+import calendar
 
 import pyautogui
+import pyperclip
 import pytesseract
 from pywinauto.application import Application
+from pywinauto.mouse import double_click
+from pywinauto.keyboard import send_keys
+import win32clipboard
 from PIL import Image, ImageEnhance
 from rich.console import Console
 
 from worker_automate_hub.api.client import (
     get_config_by_name,
     send_file,
-    get_historico_by_processo_identificador,
 )
 
 from worker_automate_hub.models.dto.rpa_historico_request_dto import (
@@ -49,6 +53,25 @@ async def sped_fiscal(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
         # Get config from BOF
         config = await get_config_by_name("login_emsys")
         console.print(task)
+        try:
+            get_config_gerar_inventario = await get_config_by_name("SPED_gerar_inventario")
+            console.print(get_config_gerar_inventario)
+            con_config = get_config_gerar_inventario.get("conConfiguracao")
+            if con_config is not None:
+                gerar_inventario = con_config.get("gerar")
+            else:
+                return RpaRetornoProcessoDTO(
+                    sucesso=False,
+                    retorno="Não foi possivel recuperar o valor da configuração Gerar Inventario, não sendo possivel seguir com o processo.",
+                    status=RpaHistoricoStatusEnum.Falha,
+                )
+        except Exception as e:
+            return RpaRetornoProcessoDTO(
+                sucesso=False,
+                retorno=f"Não foi possivel recuperar o valor da configuração Gerar Inventario, erro: {e}.",
+                status=RpaHistoricoStatusEnum.Falha,
+            )
+
 
         # Seta config entrada na var sped_processar para melhor entendimento
         sped_processar = task.configEntrada
@@ -210,9 +233,114 @@ async def sped_fiscal(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                 retorno=f"Erro ao interagir com a tela de gerar sped fiscal, erro: {e}",
                 status=RpaHistoricoStatusEnum.Falha,
             )
+
+        ano_atual = datetime.now().year
+        if str(periodo_dt) == f"02/{ano_atual}":
+            console.print("Filtrando período pelo Inventario...\n")
+            ano_passado = ano_atual - 1
+            data_busca = f"31/12/{ano_passado}"
             
+            periodo_filtro_inicio = main_window.child_window(class_name="TEditDate", found_index=1)
+            periodo_filtro_inicio.set_edit_text(data_busca)
+            await worker_sleep(2)
+            periodo_filtro_fim = main_window.child_window(class_name="TEditDate", found_index=0)
+            periodo_filtro_fim.set_edit_text(data_busca)
+            await worker_sleep(2)
+            periodo_btn_buscar = main_window.child_window(class_name="TBitBtn", found_index=0)
+            periodo_btn_buscar.click()
+            await worker_sleep(2)
+            periodo_btn_buscar = main_window.child_window(class_name="TcxGrid", found_index=0)
+            rect = periodo_btn_buscar.rectangle()
+            center_x = (rect.left + rect.right) // 2
+            center_y = (rect.top + rect.bottom) // 2
+            double_click(coords=(center_x, center_y))
+        elif gerar_inventario:
+            mes_atual = datetime.now().month
+            if mes_atual == 1:
+                mes_anterior = 12
+                ano_anterior = ano_atual - 1
+            else:
+                mes_anterior = mes_atual - 1
+                ano_anterior = ano_atual
+
+            ultimo_dia = calendar.monthrange(ano_anterior, mes_anterior)[1]
+            data_ultimo_dia = date(ano_anterior, mes_anterior, ultimo_dia)
+            data_formatada = data_ultimo_dia.strftime("%d/%m/%Y")
+
+            periodo_filtro_inicio = main_window.child_window(class_name="TEditDate", found_index=1)
+            periodo_filtro_inicio.set_edit_text(data_formatada)
+            await worker_sleep(2)
+            periodo_filtro_fim = main_window.child_window(class_name="TEditDate", found_index=0)
+            periodo_filtro_fim.set_edit_text(data_formatada)
+            await worker_sleep(2)
+            periodo_btn_buscar = main_window.child_window(class_name="TBitBtn", found_index=0)
+            periodo_btn_buscar.click()
+            await worker_sleep(2)
+            grid_inventario = main_window.child_window(class_name="TcxGrid", found_index=0)
+            rect = grid_inventario.rectangle()
+            center_x = (rect.left + rect.right) // 2
+            center_y = (rect.top + rect.bottom) // 2
+            grid_inventario.click_input(coords=(center_x, center_y))
+            await worker_sleep(2)
+            send_keys("^({HOME})")
+            pyperclip.copy('')
+            await worker_sleep(3)
 
-        await worker_sleep(2)
+            while True:
+                last_line_inventario_emsys_fiscal = 'x'
+                await worker_sleep(1)
+                with pyautogui.hold('ctrl'):
+                    pyautogui.press('c')
+
+                await worker_sleep(1)
+
+                with pyautogui.hold('ctrl'):
+                    pyautogui.press('c')
+
+                win32clipboard.OpenClipboard()
+                line_inventario_emsys_fiscal = win32clipboard.GetClipboardData().strip()
+                win32clipboard.CloseClipboard()
+                console.print(f"Linha atual copiada do Emsys Fiscal: {line_inventario_emsys_fiscal}\nUltima Linha copiada: {last_line_inventario_emsys_fiscal}")
+
+                if len(line_inventario_emsys_fiscal) <= 4:
+                    return RpaRetornoProcessoDTO(
+                        sucesso=False,
+                        retorno=f"Erro: Nenhum registro encontrado com a data informada {periodo_btn_buscar}.",
+                        status=RpaHistoricoStatusEnum.Falha,
+                    )
+
+
+                if 'no final do' in line_inventario_emsys_fiscal.lower():
+                    selecionar_registro_full_path = "assets\\emsys\\button_selecionar_registro_fiscal.png"
+                    try:
+                        button_location = pyautogui.locateCenterOnScreen(selecionar_registro_full_path, confidence=0.6)
+                        if button_location:
+                            pyautogui.click(button_location)
+                            console.print("Botão 'Selecionar Registro' clicado com sucesso!")
+                            break
+                    except Exception as e:
+                        return RpaRetornoProcessoDTO(
+                            sucesso=False,
+                            retorno=f"Erro: Não foi clicar na opção Selecionar Registro no Inventario, erro: {e}.",
+                            status=RpaHistoricoStatusEnum.Falha,
+                        )
+
+                
+                if bool(line_inventario_emsys_fiscal):
+                    if last_line_inventario_emsys_fiscal == line_inventario_emsys_fiscal:
+                        return RpaRetornoProcessoDTO(
+                            sucesso=False,
+                            retorno=f"Erro: Não foi encontrado a opção do período de busca {periodo_filtro_inicio} nas opções do inventário.",
+                            status=RpaHistoricoStatusEnum.Falha,
+                        )
+                    else:
+                        last_line_inventario_emsys_fiscal = line_inventario_emsys_fiscal
+                        pyautogui.press('down')
+
+        else:
+            console.print("Mês não é igual a Fevereiro e SPED Inventario é falso, seguindo com a geração do SPED...\n")
+        
+        await worker_sleep(3)
         pesquisar_full_path = "assets\\emsys\\button_gerar_sped_fiscal.png"
         try:
             button_location = pyautogui.locateCenterOnScreen(pesquisar_full_path, confidence=0.6)