worker-automate-hub 0.5.653__tar.gz → 0.5.655__tar.gz

{worker_automate_hub-0.5.653 → worker_automate_hub-0.5.655}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worker-automate-hub
-Version: 0.5.653
+Version: 0.5.655
 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.5.653 → worker_automate_hub-0.5.655}/pyproject.toml

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

@@ -218,7 +218,55 @@ async def lancamento_pis_cofins(task: RpaProcessoEntradaDTO) -> RpaRetornoProces
             except:
                 btn_ok = main_window.child_window(title="&OK")
                 btn_ok.click()
+            
+            await worker_sleep(3)
 
+            try:
+                selecionar_todos_itens = (
+                    ASSETS_PATH + "\\lancamento_pis_cofins\\botao_incluir.png"
+                )
+                # Tenta localizar a imagem na tela
+                localizacao = pyautogui.locateOnScreen(
+                    selecionar_todos_itens, confidence=0.9
+                )
+                await worker_sleep(3)
+                if localizacao:
+                    centro = pyautogui.center(localizacao)
+                    pyautogui.moveTo(centro)
+                    pyautogui.click()
+                    console.print("Clique realizado com sucesso!")
+                else:
+                    console.print("Imagem não encontrada na tela.")
+            except Exception as e:
+                retorno = f"Não foi possivel clicar em selecionar todos os itens na Seleção de Empresas, erro: {e} "
+                return RpaRetornoProcessoDTO(
+                    sucesso=False,
+                    retorno=retorno,
+                    status=RpaHistoricoStatusEnum.Falha,
+                    tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
+                )
+            
+            # Verificar se já existe registro no mês apurado
+            try:
+                selecionar_todos_itens = (
+                    ASSETS_PATH + "\\lancamento_pis_cofins\\ja_existe_periodo.png"
+                )
+                # Tenta localizar a imagem na tela
+                localizacao = pyautogui.locateOnScreen(
+                    selecionar_todos_itens, confidence=0.9
+                )
+                await worker_sleep(3)
+                if localizacao:
+                    app = Application().connect(title="Informação", timeout=60)
+                    main_window = app["Informação"]
+                    main_window.set_focus()
+                    informacao =  main_window.child_window(title="OK")
+                    informacao.click()
+                    console.print("Clique realizado com sucesso!")
+                else:
+                    console.print("Imagem não encontrada na tela.")
+            except:
+                pass
             await worker_sleep(5)
             console.print("Verificando se possui tela de Informação... \n")
             information_pop_up = await is_window_open("Information")
@@ -291,6 +339,7 @@ async def lancamento_pis_cofins(task: RpaProcessoEntradaDTO) -> RpaRetornoProces
                 )
                 main_window = app["TFrmMovtoApuraPisCofins"]
                 main_window.close()
+                await worker_sleep(2)
                 console.print("Tela de Movimento Apuração, fechada com sucesso... \n")
         else:
             return RpaRetornoProcessoDTO(
@@ -311,7 +360,7 @@ async def lancamento_pis_cofins(task: RpaProcessoEntradaDTO) -> RpaRetornoProces
             pyautogui.press("enter")
             await worker_sleep(2)
             pyautogui.press("enter")
-            await worker_sleep(4)
+            await worker_sleep(5)
             console.print(
                 f"\nPesquisa: 'Otimizador Cálculo PIS/COFINS' realizada com sucesso",
                 style="bold green",