worker-automate-hub 0.5.749__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
@@ -144,41 +145,42 @@ async def get_processo(uuidProcesso: str) -> RpaProcesso:
         RpaProcesso: O processo caso tenha sido encontrado.
     """
     env_config, _ = load_env_config()
-    try:
-        if not uuidProcesso:
-            raise ValueError("O uuid do processo deve ser informado")
-
-        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
-        timeout = aiohttp.ClientTimeout(total=600) 
-
-        async with aiohttp.ClientSession(
-            connector=aiohttp.TCPConnector(verify_ssl=True), timeout=timeout
-        ) as session:
-            async with session.get(
-                f"{env_config["API_BASE_URL"]}/processo/{uuidProcesso}",
-                headers=headers_basic,
-            ) as response:
-                if response.status != 200:
-                    error_content = await response.text()
-                    raise Exception(f"Erro ao obter o processo: {error_content}")
-                res = await response.json()
-                if type(res["campos"]) == str and res["campos"] == "{}":
-                    res["campos"] = {}
-                return RpaProcesso(**res)
-
-    except ValueError as e:
-        logger.error(f"Erro ao obter o processo: {str(e)}")
-        console.print(
-            f"{e}\n",
-            style="bold red",
-        )
-    except Exception as e:
-        logger.error(f"Erro ao obter o processo: {str(e)}")
-        console.print(
-            f"{e}\n",
-            style="bold red",
+    x = 0
+    while x < 10:
+        try:
+            if not uuidProcesso:
+                raise ValueError("O uuid do processo deve ser informado")
+
+            headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
+            timeout = aiohttp.ClientTimeout(total=600) 
+
+            async with aiohttp.ClientSession(
+                connector=aiohttp.TCPConnector(verify_ssl=True), timeout=timeout
+            ) as session:
+                async with session.get(
+                    f"{env_config["API_BASE_URL"]}/processo/{uuidProcesso}",
+                    headers=headers_basic,
+                ) as response:
+                    if response.status != 200:
+                        x += 1
+                        console.print(f"Erro ao obter o processo: {response.content}")
+                        continue
+                    else:
+                        res = await response.json()
+                        if type(res["campos"]) == str and res["campos"] == "{}":
+                            res["campos"] = {}
+                        return RpaProcesso(**res)
+
+        except ValueError as e:
+            x += 1
+            logger.error(f"Erro ao obter o processo: {str(e)}")
+            console.print(
+                f"{e}\n",
+                style="bold red",
         )
-        return None
+            continue
+            
+    return None
 
 
 async def get_workers():
@@ -232,30 +234,34 @@ async def get_config_by_name(name: str) -> RpaConfiguracao:
     if "API_BASE_URL" not in env_config:
         raise ValueError("URL da API não encontrada na configuração do ambiente")
 
-    try:
-        headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
-        timeout = aiohttp.ClientTimeout(total=600) 
-
-        async with aiohttp.ClientSession(
-            connector=aiohttp.TCPConnector(verify_ssl=True), timeout=timeout
-        ) as session:
-            async with session.get(
-                f"{env_config["API_BASE_URL"]}/configuracao/api/{name}",
-                headers=headers_basic,
-            ) as response:
-                if response.status != 200:
-                    raise Exception(f"Erro ao obter a configuração: {response.content}")
-                data = await response.json()
-                return RpaConfiguracao(**data)
-
-    except Exception as e:
-        err_msg = f"Erro ao obter a configuração: {e}"
-        logger.error(err_msg)
-        console.print(
-            f"{err_msg}\n",
-            style="bold red",
-        )
-        return None
+    x = 0
+    while x < 10:
+        try:
+            headers_basic = {"Authorization": f"Basic {env_config["API_AUTHORIZATION"]}"}
+            timeout = aiohttp.ClientTimeout(total=600) 
+
+            async with aiohttp.ClientSession(
+                connector=aiohttp.TCPConnector(verify_ssl=True), timeout=timeout
+            ) as session:
+                async with session.get(
+                    f"{env_config["API_BASE_URL"]}/configuracao/api/{name}",
+                    headers=headers_basic,
+                ) as response:
+                    if response.status != 200:
+                        console.print(f"Erro ao obter a configuração: {response.content}")
+                        x += 1
+                    else:
+                        data = await response.json()
+                        return RpaConfiguracao(**data)
+        except Exception as e:
+            x += 1
+            err_msg = f"Erro ao obter a configuração: {e}"
+            logger.error(err_msg)
+            console.print(
+                f"{err_msg}\n",
+                style="bold red",
+            )
+    return None
 
 
 def sync_get_config_by_name(name: str) -> RpaConfiguracao:
@@ -493,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.
@@ -533,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:
@@ -543,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.
@@ -560,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']}"}
 
@@ -577,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()
@@ -651,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}")
 
 
 
@@ -739,6 +829,8 @@ async def send_file(
         elif file_extension == "xlsx":
             filename = desArquivo
             content_type = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+        elif file_extension == "csv": 
+            content_type = "text/csv"
         else:
             raise ValueError(f"Extensão de arquivo não suportada: {file_extension}")
 
@@ -836,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/cli.py

@@ -59,124 +59,110 @@ Existem 3 subcomandos disponíveis para essa aplicação
     [green][b]worker[/][/] --version
 
  [b]Para gerar o arquivo de configuração[/]
-    [green][b]worker[/][/] --configure
+    [green][b]worker[/][/] configure
 
  [b]Para informações detalhadas
     [blue][link=https://github.com/SIM-Rede/worker-automate-hub]Repo no GIT Argenta[/][/] | [blue][link=https://pypi.org/project/worker-automate-hub/]Publicação no PyPI[/][/]
 """
 
 
-def function_help(flag: bool):
-    if flag:
+@app.callback(invoke_without_command=True)
+def main(
+    ctx: Context,
+    version: bool = Option(
+        False,
+        "--version",
+        help="Mostra a versão instalada",
+        is_flag=True,
+    ),
+):
+    """Comando principal"""
+    if ctx.invoked_subcommand:
+        return
+
+    if version:
         console.print(
             importlib.metadata.version("worker-automate-hub"),
             style="bold blue",
         )
         raise Exit(code=0)
 
+    console.print(HELP_MESSAGE)
 
-def function_configure(flag: bool):
-
-    if flag:
-        console.clear()
-        environment_names = [
-            "local",
-            "qa",
-            "main",
-        ]
-        q = [
-            inquirer.Text("vault_token", "Por favor digite o token do Vault"),
-            inquirer.List("env_list", "Selecione o ambiente", environment_names),
-        ]
-        r = inquirer.prompt(q, theme=GreenPassion())
-
-        env_sel, credentials = load_environments(r["env_list"], r["vault_token"])
-        write_env_config(env_sel, credentials)
-        workers = asyncio.run(get_workers())
-
-        if workers == None:
-            console.print("\nNenhum worker encontrado.\n", style="yellow")
-            raise Exit(code=0)
-        else:
-
-            nomes_workers = [worker["nomRobo"] for worker in workers]
-            q2 = [
-                inquirer.List(
-                    "worker_list", "Selecione um Worker", choices=nomes_workers
-                )
-            ]
-            r2 = inquirer.prompt(q2, theme=GreenPassion())
-            worker_sel = next(
-                worker for worker in workers if worker["nomRobo"] == r2["worker_list"]
-            )
-            add_worker_config(worker_sel)
-
-            q3 = [
-                inquirer.Confirm(
-                    "reg_config",
-                    message="Adicionar configuração de inicialização aos registros do Windows?",
-                )
-            ]
-            r3 = inquirer.prompt(q3, theme=GreenPassion())
-
-            if r3["reg_config"]:
-                add_start_on_boot_to_registry()
-
-            q4 = [
-                inquirer.Confirm(
-                    "assets_config",
-                    message="Atualizar a pasta assets?",
-                )
-            ]
-            r4 = inquirer.prompt(q4, theme=GreenPassion())
-
-            if r4["assets_config"]:                
-                update_assets_v2()
-
-            q5 = [
-                inquirer.Confirm(
-                    "worker_bat", message="Criar o arquivo worker-startup.bat?"
-                )
-            ]
-            r5 = inquirer.prompt(q5, theme=GreenPassion())
-
-            if r5["worker_bat"]:
-                create_worker_bat()
-
-            q6 = [
-                inquirer.Confirm(
-                    "tesseract_install", message="Iniciar a instalação do Tesseract?"
-                )
-            ]
-            r6 = inquirer.prompt(q6, theme=GreenPassion())
-
-            if r6["tesseract_install"]:
-                asyncio.run(download_tesseract())
-
-            console.print(
-                "\nConfiguração finalizada com sucesso!\n", style="bold green"
-            )
-
-            raise Exit(code=0)
 
+@app.command()
+def configure():
+    """Executa o processo interativo de configuração"""
+    console.clear()
+    environment_names = [
+        "local",
+        "qa",
+        "main",
+    ]
+    q = [
+        inquirer.Text("vault_token", "Por favor digite o token do Vault"),
+        inquirer.List("env_list", "Selecione o ambiente", environment_names),
+    ]
+    r = inquirer.prompt(q, theme=GreenPassion())
+
+    env_sel, credentials = load_environments(r["env_list"], r["vault_token"])
+    write_env_config(env_sel, credentials)
+    workers = asyncio.run(get_workers())
+
+    if workers is None:
+        console.print("\nNenhum worker encontrado.\n", style="yellow")
+        raise Exit(code=0)
 
-@app.callback(invoke_without_command=True)
-def main(
-    ctx: Context,
-    version: bool = Option(False, callback=function_help, is_flag=True),
-    configure: bool = Option(False, callback=function_configure, is_flag=True),
-):
-    if ctx.invoked_subcommand:
-        return
-    console.print(HELP_MESSAGE)
+    nomes_workers = [worker["nomRobo"] for worker in workers]
+    q2 = [inquirer.List("worker_list", "Selecione um Worker", choices=nomes_workers)]
+    r2 = inquirer.prompt(q2, theme=GreenPassion())
+    worker_sel = next(
+        worker for worker in workers if worker["nomRobo"] == r2["worker_list"]
+    )
+    add_worker_config(worker_sel)
+
+    q3 = [
+        inquirer.Confirm(
+            "reg_config",
+            message="Adicionar configuração de inicialização aos registros do Windows?",
+        )
+    ]
+    r3 = inquirer.prompt(q3, theme=GreenPassion())
+    if r3["reg_config"]:
+        add_start_on_boot_to_registry()
+
+    q4 = [
+        inquirer.Confirm(
+            "assets_config",
+            message="Atualizar a pasta assets?",
+        )
+    ]
+    r4 = inquirer.prompt(q4, theme=GreenPassion())
+    if r4["assets_config"]:
+        update_assets_v2()
+
+    q5 = [inquirer.Confirm("worker_bat", message="Criar o arquivo worker-startup.bat?")]
+    r5 = inquirer.prompt(q5, theme=GreenPassion())
+    if r5["worker_bat"]:
+        create_worker_bat()
+
+    q6 = [
+        inquirer.Confirm(
+            "tesseract_install", message="Iniciar a instalação do Tesseract?"
+        )
+    ]
+    r6 = inquirer.prompt(q6, theme=GreenPassion())
+    if r6["tesseract_install"]:
+        asyncio.run(download_tesseract())
+
+    console.print("\nConfiguração finalizada com sucesso!\n", style="bold green")
+
+    raise Exit(code=0)
 
 
 def is_command_running(command):
     """
     Verifica se um comando CLI está sendo executado em outro terminal.
-
-    :param command: O comando CLI a ser verificado (lista de strings).
-    :return: True se o comando estiver sendo executado, False caso contrário.
     """
     command_str = " ".join(command)
     rep = 0
@@ -186,19 +172,11 @@ def is_command_running(command):
             if cmdline and isinstance(cmdline, list):
                 cmdline_str = " ".join(cmdline)
                 if command_str in cmdline_str:
-                    print(cmdline_str)
                     rep += 1
-
         except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
             continue
 
-    if rep > 1:
-        return True
-    else:
-        return False
-
-
-from typer import Option
+    return rep > 1
 
 
 @app.command()
@@ -214,19 +192,21 @@ def run(
         help="Executa o download da pasta assets atualizada.",
     ),
 ):
+    """Inicializa o worker"""
     if assets:
         update_assets_v2()
-        
+
     command = ["worker", "run"]
     if not force and is_command_running(command):
         console.print(
             "\nO script já está em execução. Saindo...\n", style="bold yellow"
         )
-        raise Exit(code=0)    
-    else:
-        run_worker(stop_event)
+        raise Exit(code=0)
+
+    run_worker(stop_event)
 
 
 @app.command()
 def update():
+    """Força verificação/atualização do worker"""
     check_for_update(stop_event)