colacloud-cli 0.1.0__tar.gz → 0.3.0__tar.gz

colacloud_cli-0.3.0/.mcp.json

@@ -0,0 +1,3 @@
+{
+  "mcpServers": {}
+}

{colacloud_cli-0.1.0 → colacloud_cli-0.3.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: colacloud-cli
-Version: 0.1.0
+Version: 0.3.0
 Summary: Command-line interface for the COLA Cloud API
 Project-URL: Homepage, https://colacloud.us
-Project-URL: Documentation, https://colacloud.us/docs/api
+Project-URL: Documentation, https://docs.colacloud.us/api-reference
 Project-URL: Repository, https://github.com/cola-cloud-us/colacloud-cli
 Author-email: Jay Sobel <jay@colacloud.us>
 License: MIT
@@ -300,5 +300,5 @@ MIT License - see [LICENSE](LICENSE) for details.
 ## Links
 
 - [COLA Cloud Website](https://colacloud.us)
-- [API Documentation](https://colacloud.us/docs/api)
+- [API Documentation](https://docs.colacloud.us/api-reference)
 - [GitHub Repository](https://github.com/cola-cloud-us/colacloud-cli)

{colacloud_cli-0.1.0 → colacloud_cli-0.3.0}/README.md

@@ -273,5 +273,5 @@ MIT License - see [LICENSE](LICENSE) for details.
 ## Links
 
 - [COLA Cloud Website](https://colacloud.us)
-- [API Documentation](https://colacloud.us/docs/api)
+- [API Documentation](https://docs.colacloud.us/api-reference)
 - [GitHub Repository](https://github.com/cola-cloud-us/colacloud-cli)

{colacloud_cli-0.1.0 → colacloud_cli-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "colacloud-cli"
-version = "0.1.0"
+version = "0.3.0"
 description = "Command-line interface for the COLA Cloud API"
 readme = "README.md"
 license = { text = "MIT" }
@@ -33,7 +33,7 @@ cola = "colacloud_cli.main:cli"
 
 [project.urls]
 Homepage = "https://colacloud.us"
-Documentation = "https://colacloud.us/docs/api"
+Documentation = "https://docs.colacloud.us/api-reference"
 Repository = "https://github.com/cola-cloud-us/colacloud-cli"
 
 [build-system]
@@ -47,7 +47,17 @@ packages = ["src/colacloud_cli"]
 dev = [
     "pytest>=7.0.0",
     "pytest-cov>=4.0.0",
-    "black>=23.0.0",
-    "isort>=5.12.0",
+    "respx>=0.21.0",
+    "ruff>=0.9.0",
     "mypy>=1.0.0",
 ]
+
+[tool.ruff]
+line-length = 88
+target-version = "py310"
+
+[tool.ruff.lint]
+select = ["F", "E", "W", "I", "UP", "B", "SIM", "RUF"]
+
+[tool.ruff.lint.isort]
+known-first-party = ["colacloud_cli"]

colacloud_cli-0.3.0/scripts/smoke_test.py

@@ -0,0 +1,196 @@
+#!/usr/bin/env python3
+"""
+Live smoke test for the colacloud-cli.
+
+Runs actual CLI commands against the production API to verify
+end-to-end functionality: config handling, HTTP requests, response
+formatting, and JSON output parsing.
+
+Usage:
+    COLA_API_KEY=... uv run python scripts/smoke_test.py
+    COLA_API_KEY=... uv run python scripts/smoke_test.py --base-url http://localhost:5001/api/v1
+"""
+
+import argparse
+import json
+import os
+import subprocess
+import sys
+import time
+
+results: list[tuple[bool, str]] = []
+
+
+def run_cola(*args: str, env_override: dict | None = None) -> tuple[int, str, str]:
+    """Run a cola CLI command and return (exit_code, stdout, stderr)."""
+    env = os.environ.copy()
+    if env_override:
+        env.update(env_override)
+    proc = subprocess.run(
+        ["uv", "run", "cola", *args],
+        capture_output=True,
+        text=True,
+        timeout=30,
+        env=env,
+    )
+    return proc.returncode, proc.stdout, proc.stderr
+
+
+def check(name: str, fn):
+    """Run a single check. Calls fn(), expects it to return or raise."""
+    start = time.monotonic()
+    try:
+        detail = fn()
+        elapsed = time.monotonic() - start
+        msg = f"  OK  {name} ({elapsed:.2f}s)"
+        if detail:
+            msg += f" — {detail}"
+        results.append((True, msg))
+        print(msg)
+    except Exception as e:
+        elapsed = time.monotonic() - start
+        msg = f"FAIL  {name} — {type(e).__name__}: {e} ({elapsed:.2f}s)"
+        results.append((False, msg))
+        print(msg)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="COLA Cloud CLI smoke test")
+    parser.add_argument(
+        "--base-url",
+        default=None,
+        help="Override API base URL (default: production)",
+    )
+    args = parser.parse_args()
+
+    api_key = os.environ.get("COLA_API_KEY") or os.environ.get("COLACLOUD_API_KEY")
+    if not api_key:
+        print("Error: COLA_API_KEY or COLACLOUD_API_KEY environment variable is required")
+        sys.exit(1)
+
+    env = {"COLACLOUD_API_KEY": api_key}
+    if args.base_url:
+        env["COLACLOUD_BASE_URL"] = args.base_url
+
+    display = args.base_url or "https://app.colacloud.us/api/v1"
+    print(f"Smoke testing colacloud CLI against {display}\n")
+
+    # --- --version ---
+    def test_version():
+        rc, out, _ = run_cola("--version", env_override=env)
+        assert rc == 0, f"exit code {rc}"
+        assert "cola" in out.lower(), f"unexpected output: {out}"
+        return out.strip()
+
+    check("cola --version", test_version)
+
+    # --- colas list --json ---
+    ttb_id = None
+
+    def test_colas_list():
+        nonlocal ttb_id
+        rc, out, err = run_cola("colas", "list", "--limit", "1", "--json", env_override=env)
+        assert rc == 0, f"exit code {rc}: {err}"
+        data = json.loads(out)
+        assert "data" in data, "missing 'data' key"
+        assert len(data["data"]) > 0, "no COLAs returned"
+        assert "pagination" in data, "missing 'pagination' key"
+        cola = data["data"][0]
+        ttb_id = cola["ttb_id"]
+        return f"total={data['pagination']['total']}, first={ttb_id}"
+
+    check("cola colas list --limit 1 --json", test_colas_list)
+
+    # --- colas get --json ---
+    def test_colas_get():
+        if not ttb_id:
+            return "skipped — no ttb_id from list"
+        rc, out, err = run_cola("colas", "get", ttb_id, "--json", env_override=env)
+        assert rc == 0, f"exit code {rc}: {err}"
+        data = json.loads(out)
+        assert "data" in data, "missing 'data' key"
+        assert data["data"]["ttb_id"] == ttb_id
+        return f"ttb_id={ttb_id}, type={data['data'].get('product_type')}"
+
+    check(f"cola colas get {ttb_id} --json", test_colas_get)
+
+    # --- colas search --json ---
+    def test_colas_search():
+        rc, out, err = run_cola(
+            "colas", "search", "bourbon", "--limit", "5", "--json", env_override=env
+        )
+        assert rc == 0, f"exit code {rc}: {err}"
+        data = json.loads(out)
+        assert "data" in data
+        return f"found {len(data['data'])} results"
+
+    check('cola colas search "bourbon" --json', test_colas_search)
+
+    # --- permittees list --json ---
+    permit_number = None
+
+    def test_permittees_list():
+        nonlocal permit_number
+        rc, out, err = run_cola(
+            "permittees", "list", "--limit", "1", "--json", env_override=env
+        )
+        assert rc == 0, f"exit code {rc}: {err}"
+        data = json.loads(out)
+        assert "data" in data
+        assert len(data["data"]) > 0
+        permit_number = data["data"][0]["permit_number"]
+        return f"total={data['pagination']['total']}, first={permit_number}"
+
+    check("cola permittees list --limit 1 --json", test_permittees_list)
+
+    # --- permittees get --json ---
+    def test_permittees_get():
+        if not permit_number:
+            return "skipped — no permit_number from list"
+        rc, out, err = run_cola(
+            "permittees", "get", permit_number, "--json", env_override=env
+        )
+        assert rc == 0, f"exit code {rc}: {err}"
+        data = json.loads(out)
+        assert "data" in data
+        return f"permit={permit_number}, company={data['data'].get('company_name')}"
+
+    check(f"cola permittees get {permit_number} --json", test_permittees_get)
+
+    # --- usage --json ---
+    def test_usage():
+        rc, out, err = run_cola("usage", "--json", env_override=env)
+        assert rc == 0, f"exit code {rc}: {err}"
+        data = json.loads(out)
+        assert "data" in data
+        usage = data["data"]
+        return f"tier={usage.get('tier')}, used={usage.get('requests_used')}/{usage.get('monthly_limit')}"
+
+    check("cola usage --json", test_usage)
+
+    # --- colas list (table output, no --json) ---
+    def test_table_output():
+        rc, out, err = run_cola("colas", "list", "--limit", "3", env_override=env)
+        assert rc == 0, f"exit code {rc}: {err}"
+        assert len(out.strip()) > 0, "empty output"
+        return f"{len(out.splitlines())} lines of output"
+
+    check("cola colas list --limit 3 (table)", test_table_output)
+
+    # --- Summary ---
+    total = len(results)
+    passed = sum(1 for p, _ in results if p)
+    failed = total - passed
+
+    print(f"\n{'=' * 40}")
+    print(f"  {passed}/{total} passed", end="")
+    if failed:
+        print(f", {failed} failed")
+    else:
+        print()
+
+    sys.exit(0 if failed == 0 else 1)
+
+
+if __name__ == "__main__":
+    main()

{colacloud_cli-0.1.0 → colacloud_cli-0.3.0}/src/colacloud_cli/api.py

@@ -1,6 +1,7 @@
 """COLA Cloud API client wrapper."""
 
-from typing import Any, Optional
+import contextlib
+from typing import Any
 
 import httpx
 
@@ -14,9 +15,9 @@ class APIError(Exception):
     def __init__(
         self,
         message: str,
-        status_code: Optional[int] = None,
-        error_code: Optional[str] = None,
-        details: Optional[dict] = None,
+        status_code: int | None = None,
+        error_code: str | None = None,
+        details: dict | None = None,
     ):
         super().__init__(message)
         self.message = message
@@ -45,7 +46,7 @@ class RateLimitError(APIError):
     def __init__(
         self,
         message: str,
-        retry_after: Optional[int] = None,
+        retry_after: int | None = None,
         **kwargs,
     ):
         super().__init__(message, **kwargs)
@@ -57,8 +58,8 @@ class ColaCloudClient:
 
     def __init__(
         self,
-        api_key: Optional[str] = None,
-        base_url: Optional[str] = None,
+        api_key: str | None = None,
+        base_url: str | None = None,
         timeout: float = 30.0,
     ):
         """Initialize the API client.
@@ -117,7 +118,9 @@ class ColaCloudClient:
         # Extract error information
         error_info = data.get("error", {})
         error_code = error_info.get("code", "unknown_error")
-        error_message = error_info.get("message", response.reason_phrase or "Unknown error")
+        error_message = error_info.get(
+            "message", response.reason_phrase or "Unknown error"
+        )
         error_details = error_info.get("details", {})
 
         # Handle specific error types
@@ -132,10 +135,8 @@ class ColaCloudClient:
         if response.status_code == 429:
             retry_after = None
             if "Retry-After" in response.headers:
-                try:
+                with contextlib.suppress(ValueError):
                     retry_after = int(response.headers["Retry-After"])
-                except ValueError:
-                    pass
 
             raise RateLimitError(
                 error_message,
@@ -160,8 +161,8 @@ class ColaCloudClient:
         """
         if not self.api_key:
             raise AuthenticationError(
-                "API key not configured. Run 'cola config set-key' to set your API key, "
-                "or set the COLACLOUD_API_KEY environment variable."
+                "API key not configured. Run 'cola config set-key' to set "
+                "your API key, or set the COLACLOUD_API_KEY environment variable."
             )
 
     def close(self) -> None:
@@ -178,14 +179,14 @@ class ColaCloudClient:
 
     def list_colas(
         self,
-        query: Optional[str] = None,
-        product_type: Optional[str] = None,
-        origin: Optional[str] = None,
-        brand_name: Optional[str] = None,
-        approval_date_from: Optional[str] = None,
-        approval_date_to: Optional[str] = None,
-        abv_min: Optional[float] = None,
-        abv_max: Optional[float] = None,
+        query: str | None = None,
+        product_type: str | None = None,
+        origin: str | None = None,
+        brand_name: str | None = None,
+        approval_date_from: str | None = None,
+        approval_date_to: str | None = None,
+        abv_min: float | None = None,
+        abv_max: float | None = None,
         page: int = 1,
         per_page: int = 20,
     ) -> dict[str, Any]:
@@ -193,7 +194,7 @@ class ColaCloudClient:
 
         Args:
             query: Full-text search query.
-            product_type: Filter by product type (malt beverage, wine, distilled spirits).
+            product_type: Filter by product type.
             origin: Filter by country/state.
             brand_name: Filter by brand name (partial match).
             approval_date_from: Filter by minimum approval date (YYYY-MM-DD).
@@ -248,9 +249,9 @@ class ColaCloudClient:
 
     def list_permittees(
         self,
-        query: Optional[str] = None,
-        state: Optional[str] = None,
-        is_active: Optional[bool] = None,
+        query: str | None = None,
+        state: str | None = None,
+        is_active: bool | None = None,
         page: int = 1,
         per_page: int = 20,
     ) -> dict[str, Any]:
@@ -310,6 +311,163 @@ class ColaCloudClient:
         response = self._client.get(f"/barcode/{barcode_value}")
         return self._handle_response(response)
 
+    # Processing times endpoints
+
+    def list_processing_times(
+        self,
+        commodity: str | None = None,
+    ) -> dict[str, Any]:
+        """Get COLA processing times.
+
+        Args:
+            commodity: Filter by commodity type.
+
+        Returns:
+            API response with processing times data.
+        """
+        self._require_api_key()
+
+        params: dict[str, Any] = {}
+        if commodity:
+            params["commodity"] = commodity
+
+        response = self._client.get("/processing-times", params=params)
+        return self._handle_response(response)
+
+    def list_formula_processing_times(
+        self,
+        formula_type: str | None = None,
+        commodity: str | None = None,
+    ) -> dict[str, Any]:
+        """Get formula processing times.
+
+        Args:
+            formula_type: Filter by formula type.
+            commodity: Filter by commodity type.
+
+        Returns:
+            API response with formula processing times data.
+        """
+        self._require_api_key()
+
+        params: dict[str, Any] = {}
+        if formula_type:
+            params["formula_type"] = formula_type
+        if commodity:
+            params["commodity"] = commodity
+
+        response = self._client.get("/processing-times/formula", params=params)
+        return self._handle_response(response)
+
+    def list_registration_processing_times(
+        self,
+        category: str | None = None,
+        application_type: str | None = None,
+    ) -> dict[str, Any]:
+        """Get registration processing times.
+
+        Args:
+            category: Filter by category.
+            application_type: Filter by application type.
+
+        Returns:
+            API response with registration processing times data.
+        """
+        self._require_api_key()
+
+        params: dict[str, Any] = {}
+        if category:
+            params["category"] = category
+        if application_type:
+            params["application_type"] = application_type
+
+        response = self._client.get("/processing-times/registration", params=params)
+        return self._handle_response(response)
+
+    # Production reports endpoint
+
+    def list_production_reports(
+        self,
+        commodity: str | None = None,
+        year: int | None = None,
+        month: int | None = None,
+        report_type: str | None = None,
+        statistical_group: str | None = None,
+        page: int = 1,
+        per_page: int = 100,
+    ) -> dict[str, Any]:
+        """Get production reports.
+
+        Args:
+            commodity: Filter by commodity type.
+            year: Filter by year.
+            month: Filter by month.
+            report_type: Filter by report type.
+            statistical_group: Filter by statistical group.
+            page: Page number.
+            per_page: Results per page (max 100).
+
+        Returns:
+            API response with production reports data.
+        """
+        self._require_api_key()
+
+        params: dict[str, Any] = {"page": page, "per_page": per_page}
+        if commodity:
+            params["commodity"] = commodity
+        if year is not None:
+            params["year"] = year
+        if month is not None:
+            params["month"] = month
+        if report_type:
+            params["report_type"] = report_type
+        if statistical_group:
+            params["statistical_group"] = statistical_group
+
+        response = self._client.get("/production-reports", params=params)
+        return self._handle_response(response)
+
+    # AVA endpoints
+
+    def list_avas(
+        self,
+        state: str | None = None,
+        query: str | None = None,
+    ) -> dict[str, Any]:
+        """List American Viticultural Areas (AVAs).
+
+        Args:
+            state: Filter by state.
+            query: Search by name.
+
+        Returns:
+            API response with AVA data.
+        """
+        self._require_api_key()
+
+        params: dict[str, Any] = {}
+        if state:
+            params["state"] = state
+        if query:
+            params["q"] = query
+
+        response = self._client.get("/avas", params=params)
+        return self._handle_response(response)
+
+    def get_ava(self, ava_id: str) -> dict[str, Any]:
+        """Get a single AVA by ID.
+
+        Args:
+            ava_id: The AVA identifier.
+
+        Returns:
+            API response with AVA details.
+        """
+        self._require_api_key()
+
+        response = self._client.get(f"/avas/{ava_id}")
+        return self._handle_response(response)
+
     # Usage endpoint
 
     def get_usage(self) -> dict[str, Any]:

colacloud_cli-0.3.0/src/colacloud_cli/commands/avas.py

@@ -0,0 +1,91 @@
+"""AVA commands for COLA Cloud CLI."""
+
+import json
+
+import click
+
+from colacloud_cli.api import APIError, get_client
+from colacloud_cli.commands.utils import console, handle_api_error
+
+
+@click.group(name="avas")
+def avas_group():
+    """Browse American Viticultural Areas (AVAs)."""
+    pass
+
+
+@avas_group.command(name="list")
+@click.option("--state", help="Filter by state (e.g., CA, OR, WA).")
+@click.option("-q", "--query", help="Search by AVA name.")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON.")
+def list_avas(state: str | None, query: str | None, as_json: bool):
+    """List American Viticultural Areas.
+
+    Browse and search federally recognized wine grape-growing regions.
+
+    Examples:
+
+    \b
+        # List all AVAs
+        cola avas list
+
+    \b
+        # Filter by state
+        cola avas list --state CA
+
+    \b
+        # Search by name
+        cola avas list -q "napa"
+    """
+    try:
+        with get_client() as client:
+            result = client.list_avas(state=state, query=query)
+
+        if as_json:
+            click.echo(json.dumps(result, indent=2))
+        else:
+            data = result.get("data", [])
+            meta = result.get("meta", {})
+
+            if not data:
+                console.print("[yellow]No AVAs found matching your criteria.[/]")
+                return
+
+            console.print(json.dumps(data, indent=2))
+            total = meta.get("total", len(data))
+            console.print(f"\n[dim]{total} result(s)[/]")
+
+    except APIError as e:
+        handle_api_error(e)
+
+
+@avas_group.command(name="get")
+@click.argument("ava_id")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON.")
+def get_ava(ava_id: str, as_json: bool):
+    """Get detailed information about a specific AVA.
+
+    AVA_ID is the unique identifier for the American Viticultural Area.
+
+    Examples:
+
+    \b
+        # Get AVA details
+        cola avas get napa-valley
+
+    \b
+        # Output as JSON
+        cola avas get napa-valley --json
+    """
+    try:
+        with get_client() as client:
+            result = client.get_ava(ava_id)
+
+        if as_json:
+            click.echo(json.dumps(result, indent=2))
+        else:
+            data = result.get("data", {})
+            console.print(json.dumps(data, indent=2))
+
+    except APIError as e:
+        handle_api_error(e)

{colacloud_cli-0.1.0 → colacloud_cli-0.3.0}/src/colacloud_cli/commands/colas.py

@@ -23,16 +23,28 @@ def colas_group():
 @click.option("-q", "--query", help="Full-text search query.")
 @click.option(
     "--product-type",
-    type=click.Choice(["malt beverage", "wine", "distilled spirits"], case_sensitive=False),
+    type=click.Choice(
+        ["malt beverage", "wine", "distilled spirits"], case_sensitive=False
+    ),
     help="Filter by product type.",
 )
 @click.option("--origin", help="Filter by origin (country/state).")
 @click.option("--brand", "brand_name", help="Filter by brand name (partial match).")
-@click.option("--date-from", "approval_date_from", help="Filter by minimum approval date (YYYY-MM-DD).")
-@click.option("--date-to", "approval_date_to", help="Filter by maximum approval date (YYYY-MM-DD).")
+@click.option(
+    "--date-from",
+    "approval_date_from",
+    help="Filter by minimum approval date (YYYY-MM-DD).",
+)
+@click.option(
+    "--date-to",
+    "approval_date_to",
+    help="Filter by maximum approval date (YYYY-MM-DD).",
+)
 @click.option("--abv-min", type=float, help="Filter by minimum ABV.")
 @click.option("--abv-max", type=float, help="Filter by maximum ABV.")
-@click.option("--limit", "per_page", default=20, type=int, help="Results per page (max 100).")
+@click.option(
+    "--limit", "per_page", default=20, type=int, help="Results per page (max 100)."
+)
 @click.option("--page", default=1, type=int, help="Page number.")
 @click.option("--json", "as_json", is_flag=True, help="Output as JSON.")
 def list_colas(
@@ -135,7 +147,9 @@ def get_cola(ttb_id: str, as_json: bool):
 
 @colas_group.command(name="search")
 @click.argument("query")
-@click.option("--limit", "per_page", default=20, type=int, help="Results per page (max 100).")
+@click.option(
+    "--limit", "per_page", default=20, type=int, help="Results per page (max 100)."
+)
 @click.option("--page", default=1, type=int, help="Page number.")
 @click.option("--json", "as_json", is_flag=True, help="Output as JSON.")
 def search_colas(query: str, per_page: int, page: int, as_json: bool):

{colacloud_cli-0.1.0 → colacloud_cli-0.3.0}/src/colacloud_cli/commands/config.py

@@ -49,7 +49,9 @@ def set_key(key: str | None):
     config.set_api_key(key)
 
     console.print("[green]Success![/] API key saved to ~/.colacloud/config.json")
-    console.print("[dim]The config file has been set to mode 600 (owner read/write only).[/]")
+    console.print(
+        "[dim]The config file has been set to mode 600 (owner read/write only).[/]"
+    )
 
 
 @config_group.command(name="show")