worker-automate-hub 0.5.753__tar.gz → 0.5.838__tar.gz

{worker_automate_hub-0.5.753 → worker_automate_hub-0.5.838}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: worker-automate-hub
-Version: 0.5.753
+Version: 0.5.838
 Summary: Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta.
 Author: Joel Paim
 Requires-Python: >=3.12,<4.0
@@ -10,6 +10,7 @@ Classifier: Natural Language :: Portuguese (Brazilian)
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Utilities
 Requires-Dist: aiohttp (>=3.9.5,<4.0.0)
 Requires-Dist: beautifulsoup4 (>=4.12.3,<5.0.0)
@@ -49,6 +50,7 @@ Requires-Dist: timedelta (>=2020.12.3,<2021.0.0)
 Requires-Dist: toml (>=0.10.2,<0.11.0)
 Requires-Dist: torch (>=2.4.0,<3.0.0)
 Requires-Dist: typer (>=0.12.3,<0.13.0)
+Requires-Dist: unidecode (>=1.4.0,<2.0.0)
 Requires-Dist: webdriver-manager (>=4.0.2,<5.0.0)
 Requires-Dist: win10toast (>=0.9,<0.10)
 Requires-Dist: xhtml2pdf (>=0.2.16,<0.3.0)

{worker_automate_hub-0.5.753 → worker_automate_hub-0.5.838}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.5.753"
+version = "0.5.838"
 description = "Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta."
 authors = ["Joel Paim"]
 readme = "README.md"
@@ -61,6 +61,7 @@ gitpython = "^3.1.43"
 selenium = "^4.27.1"
 webdriver-manager = "^4.0.2"
 xlrd = "^2.0.1"
+unidecode = "^1.4.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.2"

{worker_automate_hub-0.5.753 → worker_automate_hub-0.5.838}/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")
+    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) 
+            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)
+            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:
-        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",
+        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():
@@ -497,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.
@@ -581,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()
@@ -655,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}")
 
 
 
@@ -743,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}")
 

worker_automate_hub-0.5.838/worker_automate_hub/cli.py

@@ -0,0 +1,212 @@
+import asyncio
+import importlib.metadata
+import threading
+
+import inquirer
+import psutil
+from inquirer.themes import GreenPassion
+from rich.console import Console
+from typer import Context, Exit, Option, Typer
+
+from worker_automate_hub.api.client import get_workers, load_environments
+from worker_automate_hub.core.so_manipulation import (
+    add_worker_config,
+    download_tesseract,
+    update_assets_v2,
+    write_env_config,
+)
+from worker_automate_hub.utils.updater import check_for_update
+from worker_automate_hub.utils.util import (
+    add_start_on_boot_to_registry,
+    create_worker_bat,
+)
+
+from .worker import run_worker
+
+console = Console()
+app = Typer()
+stop_event = threading.Event()  # Evento para parar as threads
+
+
+HELP_MESSAGE = """
+[b]Worker[/] - Grupo Argenta
+
+Forma de uso: [b]worker [SUBCOMANDO] [ARGUMENTOS][/]
+
+Existem 3 subcomandos disponíveis para essa aplicação
+
+- [b]run[/]: Inicializa o Worker na máquina atual e começa a solicitar trabalho para o orquestrador.
+- [b]validate[/]: Verifica se o Worker atual está configurado corretamente e pronto para ser inicializado.
+- [b]assets[/]: Realiza a limpeza e depois download na pasta assets de todos arquivos utilizado pelo worker durante execução.
+
+[b]Exemplos de uso:[/]
+ [b][blue]RUN[/][/]
+    [green][b]worker[/][/] [b]run[/]
+
+ [b][blue]UPDATE[/][/]
+    [green][b]worker[/][/] [b]update[/]
+
+ [b][blue]VALIDATE[/][/]
+    [green][b]worker[/][/] [b]validate[/]
+
+---
+
+[b]Help:[/]
+ [b]Para mais informações[/]
+    [green][b]worker[/][/] --help
+
+ [b]Para ver a versão instalada[/]
+    [green][b]worker[/][/] --version
+
+ [b]Para gerar o arquivo de configuração[/]
+    [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[/][/]
+"""
+
+
+@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)
+
+
+@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)
+
+    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.
+    """
+    command_str = " ".join(command)
+    rep = 0
+    for proc in psutil.process_iter(["pid", "name", "cmdline"]):
+        try:
+            cmdline = proc.info["cmdline"]
+            if cmdline and isinstance(cmdline, list):
+                cmdline_str = " ".join(cmdline)
+                if command_str in cmdline_str:
+                    rep += 1
+        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+            continue
+
+    return rep > 1
+
+
+@app.command()
+def run(
+    force: bool = Option(
+        False,
+        "--force",
+        help="Força a execução mesmo se outro comando já estiver rodando.",
+    ),
+    assets: bool = Option(
+        False,
+        "--assets",
+        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)
+
+    run_worker(stop_event)
+
+
+@app.command()
+def update():
+    """Força verificação/atualização do worker"""
+    check_for_update(stop_event)