get-sinapi 3.1.0__tar.gz → 3.1.1__tar.gz

{get_sinapi-3.1.0 → get_sinapi-3.1.1}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2026 NR Data Analysis
+Copyright (c) 2026 Nadison Roberto | NR TECH
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

{get_sinapi-3.1.0/src/get_sinapi.egg-info → get_sinapi-3.1.1}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: get-sinapi
-Version: 3.1.0
+Version: 3.1.1
 Summary: Pipeline rastreável de dados públicos do SINAPI/CAIXA
-Author: NR TECH
+Author: Nadison Roberto
 License-Expression: MIT
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{get_sinapi-3.1.0 → get_sinapi-3.1.1}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "get-sinapi"
-version = "3.1.0"
+version = "3.1.1"
 description = "Pipeline rastreável de dados públicos do SINAPI/CAIXA"
 readme = "README.md"
 requires-python = ">=3.11"
 license = "MIT"
-authors = [{ name = "NR TECH" }]
+authors = [{ name = "Nadison Roberto" }]
 dependencies = [
   "requests>=2.31",
   "curl-cffi>=0.7",

{get_sinapi-3.1.0 → get_sinapi-3.1.1/src/get_sinapi.egg-info}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: get-sinapi
-Version: 3.1.0
+Version: 3.1.1
 Summary: Pipeline rastreável de dados públicos do SINAPI/CAIXA
-Author: NR TECH
+Author: Nadison Roberto
 License-Expression: MIT
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{get_sinapi-3.1.0 → get_sinapi-3.1.1}/src/get_sinapi.egg-info/SOURCES.txt

@@ -36,6 +36,7 @@ tests/test_api_connection_lifecycle.py
 tests/test_api_literal_search.py
 tests/test_bootstrap.py
 tests/test_caixa_collector.py
+tests/test_cli.py
 tests/test_compositions.py
 tests/test_config.py
 tests/test_config_finite_backoff.py

{get_sinapi-3.1.0 → get_sinapi-3.1.1}/src/getsinapi/__init__.py

@@ -1,2 +1,2 @@
 """GET-SINAPI: pipeline de dados do SINAPI/CAIXA."""
-__version__ = "3.1.0"
+__version__ = "3.1.1"

{get_sinapi-3.1.0 → get_sinapi-3.1.1}/src/getsinapi/cli.py

@@ -4,11 +4,91 @@ from __future__ import annotations
 
 import argparse
 import os
+import platform
+import sys
 from pathlib import Path
 
 from getsinapi import __version__
 from getsinapi.config import Settings
 
+CYAN = "\033[96m"
+BLUE = "\033[94m"
+YELLOW = "\033[93m"
+GREEN = "\033[92m"
+RESET = "\033[0m"
+
+
+def _enable_ansi() -> bool:
+    """Habilita cores ANSI no terminal, com fallback seguro."""
+    if not sys.stdout.isatty():
+        return False
+    if platform.system() != "Windows":
+        return True
+    try:
+        import ctypes
+        import msvcrt
+
+        handle = msvcrt.get_osfhandle(sys.stdout.fileno())
+        mode = ctypes.c_uint()
+        kernel32 = ctypes.windll.kernel32
+        if not kernel32.GetConsoleMode(handle, ctypes.byref(mode)):
+            return False
+        return bool(kernel32.SetConsoleMode(handle, mode.value | 0x0004))
+    except (AttributeError, OSError, ValueError):
+        return False
+
+
+ANSI_ENABLED = _enable_ansi()
+
+
+def _paint(value: str, color: str) -> str:
+    return f"{color}{value}{RESET}" if ANSI_ENABLED else value
+
+
+def _rule(char: str = "─", color: str = CYAN) -> None:
+    """Desenha um separador padronizado para os painéis do terminal."""
+    print(_paint("  " + char * 66, color))
+
+
+def _section(number: str, title: str, subtitle: str | None = None) -> None:
+    """Exibe uma seção numerada com hierarquia visual consistente."""
+    print()
+    print(_paint(f"  {number}  |  {title.upper()}", BLUE))
+    _rule()
+    if subtitle:
+        print(f"  {subtitle}")
+
+
+def _show_header() -> None:
+    """Exibe a identidade do produto e os avisos de titularidade no terminal."""
+    width = 68
+    print()
+    print(_paint("╔" + "═" * width + "╗", CYAN))
+    print(_paint("║" + "GET-SINAPI".center(width) + "║", CYAN))
+    print(_paint("║" + "PLATAFORMA DE DADOS SINAPI | CAIXA".center(width) + "║", BLUE))
+    print(_paint("║" + "NR TECH  •  TERMINAL PYTHON".center(width) + "║", CYAN))
+    print(_paint("║" + f"VERSÃO {__version__}".center(width) + "║", CYAN))
+    print(_paint("╚" + "═" * width + "╝", CYAN))
+    print("  FONTE       : Publicações oficiais da CAIXA / SINAPI")
+    print("  RECURSOS    : Consulta local • Excel • CSV • Exportação em lote")
+    print(_paint("  TITULAR     : © 2026 Nadison Roberto | NR TECH", BLUE))
+    print(_paint("  LICENÇA     : MIT — consulte o arquivo LICENSE", BLUE))
+    print()
+
+
+def _parse_multi_selection(answer: str, total: int) -> list[int] | None:
+    """Converte números separados por vírgula em índices; None representa 'todas'."""
+    value = answer.strip().upper()
+    if not value or value == "A":
+        return None
+    tokens = [item.strip() for item in value.split(",")]
+    if any(not token.isdigit() for token in tokens):
+        raise ValueError("Use números separados por vírgula ou A para todas.")
+    indexes = [int(token) for token in tokens]
+    if any(index < 1 or index > total for index in indexes):
+        raise ValueError(f"Seleção fora do intervalo. Informe valores de 1 a {total}.")
+    return list(dict.fromkeys(indexes))
+
 
 def build_parser() -> argparse.ArgumentParser:
     parser = argparse.ArgumentParser(
@@ -54,79 +134,90 @@ def _export_menu(settings: Settings) -> int:
         desktop_data / "getsinapi.sqlite3",
         settings.data_dir / "getsinapi.sqlite3",
     ]
-    database = next((path for path in candidates if path and path.is_file()),
-                    desktop_data / "getsinapi.sqlite3" if os.name == "nt" else settings.data_dir / "getsinapi.sqlite3")
+    database = next(
+        (path for path in candidates if path and path.is_file()),
+        desktop_data / "getsinapi.sqlite3" if os.name == "nt" else settings.data_dir / "getsinapi.sqlite3",
+    )
     if not database.is_file():
         print(f"Banco SINAPI não encontrado: {database}")
         print("Inicialize a base com os comandos download, extract, parse e load.")
         return 2
 
-    print("\n" + "=" * 62)
-    print("GET-SINAPI | CENTRAL DE EXPORTAÇÃO EM LOTE")
-    print("=" * 62)
-    print("1. Exportar tudo (insumos, composições e componentes)")
-    print("2. Escolher categorias")
+    print(_paint("  ╔" + "═" * 66 + "╗", CYAN))
+    print(_paint("  ║" + "CENTRAL DE EXPORTAÇÃO".center(66) + "║", CYAN))
+    print(_paint("  ╚" + "═" * 66 + "╝", CYAN))
+    print("  Selecione os dados, aplique os filtros e defina o formato de saída.")
+    _section("01", "Modo de exportação", "Escolha entre uma exportação completa ou personalizada.")
+    print(_paint("  [1]", CYAN) + " Exportação completa — insumos, composições e componentes")
+    print(_paint("  [2]", CYAN) + " Exportação personalizada — selecionar categorias")
     mode = _ask_choice("Selecione", {"1": "Tudo", "2": "Personalizado"}, "1")
 
     categories = ["INSUMO", "COMPOSICAO", "COMPONENTE"]
     if mode == "2":
-        print("\nCategorias disponíveis:")
-        print("1. Insumos   2. Composições   3. Componentes")
-        selected = input("Informe uma ou mais opções separadas por vírgula (ex.: 1,3): ").strip()
+        _section("02", "Categorias", "Informe uma ou mais categorias separadas por vírgula.")
+        print(_paint("  [1]", CYAN) + " Insumos")
+        print(_paint("  [2]", CYAN) + " Composições")
+        print(_paint("  [3]", CYAN) + " Componentes")
+        selected = input("Categorias (ex.: 1,3): ").strip()
         mapping = {"1": "INSUMO", "2": "COMPOSICAO", "3": "COMPONENTE"}
-        chosen = [mapping[item.strip()] for item in selected.split(",") if item.strip() in mapping]
-        if not chosen:
-            print("Nenhuma categoria válida foi selecionada.")
+        try:
+            indexes = _parse_multi_selection(selected, 3)
+            categories = list(mapping.values()) if indexes is None else [
+                mapping[str(index)] for index in indexes
+            ]
+        except ValueError as exc:
+            print(_paint(f"Seleção inválida: {exc}", YELLOW))
             return 2
-        categories = list(dict.fromkeys(chosen))
 
     refs = available_references(database)
     if not refs:
         print("A base SQLite não possui competências carregadas.")
         return 2
-    print("\nCompetências disponíveis:")
+    _section("03", "Competências disponíveis", "Selecione números separados por vírgula ou A para todas.")
     for index, reference in enumerate(refs, start=1):
-        print(f"{index}. {reference}")
-    print("A. Todas as competências")
-    reference_choice = input("Competência [A]: ").strip().upper() or "A"
-    if reference_choice == "A":
-        references = None
-    elif reference_choice.isdigit() and 1 <= int(reference_choice) <= len(refs):
-        references = [refs[int(reference_choice) - 1]]
-    else:
-        print("Competência inválida.")
+        print(_paint(f"  [{index:>2}] {reference}", CYAN))
+    print(_paint("  [ A]", CYAN) + " Todas as competências")
+    reference_choice = input("Competências (ex.: 1,2,5 ou A) [A]: ").strip().upper() or "A"
+    try:
+        indexes = _parse_multi_selection(reference_choice, len(refs))
+        references = None if indexes is None else [refs[index - 1] for index in indexes]
+    except ValueError as exc:
+        print(_paint(f"Competência inválida: {exc}", YELLOW))
         return 2
 
-    uf = input("\nUF (duas letras) ou ENTER para todas: ").strip().upper() or None
+    _section("04", "Filtros complementares", "Pressione ENTER para manter o filtro sem restrição.")
+    uf = input("  UF (duas letras) ou ENTER para todas: ").strip().upper() or None
     if uf and (len(uf) != 2 or not uf.isalpha()):
         print("UF inválida.")
         return 2
 
     regime_choice = _ask_choice(
-        "Regime: 1 todos | 2 desonerado | 3 não desonerado",
+        "  Regime: [1] Todos | [2] Desonerado | [3] Não desonerado",
         {"1": "all", "2": "desonerado", "3": "nao_desonerado"},
         "1",
     )
     regime = None if regime_choice == "1" else regime_choice
 
     encargo_choice = _ask_choice(
-        "Encargos: 1 todos | 2 com encargos | 3 sem encargos",
+        "  Encargos: [1] Todos | [2] Com encargos | [3] Sem encargos",
         {"1": "all", "2": "com_encargos", "3": "sem_encargos"},
         "1",
     )
     encargo_mode = None if encargo_choice == "1" else encargo_choice
 
-    search = input("Pesquisar por código/descrição (ENTER para sem filtro): ").strip() or None
+    search = input("  Código ou descrição (ENTER para sem filtro): ").strip() or None
+    _section("05", "Formato e destino", "Os arquivos serão gravados na pasta informada.")
     file_format = _ask_choice(
-        "Formato: 1 Excel (.xlsx) | 2 CSV | 3 Excel + CSV",
+        "  Formato: [1] Excel (.xlsx) | [2] CSV | [3] Excel + CSV",
         {"1": "xlsx", "2": "csv", "3": "both"},
         "1",
     )
     default_output = database.parent / "exports"
-    output_text = input(f"Pasta de destino [{default_output}]: ").strip()
+    output_text = input(f"  Pasta de destino [{default_output}]: ").strip()
     output_dir = Path(output_text).expanduser() if output_text else default_output
 
-    print("\nIniciando extração da base local...")
+    _section("06", "Processamento", "Preparando a exportação a partir da base SQLite local.")
+    print(_paint("  PROCESSANDO...", YELLOW))
     generated = export_catalog(
         database,
         output_dir,
@@ -138,15 +229,19 @@ def _export_menu(settings: Settings) -> int:
         encargo_mode=encargo_mode,
         search=search,
     )
-    print(f"\nExportação concluída. Arquivos gerados: {len(generated)}")
+    print(_paint("\n  STATUS  : EXPORTAÇÃO CONCLUÍDA", GREEN))
+    print(_paint(f"  ARQUIVOS: {len(generated)}", GREEN))
     for path in generated:
-        print(f"  - {path}")
+        print(_paint("  [OK]", GREEN) + f" {path}")
+    _rule("═")
+    print(_paint("  © 2026 Nadison Roberto | NR TECH", BLUE))
     return 0
 
 
 def main() -> int:
     parser = build_parser()
     args = parser.parse_args()
+    _show_header()
     settings = Settings.from_env()
 
     if args.command in (None, "export"):

get_sinapi-3.1.1/tests/test_cli.py

@@ -0,0 +1,36 @@
+import pytest
+
+from getsinapi.cli import _parse_multi_selection, _show_header
+
+
+def test_header_displays_product_identity_and_copyright(capsys, monkeypatch):
+    monkeypatch.setattr("getsinapi.cli.ANSI_ENABLED", False)
+    _show_header()
+    output = capsys.readouterr().out
+    assert "GET-SINAPI" in output
+    assert "VERSÃO 3.1.1" in output
+    assert "© 2026 Nadison Roberto | NR TECH" in output
+    assert "LICENÇA     : MIT" in output
+
+
+
+@pytest.mark.parametrize(
+    ("answer", "expected"),
+    [
+        ("1,2", [1, 2]),
+        (" 1, 3 ", [1, 3]),
+        ("2,2,1", [2, 1]),
+        ("A", None),
+        ("a", None),
+        ("", None),
+        ("   ", None),
+    ],
+)
+def test_parse_multi_selection_accepts_multiple_indexes_and_all(answer, expected):
+    assert _parse_multi_selection(answer, total=3) == expected
+
+
+@pytest.mark.parametrize("answer", ["0", "4", "1,,2", "1,x", "-1", "1.5"])
+def test_parse_multi_selection_rejects_invalid_values(answer):
+    with pytest.raises(ValueError):
+        _parse_multi_selection(answer, total=3)