entropy-data 0.3.0__py3-none-any.whl

entropy_data/config.py

@@ -0,0 +1,141 @@
+"""Connection configuration management for ~/.entropy-data/config.toml."""
+
+import os
+import stat
+import tomllib
+from dataclasses import dataclass
+from pathlib import Path
+
+import tomli_w
+
+CONFIG_DIR = Path.home() / ".entropy-data"
+CONFIG_FILE = CONFIG_DIR / "config.toml"
+DEFAULT_HOST = "https://api.entropy-data.com"
+
+
+class ConfigurationError(Exception):
+    """Missing or invalid configuration."""
+
+
+@dataclass
+class ConnectionConfig:
+    api_key: str
+    host: str = DEFAULT_HOST
+
+
+def load_config() -> dict:
+    """Read ~/.entropy-data/config.toml, return empty dict if missing."""
+    if not CONFIG_FILE.exists():
+        return {}
+    with open(CONFIG_FILE, "rb") as f:
+        return tomllib.load(f)
+
+
+def save_config(config: dict) -> None:
+    """Write config.toml with 0600 permissions."""
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    with open(CONFIG_FILE, "wb") as f:
+        tomli_w.dump(config, f)
+    CONFIG_FILE.chmod(stat.S_IRUSR | stat.S_IWUSR)
+
+
+def resolve_connection(
+    connection_name: str | None = None,
+    cli_api_key: str | None = None,
+    cli_host: str | None = None,
+) -> ConnectionConfig:
+    """Resolve connection with precedence: CLI options > env vars > config file."""
+    api_key = cli_api_key
+    host = cli_host
+
+    # Layer 2: environment variables
+    if api_key is None:
+        api_key = os.getenv("ENTROPY_DATA_API_KEY")
+    if host is None:
+        host = os.getenv("ENTROPY_DATA_HOST")
+
+    # Layer 3: config file
+    if api_key is None or host is None:
+        config = load_config()
+        connections = config.get("connections", {})
+
+        name = connection_name or config.get("default_connection_name")
+        if connection_name and connection_name not in connections:
+            raise ConfigurationError(f"Connection '{connection_name}' not found.")
+        if name and name in connections:
+            conn = connections[name]
+            if api_key is None:
+                api_key = conn.get("api_key")
+            if host is None:
+                host = conn.get("host")
+
+    # Default host
+    if host is None:
+        host = DEFAULT_HOST
+
+    if api_key is None:
+        raise ConfigurationError(
+            "No API key found. Set ENTROPY_DATA_API_KEY, use --api-key, or run: entropy-data connection add <name>"
+        )
+
+    return ConnectionConfig(api_key=api_key, host=host)
+
+
+def add_connection(name: str, api_key: str, host: str = DEFAULT_HOST) -> None:
+    """Add or update a named connection."""
+    if not name or not name.strip():
+        raise ConfigurationError("Connection name must not be empty.")
+    config = load_config()
+    if "connections" not in config:
+        config["connections"] = {}
+    config["connections"][name] = {"api_key": api_key, "host": host}
+    # Set as default if it's the first connection
+    if "default_connection_name" not in config:
+        config["default_connection_name"] = name
+    save_config(config)
+
+
+def remove_connection(name: str) -> None:
+    """Remove a named connection."""
+    config = load_config()
+    connections = config.get("connections", {})
+    if name not in connections:
+        raise ConfigurationError(f"Connection '{name}' not found.")
+    del connections[name]
+    # Clear default if we removed it
+    if config.get("default_connection_name") == name:
+        if connections:
+            config["default_connection_name"] = next(iter(connections))
+        else:
+            config.pop("default_connection_name", None)
+    save_config(config)
+
+
+def set_default_connection(name: str) -> None:
+    """Set the default connection."""
+    config = load_config()
+    connections = config.get("connections", {})
+    if name not in connections:
+        raise ConfigurationError(f"Connection '{name}' not found.")
+    config["default_connection_name"] = name
+    save_config(config)
+
+
+def list_connections() -> list[dict]:
+    """List all connections with masked API keys."""
+    config = load_config()
+    default_name = config.get("default_connection_name")
+    connections = config.get("connections", {})
+    result = []
+    for name, conn in connections.items():
+        api_key = conn.get("api_key", "")
+        masked = api_key[:4] + "..." + api_key[-4:] if len(api_key) > 8 else "****"
+        result.append(
+            {
+                "name": name,
+                "host": conn.get("host", DEFAULT_HOST),
+                "api_key": masked,
+                "default": name == default_name,
+            }
+        )
+    return result

entropy_data/output.py

@@ -0,0 +1,119 @@
+"""Output formatting for CLI results."""
+
+import json
+from enum import Enum
+
+from rich.console import Console
+from rich.table import Table
+
+console = Console()
+error_console = Console(stderr=True)
+
+
+class OutputFormat(str, Enum):
+    table = "table"
+    json = "json"
+
+
+# Column definitions per resource type: list of (header, dict_key)
+RESOURCE_COLUMNS: dict[str, list[tuple[str, str]]] = {
+    "dataproducts": [("ID", "id"), ("Title", "name"), ("Status", "status"), ("Owner", "team.name")],
+    "datacontracts": [("ID", "id"), ("Title", "name"), ("Version", "version"), ("Owner", "team.name")],
+    "access": [
+        ("ID", "id"),
+        ("Purpose", "info.purpose"),
+        ("Status", "info.status"),
+        ("Active", "info.active"),
+        ("Provider", "provider.dataProductId"),
+        ("Consumer", "consumer.teamId"),
+    ],
+    "teams": [("ID", "id"), ("Name", "name"), ("Type", "type"), ("Parent", "parent")],
+    "sourcesystems": [("ID", "id"), ("Name", "name"), ("Owner", "owner")],
+    "definitions": [("ID", "id"), ("Name", "title"), ("Owner", "owner")],
+    "certifications": [("ID", "id"), ("Name", "name"), ("Rank", "rank"), ("Tag", "tag")],
+    "example-data": [("ID", "id"), ("Data Product", "dataProductId"), ("Schema", "schemaName")],
+    "test-results": [("ID", "id"), ("Data Contract", "dataContractId"), ("Result", "result")],
+    "events": [("ID", "id"), ("Type", "type"), ("Subject", "subject"), ("Time", "time")],
+    "costs": [("ID", "id"), ("Data Product", "dataProductId"), ("Amount", "amount"), ("Currency", "currency")],
+    "assets": [
+        ("ID", "id"),
+        ("Name", "info.name"),
+        ("Type", "info.type"),
+        ("Source", "info.source"),
+        ("Owner", "info.owner"),
+    ],
+    "tags": [("ID", "id"), ("Owner", "info.owner"), ("Description", "info.description")],
+    "lineage": [
+        ("Event Type", "eventType"),
+        ("Event Time", "eventTime"),
+        ("Job", "job.name"),
+        ("Namespace", "job.namespace"),
+    ],
+    "usage": [],
+}
+
+
+def _get_nested(data: dict, key: str) -> str:
+    """Get a nested value from a dict using dot notation."""
+    parts = key.split(".")
+    current = data
+    for part in parts:
+        if isinstance(current, dict):
+            current = current.get(part)
+        else:
+            return ""
+    return str(current) if current is not None else ""
+
+
+def print_resource(data: dict, resource_type: str, fmt: OutputFormat) -> None:
+    """Print a single resource."""
+    if fmt == OutputFormat.json:
+        console.print_json(json.dumps(data))
+        return
+
+    columns = RESOURCE_COLUMNS.get(resource_type, [])
+    if columns:
+        table = Table(show_header=True)
+        for header, _ in columns:
+            table.add_column(header)
+        table.add_row(*[_get_nested(data, key) for _, key in columns])
+        console.print(table)
+    else:
+        console.print_json(json.dumps(data))
+
+
+def print_resource_list(
+    data: list[dict], resource_type: str, fmt: OutputFormat, has_next_page: bool = False, page: int = 0
+) -> None:
+    """Print a list of resources."""
+    if fmt == OutputFormat.json:
+        console.print_json(json.dumps(data))
+        return
+
+    columns = RESOURCE_COLUMNS.get(resource_type, [])
+    if not columns:
+        console.print_json(json.dumps(data))
+        return
+
+    table = Table(show_header=True, title=f"{resource_type} (page {page})")
+    for header, _ in columns:
+        table.add_column(header)
+    for item in data:
+        table.add_row(*[_get_nested(item, key) for _, key in columns])
+    console.print(table)
+
+    if has_next_page:
+        console.print(f"\nMore results available. Use --page {page + 1} to see the next page.")
+
+
+def print_success(message: str) -> None:
+    console.print(f"[green]{message}[/green]")
+
+
+def print_link(url: str) -> None:
+    if url:
+        console.print(f"Open {url}")
+
+
+def print_error(message: str) -> None:
+    error_console.print(f"[red]Error: {message}[/red]")

entropy_data/util.py

@@ -0,0 +1,22 @@
+"""Shared utilities."""
+
+import json
+import sys
+from pathlib import Path
+
+import yaml
+
+
+def read_body(file: Path) -> dict:
+    """Read JSON or YAML from file path or stdin (-)."""
+    if str(file) == "-":
+        content = sys.stdin.read()
+    else:
+        content = file.read_text()
+    try:
+        data = json.loads(content)
+    except json.JSONDecodeError:
+        data = yaml.safe_load(content)
+    if not isinstance(data, dict):
+        raise ValueError(f"Expected a JSON/YAML object, got {type(data).__name__}.")
+    return data

entropy_data-0.3.0.dist-info/METADATA

@@ -0,0 +1,23 @@
+Metadata-Version: 2.4
+Name: entropy-data
+Version: 0.3.0
+Summary: CLI for Entropy Data
+Project-URL: Homepage, https://entropy-data.com
+Project-URL: Documentation, https://docs.entropy-data.com
+Project-URL: Repository, https://github.com/entropy-data/entropy-data-cli
+Author-email: Entropy Data <support@entropy-data.com>
+License-Expression: MIT
+License-File: LICENSE
+Requires-Python: >=3.12
+Requires-Dist: pydantic<3.0,>=2.10
+Requires-Dist: python-dotenv<2.0,>=1.0
+Requires-Dist: pyyaml<7.0,>=6.0
+Requires-Dist: requests<3.0,>=2.32
+Requires-Dist: rich<16.0,>=14.0
+Requires-Dist: tomli-w<2.0,>=1.0
+Requires-Dist: typer<1.0,>=0.15.0
+Provides-Extra: dev
+Requires-Dist: pre-commit>=3.7; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: responses>=0.25; extra == 'dev'
+Requires-Dist: ruff>=0.8; extra == 'dev'

entropy_data-0.3.0.dist-info/RECORD

@@ -0,0 +1,33 @@
+entropy_data/__init__.py,sha256=7gUBK-cksVtizDORiQOtbtPFmpRbZbMORdHsf0DwuuY,147
+entropy_data/__main__.py,sha256=TPkjTet86UW21BUomzL6QcGH_jKFOFwM2iHApJ-sohk,40
+entropy_data/cli.py,sha256=uZ7RUp1FNzVGm4u_dtWEYz2Ri5JgibAOZsA6gGZGEDc,5901
+entropy_data/client.py,sha256=ADjeDNVH18R-DX-3aMxYdNdnq0Iz00mgC6T3WFkcCCg,6790
+entropy_data/config.py,sha256=yYm5_y5dEs6uEwhey79FBeLwOOLqhm-hZ1kjtAcCuoc,4476
+entropy_data/output.py,sha256=eesJqR_bxMqek9_bGqHaxFKQZcL5XLzzljb3zPrWqy0,3958
+entropy_data/util.py,sha256=jtoyru6G0Ukroj_jQ7m1McAKEoroekxZnPAsoyJ95Yc,537
+entropy_data/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+entropy_data/commands/access.py,sha256=-ymE9t6202zEK82S2wYJRug5mWTE9WJW9zvp_iQpDiI,4102
+entropy_data/commands/api_keys.py,sha256=Jj1_lj3tUc1LhpK4H5rjfWYrqNZtZyO97aH4Q3VsWqs,2206
+entropy_data/commands/assets.py,sha256=aPy652XANlOqG4eK0nXuhZgQ5Yiywn9EBQhKOOL_oSA,2566
+entropy_data/commands/certifications.py,sha256=pIukkTX4k3-VHTF_VPU_e2-WcfkWw-UB9VKYAKK7N4s,2714
+entropy_data/commands/connection.py,sha256=Os7QR0SGugMoHgxRYIBOFM14kGg5GkGsXhHSJ8Ycq98,2707
+entropy_data/commands/costs.py,sha256=VABkHbv3NK_pfQFkl3EJktb-x5Oo40kT6r6KExg7q6U,1846
+entropy_data/commands/datacontracts.py,sha256=Rg8b8zAlHfc2y4eL4ITfTih9vPiX5E_P5yb98vFy824,4032
+entropy_data/commands/dataproducts.py,sha256=Esexadd3kQyDIztYt_LLyoLy0JSiq8yC9kH28QXrLf0,3280
+entropy_data/commands/definitions.py,sha256=lWOegYFHRlFrl7rr8q0Amek67n3SOWEZ2x2QCQrPohU,2651
+entropy_data/commands/events.py,sha256=dKfVuWCDJRTuyeJaMNuG1Ly_JbFIU_pPxhYCRzsO9D8,976
+entropy_data/commands/example_data.py,sha256=D5ZZuUji-p6a6uJQCBrJwiUzb8VcW_nLHV0vhKybBGk,2807
+entropy_data/commands/import_export.py,sha256=DuHNM11yfye1gymYH5n--sjqWpTE9wua0VVLnEfwsU8,4289
+entropy_data/commands/lineage.py,sha256=NpUyaVMmDV7Jrkb66tNGHmJiDFnHP_AzMn34QrXB1Sc,3961
+entropy_data/commands/search.py,sha256=YO0cGjDHCiQh9OEHPsNDTmYg4RHy0yP1bg8NIhHTiXc,1824
+entropy_data/commands/settings.py,sha256=EBSNAV9AGpF5_4yXjxSLBD17-w361zG9CAU3Xipc6HM,2349
+entropy_data/commands/sourcesystems.py,sha256=a5tzl69ntolVV_Nn94sdT1siPBMjtC8JHiARBcYJLgE,2703
+entropy_data/commands/tags.py,sha256=IQ9HxljQ7r6bNweitGUxDLSslfmamztNHhV0TDB6bEA,2760
+entropy_data/commands/teams.py,sha256=NeUAW59wHEQsrM1c-iw3bsJ3GQUVPV4g5ySls74p1m8,2525
+entropy_data/commands/test_results.py,sha256=64uNhlXsIx5J-mfoVkO5j8Naox0-T_aF-00g7vOgWvk,2848
+entropy_data/commands/usage.py,sha256=sE1JH5_3iNSRTwYFEZqHWBeRiAcjYMTv_M8nByiE6NA,3473
+entropy_data-0.3.0.dist-info/METADATA,sha256=Hz9pp9brktLNN-BLY0d3vqIE_kV-7LCJ84_qD2c8sjQ,828
+entropy_data-0.3.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+entropy_data-0.3.0.dist-info/entry_points.txt,sha256=2MI80plj9d4TOc2jMFpcyBGOidIQkH4-1jiV-oT1shk,54
+entropy_data-0.3.0.dist-info/licenses/LICENSE,sha256=gn-OXYQ3yCus2bnJ2MHbmkOx7JUW5fBqzzN7U_Snbw4,1069
+entropy_data-0.3.0.dist-info/RECORD,,

entropy_data-0.3.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

entropy_data-0.3.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+entropy-data = entropy_data.cli:app

entropy_data-0.3.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Entropy Data
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.