worker-automate-hub 0.5.820__py3-none-any.whl → 0.5.912__py3-none-any.whl

worker_automate_hub/api/client.py

@@ -2,12 +2,13 @@ from decimal import ROUND_HALF_UP, Decimal
 import threading
 from typing import Optional
 import aiohttp
-
+import re
+from collections import defaultdict
 import aiohttp
 import requests
 from aiohttp import ClientSession
 from rich.console import Console
-
+from typing import List, Dict, Any
 from worker_automate_hub.api.helpers.api_helpers import handle_api_response
 from worker_automate_hub.config.settings import load_env_config
 from worker_automate_hub.models.dao.rpa_configuracao import RpaConfiguracao
@@ -498,6 +499,89 @@ async def get_valor_remessa_cobranca(date: str):
         logger.info(err_msg)
 
 
+async def get_notas_produtos(codFornecedor: int, codEmpresa: int, itens: List[Dict[str, Any]]) -> Dict[str, Any]:
+    """
+    Função única:
+      - Converte itens (aceita chaves codigo/codigoProduto e quantidade/qtd)
+      - Chama /nf-supplier/checker com Authorization: Basic
+      - Normaliza tipos do retorno
+      - Entrega:
+          {
+            "lista": [ {codItem, qtdTotal, notas, valorUnitario}, ... ],
+            "por_codigo": { codItem: {...}, ... }
+          }
+
+    Retorno esperado da API (ex):
+      [
+        {"codItem": 19969, "qtdTotal": 15, "notas": ["1418727","1410744"], "valorUnitario": 5.29},
+        {"codItem": 29272, "qtdTotal": 10, "notas": ["1418727"], "valorUnitario": 7.12}
+      ]
+    """
+    # --- Carrega config
+    env_config, _ = load_env_config()
+    url_base = env_config["API_BASE_URL"].rstrip("/")
+    url = f"{url_base}/nf-supplier/checker"
+
+    # --- Header Basic (aceita token puro ou já "Basic ...")
+    token = (env_config.get("API_AUTHORIZATION") or "").strip()
+    auth_header = token if token.lower().startswith("basic ") else f"Basic {token}"
+
+    # --- Converte itens de entrada
+    itens_convertidos: List[Dict[str, int]] = []
+    for it in itens or []:
+        codigo = re.findall(r"\d+", it.get("descricaoProduto"))[0]
+        quantidade = it.get("quantidade", it.get("qtd"))
+        if codigo is None or quantidade is None:
+            logger.warning(f"Item incompleto: {it}")
+            console.print(f"⚠️ Item incompleto: {it}", style="yellow")
+            continue
+        try:
+            itens_convertidos.append({"codigo": int(codigo), "quantidade": int(quantidade)})
+        except Exception:
+            logger.warning(f"Item inválido (não numérico): {it}")
+            console.print(f"⚠️ Item inválido (não numérico): {it}", style="yellow")
+
+    body = {
+        "codFornecedor": int(codFornecedor),
+        "codEmpresa": int(codEmpresa),
+        "itens": itens_convertidos,
+    }
+    headers = {"Authorization": auth_header, "Content-Type": "application/json"}
+
+    # --- Chamada HTTP
+    async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(verify_ssl=False)) as session:
+        async with session.post(url, json=body, headers=headers) as resp:
+            text = await resp.text()
+            if resp.status != 200:
+                raise RuntimeError(f"HTTP {resp.status} ao chamar {url}: {text}")
+            try:
+                data = await resp.json()
+            except Exception:
+                raise RuntimeError(f"Resposta não-JSON do servidor: {text[:600]}")
+
+    console.print(f"✅ Resposta da API: {data}", style="bold green")
+    logger.info(f"nf-supplier/checker -> {data}")
+
+    if not isinstance(data, list):
+        raise ValueError(f"Formato inesperado da API: {type(data)} -> {data}")
+
+    # --- Normaliza tipos e monta índices
+    lista_norm: List[Dict[str, Any]] = []
+    por_codigo: Dict[int, Dict[str, Any]] = {}
+
+    for row in data:
+        cod = int(row.get("codItem"))
+        item_norm = {
+            "codItem": cod,
+            "qtdTotal": int(row.get("qtdTotal", 0)),
+            "notas": [str(n) for n in (row.get("notas") or [])],
+            "valorUnitario": float(row.get("valorUnitario", 0.0)),
+        }
+        lista_norm.append(item_norm)
+        por_codigo[cod] = item_norm
+
+    return {"lista": lista_norm, "por_codigo": por_codigo}
+
 async def get_status_nf_emsys(chave: int):
     """
     Procura o status de nota fiscal no EMSYS.
@@ -538,7 +622,7 @@ async def get_dados_nf_emsys(
     chave: Optional[int] = None,
     numero_nota: Optional[int] = None,
     serie_nota: Optional[int] = None,
-    filial_nota: Optional[int] = None
+    fornecedor: Optional[int] = None
 ):
     """
     Consulta a NF no EMSYS (ahead-nota) e retorna os campos essenciais:
@@ -548,7 +632,7 @@ async def get_dados_nf_emsys(
         chave (int, opcional): Chave de acesso da NF.
         numero_nota (int, opcional): Número da NF.
         serie_nota (int, opcional): Série da NF (obrigatória se numero_nota for informado).
-        filial_nota (int, opcional): Filial da NF (obrigatória se numero_nota for informado).
+        fornecedor (int, opcional): Filial da NF (obrigatória se numero_nota for informado).
 
     Returns:
         list[dict]: Lista de notas no formato esperado.
@@ -565,14 +649,14 @@ async def get_dados_nf_emsys(
 
     # Caso 2: veio numero_nota → exige serie e filial
     elif numero_nota is not None:
-        if serie_nota is None or filial_nota is None:
-            raise ValueError("Para buscar por número da nota é obrigatório informar também 'serie_nota' e 'filial_nota'.")
+        if serie_nota is None or fornecedor is None:
+            raise ValueError("Para buscar por número da nota é obrigatório informar também 'serie_nota' e 'fornecedor'.")
         params["numeroNfe"] = numero_nota
         params["serieNfe"] = serie_nota
-        params["empresaCodigo"] = filial_nota
+        params["fornecedorCnpj"] = fornecedor
 
     else:
-        raise ValueError("É necessário informar 'chave' ou ('numero_nota' + 'serie_nota' + 'filial_nota').")
+        raise ValueError("É necessário informar 'chave' ou ('numero_nota' + 'serie_nota' + 'fornecedor').")
 
     headers_basic = {"Authorization": f"Basic {env_config['API_AUTHORIZATION']}"}
 
@@ -582,8 +666,9 @@ async def get_dados_nf_emsys(
         ) as session:
             async with session.get(url, headers=headers_basic, params=params) as response:
                 if response.status != 200:
+                    body = await response.text()  # ✅ lê o conteúdo do body
                     raise Exception(
-                        f"Erro ao comunicar com endpoint do Simplifica: {await response.text()}"
+                        f"({response.status}): {body}"
                     )
 
                 data = await response.json()
@@ -656,7 +741,7 @@ async def get_dados_nf_emsys(
                 return resultado
 
     except Exception as e:
-        raise Exception(f"Erro ao comunicar com endpoint do Simplifica: {e}")
+        raise Exception(f"{e}")
 
 
 
@@ -843,3 +928,29 @@ async def download_file_from_historico(uuid: str):
         )
         console.print(f"\n{err_msg}\n", style="bold green")
         logger.info(err_msg)
+
+async def get_mfa_code(key: str):
+    try:
+        env, _ = load_env_config()
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": "Basic " + env["API_AUTHORIZATION"],
+        }
+        payload = {"key": key}
+
+        response = requests.post(
+            env["API_BASE_URL"] + "/redis/get-redis-code",
+            json=payload,
+            headers=headers,
+        )
+
+        if response.status_code == 200 and response.json() is not None:
+            return {"code": response.json(), "status_code": 200}
+        else:
+            raise Exception(
+                f"Error to get mfa code, message: {response.text}, status_code {response.status_code}"
+            )
+
+    except Exception as e:
+        logger.error(f"Error to get mfa code: {str(e)}")
+        return {"code": None, "status_code": 500}

worker_automate_hub/api/rpa_historico_service.py

@@ -53,6 +53,7 @@ async def create_store_historico(
             or task.configEntrada.get("chaveCte")
             or task.configEntrada.get("empresa")
             or task.configEntrada.get("uuidSimplifica")
+            or task.configEntrada.get("identificador")
             or ""
         )
 

worker_automate_hub/tasks/jobs/abertura_livros_fiscais.py

@@ -8,7 +8,7 @@ import os
 from pywinauto.findwindows import ElementNotFoundError
 from pywinauto.keyboard import send_keys
 
-# sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..')))
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..')))
 from worker_automate_hub.utils.logger import logger
 from worker_automate_hub.models.dto.rpa_historico_request_dto import (
     RpaHistoricoStatusEnum,
@@ -45,6 +45,8 @@ console = Console()
 
 emsys = EMSys()
 
+# ASSETS_PATH = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros"
+ASSETS_PATH  = r"assets\abertura_livros"
 
 @repeat(times=10, delay=5)
 async def wait_aguarde_window_closed(app, timeout=60):
@@ -136,7 +138,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             console.print("Aguardando janela 'Movimento de Livro Fiscal' aparecer...")
 
             # Tempo limite de espera (em segundos)
-            timeout = 60
+            timeout = 3600
             inicio = time.time()
 
             # Espera até a janela aparecer
@@ -192,8 +194,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             saida_checkbox.click_input()
 
             console.print("Aguardar marcar caixa de saida")
-            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\saida_marcada.png"
-            imagem = "assets\\abertura_livros\\saida_marcada.png"
+            imagem = fr"{ASSETS_PATH}\saida_marcada.png"
 
             tempo_limite = 600  # 10 minutos
             intervalo = 2  # segundos entre verificações
@@ -329,8 +330,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             await worker_sleep(5)
 
             console.print("Aguardar o término de carregar")
-            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\livros_incluidos.png"
-            imagem = "assets\\abertura_livros\\livros_incluidos.png"
+            imagem = fr"{ASSETS_PATH}\livros_incluidos.png"
 
             tempo_limite = 6600  # 1h
             intervalo = 2  # segundos entre as verificações
@@ -372,8 +372,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
                 # 2. Verifica e trata janela de confirmação TMessageForm
                 try:
-                    # img_dialog = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\notas_rejeitadas.png"
-                    img_dialog = "assets\\abertura_livros\\notas_rejeitadas.png"
+                    img_dialog = fr"{ASSETS_PATH}\notas_rejeitadas.png"
                     if os.path.exists(img_dialog):
                         caixa = pyautogui.locateOnScreen(
                             img_dialog, confidence=0.9, grayscale=True
@@ -397,8 +396,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
                             await worker_sleep(5)
 
-                            # img_dialog = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\gerar_rel_notas_rejeitadas.png"
-                            img_dialog = "assets\\abertura_livros\\gerar_rel_notas_rejeitadas.png"
+                            img_dialog = fr"{ASSETS_PATH}\gerar_rel_notas_rejeitadas.png"
                             if os.path.exists(img_dialog):
                                 caixa = pyautogui.locateOnScreen(
                                     img_dialog, confidence=0.9, grayscale=True
@@ -476,8 +474,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             # Verificando se a foi confirmado os livres
             console.print("Verificando se os livros foram confirmados")
 
-            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\confirmado_livros.png"
-            imagem = "assets\\abertura_livros\\confirmado_livros.png"
+            imagem = fr"{ASSETS_PATH}\confirmado_livros.png"
             if os.path.exists(imagem):
                 caixa = pyautogui.locateOnScreen(imagem, confidence=0.9, grayscale=True)
             else:
@@ -610,8 +607,7 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
 
             console.print("Clicando no botão incluir apuração")
             # Clicar em incluir apuração
-            # imagem = r"C:\Users\automatehub\Documents\GitHub\worker-automate-hub\assets\abertura_livros\btn_incluir_apuracao.png"
-            imagem = "assets\\abertura_livros\\btn_incluir_apuracao.png"
+            imagem = fr"{ASSETS_PATH}\btn_incluir_apuracao.png"
 
             # Tenta localizar a imagem na tela
             localizacao = pyautogui.locateCenterOnScreen(imagem, confidence=0.9)
@@ -648,3 +644,4 @@ async def abertura_livros_fiscais(task: RpaProcessoEntradaDTO) -> RpaRetornoProc
             status=RpaHistoricoStatusEnum.Falha,
             tags=[RpaTagDTO(descricao=RpaTagEnum.Tecnico)],
         )
+

worker_automate_hub/tasks/jobs/descartes.py

@@ -432,7 +432,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                     task.configEntrada["filialEmpresaOrigem"] + ALMOXARIFADO_DEFAULT
                 )
                 pyautogui.hotkey("tab")
-                await worker_sleep(2)
+                await worker_sleep(10)
                 console.print(
                     f"\nDigitou almoxarifado {task.configEntrada["filialEmpresaOrigem"] + ALMOXARIFADO_DEFAULT}",
                     style="bold green",
@@ -462,7 +462,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
             pyautogui.hotkey("del")
             pyautogui.write(item["codigoProduto"])
             pyautogui.hotkey("tab")
-            await worker_sleep(2)
+            await worker_sleep(10)
             console.print(f"\nDigitou item {item['codigoProduto']}", style="bold green")
 
             # Checa tela de pesquisa de item
@@ -523,7 +523,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
 
             screenshot_path = take_screenshot()
 
-            await worker_sleep(2)
+            await worker_sleep(5)
 
             # Seleciona o Saldo Disponivel e verifica se ah possibilidade do descarte
             console.print(
@@ -536,7 +536,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                     class_name="TDBIEditNumber", found_index=9
                 ).window_text()
             except Exception as error:
-                console.print(f"Erro ao selecionar o Saldo Disponivel: {error}")
+                console.print(f"Erro ao selecionar o Saldo Disponivel: {str(error)}")
                 await send_to_webhook(
                     task.configEntrada["urlRetorno"],
                     "ERRO",
@@ -572,7 +572,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                     f"Saldo Disponivel: '{amount_avaliable}'", style="bold green"
                 )
             except Exception as error:
-                console.print(f"Erro ao converter o Saldo Disponível: {error}")
+                console.print(f"Erro ao converter o Saldo Disponível: {str(error)}")
                 await send_to_webhook(
                     task.configEntrada["urlRetorno"],
                     "ERRO",
@@ -772,7 +772,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
                     )
                 except Exception as e:
                     console.print(
-                        f"Falha ao clicar no botão OK de pré-venda incluída: {e}",
+                        f"Falha ao clicar no botão OK de pré-venda incluída: {str(e)}",
                         style="red",
                     )
                     return RpaRetornoProcessoDTO(
@@ -794,7 +794,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
 
         except Exception as e:
             console.print(
-                f"O Botão OK de pré-venda incluída não foi encontrado: {e}", style="red"
+                f"O Botão OK de pré-venda incluída não foi encontrado: {str(e)}", style="red"
             )
             return RpaRetornoProcessoDTO(
                 sucesso=False,
@@ -1041,7 +1041,7 @@ async def descartes(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoDTO:
             )
 
     except Exception as ex:
-        log_msg = f"Erro Processo Descartes: {ex} | Número da nota: {nota_fiscal} | Valor: {valor_nota}"
+        log_msg = f"Erro Processo Descartes: {str(ex)} | Número da nota: {nota_fiscal} | Valor: {valor_nota}"
         logger.error(log_msg)
         console.print(log_msg, style="bold red")
         await send_to_webhook(