sqr-cli 0.1.0__tar.gz

sqr_cli-0.1.0/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: sqr-cli
+Version: 0.1.0
+Summary: CLI for the YYS-SQR watermarking & digital product passport API
+License: MIT
+Project-URL: Homepage, https://yys-sqr-render-bsbe.onrender.com
+Project-URL: Documentation, https://yys-sqr-render-bsbe.onrender.com/api/v2/docs
+Requires-Python: >=3.8
+Requires-Dist: requests>=2.28.0

sqr_cli-0.1.0/pyproject.toml

@@ -0,0 +1,18 @@
+[build-system]
+requires = ["setuptools>=68.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "sqr-cli"
+version = "0.1.0"
+description = "CLI for the YYS-SQR watermarking & digital product passport API"
+requires-python = ">=3.8"
+dependencies = ["requests>=2.28.0"]
+license = {text = "MIT"}
+
+[project.scripts]
+sqr = "sqr_cli.main:main"
+
+[project.urls]
+Homepage = "https://yys-sqr-render-bsbe.onrender.com"
+Documentation = "https://yys-sqr-render-bsbe.onrender.com/api/v2/docs"

sqr_cli-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

sqr_cli-0.1.0/sqr_cli/__init__.py

@@ -0,0 +1,3 @@
+"""YYS-SQR CLI — command-line client for the YYS-SQR API."""
+
+__version__ = "0.1.0"

sqr_cli-0.1.0/sqr_cli/__main__.py

@@ -0,0 +1,4 @@
+"""Allow running as `python -m sqr_cli`."""
+from .main import main
+
+main()

sqr_cli-0.1.0/sqr_cli/main.py

@@ -0,0 +1,375 @@
+#!/usr/bin/env python3
+"""
+YYS-SQR CLI — command-line client for the YYS-SQR API.
+
+Install:  pip install sqr-cli
+Config:   ~/.sqr/config.json
+"""
+import argparse
+import base64
+import json
+import os
+import sys
+
+import requests
+
+from . import __version__
+
+CONFIG_DIR = os.path.expanduser("~/.sqr")
+CONFIG_FILE = os.path.join(CONFIG_DIR, "config.json")
+DEFAULT_API_URL = "https://yys-sqr-render-bsbe.onrender.com"
+
+
+# ---------------------------------------------------------------------------
+# Config helpers
+# ---------------------------------------------------------------------------
+
+def load_config():
+    if os.path.exists(CONFIG_FILE):
+        with open(CONFIG_FILE) as f:
+            return json.load(f)
+    return {}
+
+
+def save_config(cfg):
+    os.makedirs(CONFIG_DIR, exist_ok=True)
+    with open(CONFIG_FILE, "w") as f:
+        json.dump(cfg, f, indent=2)
+
+
+def get_api_url():
+    cfg = load_config()
+    return cfg.get("api_url", DEFAULT_API_URL).rstrip("/")
+
+
+def get_headers():
+    cfg = load_config()
+    headers = {"Content-Type": "application/json"}
+    key = cfg.get("api_key")
+    if key:
+        headers["Authorization"] = f"Bearer {key}"
+    return headers
+
+
+# ---------------------------------------------------------------------------
+# HTTP helpers
+# ---------------------------------------------------------------------------
+
+def _request(method, path, **kwargs):
+    url = f"{get_api_url()}{path}"
+    headers = kwargs.pop("headers", get_headers())
+    try:
+        resp = getattr(requests, method)(url, headers=headers, timeout=60, **kwargs)
+    except requests.ConnectionError:
+        print(f"Error: could not connect to {url}")
+        sys.exit(1)
+    except requests.Timeout:
+        print("Error: request timed out")
+        sys.exit(1)
+
+    try:
+        body = resp.json()
+    except ValueError:
+        body = resp.text
+
+    if resp.status_code >= 400:
+        err = body.get("error", body) if isinstance(body, dict) else body
+        print(f"Error ({resp.status_code}): {err}")
+        sys.exit(1)
+
+    return body
+
+
+def _print_json(data):
+    print(json.dumps(data, indent=2))
+
+
+# ---------------------------------------------------------------------------
+# Commands
+# ---------------------------------------------------------------------------
+
+def cmd_configure(args):
+    cfg = load_config()
+    api_url = input(f"API URL [{cfg.get('api_url', DEFAULT_API_URL)}]: ").strip()
+    if api_url:
+        cfg["api_url"] = api_url
+    elif "api_url" not in cfg:
+        cfg["api_url"] = DEFAULT_API_URL
+
+    api_key = input(f"API key [{cfg.get('api_key', '')}]: ").strip()
+    if api_key:
+        cfg["api_key"] = api_key
+
+    save_config(cfg)
+    print(f"Config saved to {CONFIG_FILE}")
+
+
+def cmd_health(args):
+    _print_json(_request("get", "/api/health"))
+
+
+# -- Templates ---------------------------------------------------------------
+
+def cmd_templates_list(args):
+    _print_json(_request("get", "/api/templates"))
+
+
+def cmd_templates_get(args):
+    _print_json(_request("get", f"/api/templates/{args.id}"))
+
+
+# -- Records -----------------------------------------------------------------
+
+def cmd_records_list(args):
+    params = {}
+    if args.page:
+        params["page"] = args.page
+    if args.per_page:
+        params["per_page"] = args.per_page
+    _print_json(_request("get", "/api/records", params=params))
+
+
+def cmd_records_get(args):
+    _print_json(_request("get", f"/api/records/{args.id}"))
+
+
+def cmd_records_create(args):
+    if not args.image:
+        print("Error: --image / -i is required")
+        sys.exit(1)
+
+    img_path = args.image
+    if not os.path.isfile(img_path):
+        print(f"Error: file not found: {img_path}")
+        sys.exit(1)
+
+    with open(img_path, "rb") as f:
+        image_b64 = base64.b64encode(f.read()).decode()
+
+    metadata = {}
+    for kv in (args.set or []):
+        if "=" not in kv:
+            print(f"Error: --set value must be key=value, got: {kv}")
+            sys.exit(1)
+        k, v = kv.split("=", 1)
+        metadata[k] = v
+
+    payload = {
+        "image": image_b64,
+        "template_id": args.template or "custom",
+        "metadata": metadata,
+    }
+    _print_json(_request("post", "/api/records", json=payload))
+
+
+def cmd_records_update(args):
+    metadata = {}
+    for kv in (args.set or []):
+        if "=" not in kv:
+            print(f"Error: --set value must be key=value, got: {kv}")
+            sys.exit(1)
+        k, v = kv.split("=", 1)
+        metadata[k] = v
+
+    if not metadata:
+        print("Error: provide at least one --set key=value")
+        sys.exit(1)
+
+    _print_json(_request("patch", f"/api/records/{args.id}", json={"metadata": metadata}))
+
+
+# -- Scan / Embed ------------------------------------------------------------
+
+def cmd_scan(args):
+    img_path = args.image_path
+    if not os.path.isfile(img_path):
+        print(f"Error: file not found: {img_path}")
+        sys.exit(1)
+
+    with open(img_path, "rb") as f:
+        image_b64 = base64.b64encode(f.read()).decode()
+
+    payload = {"image": image_b64}
+    if args.corners:
+        try:
+            corners = json.loads(args.corners)
+        except json.JSONDecodeError:
+            print("Error: --corners must be valid JSON, e.g. '[[0,0],[100,0],[100,100],[0,100]]'")
+            sys.exit(1)
+        payload["corners"] = corners
+
+    _print_json(_request("post", "/api/scan", json=payload))
+
+
+def cmd_embed(args):
+    if not args.image:
+        print("Error: --image / -i is required")
+        sys.exit(1)
+    if not args.message:
+        print("Error: --message / -m is required")
+        sys.exit(1)
+
+    img_path = args.image
+    if not os.path.isfile(img_path):
+        print(f"Error: file not found: {img_path}")
+        sys.exit(1)
+
+    with open(img_path, "rb") as f:
+        image_b64 = base64.b64encode(f.read()).decode()
+
+    result = _request("post", "/api/embed", json={"image": image_b64, "message": args.message})
+
+    # Save watermarked image if present
+    if args.output and isinstance(result, dict) and result.get("watermarked_image"):
+        with open(args.output, "wb") as f:
+            f.write(base64.b64decode(result["watermarked_image"]))
+        print(f"Watermarked image saved to {args.output}")
+        result.pop("watermarked_image", None)
+
+    _print_json(result)
+
+
+# -- Keys -------------------------------------------------------------------
+
+def cmd_keys_create(args):
+    payload = {"name": args.name or "default"}
+    _print_json(_request("post", "/api/keys", json=payload))
+
+
+def cmd_keys_list(args):
+    _print_json(_request("get", "/api/keys"))
+
+
+def cmd_keys_revoke(args):
+    _print_json(_request("post", f"/api/keys/{args.id}/revoke"))
+
+
+# ---------------------------------------------------------------------------
+# Argument parser
+# ---------------------------------------------------------------------------
+
+def build_parser():
+    parser = argparse.ArgumentParser(
+        prog="sqr",
+        description="YYS-SQR CLI — watermarking & digital product passport platform",
+    )
+    parser.add_argument("-v", "--version", action="version", version=f"%(prog)s {__version__}")
+    sub = parser.add_subparsers(dest="command")
+
+    # configure
+    sub.add_parser("configure", help="Save API URL + key to ~/.sqr/config.json")
+
+    # health
+    sub.add_parser("health", help="GET /api/health")
+
+    # templates
+    tmpl = sub.add_parser("templates", help="Template commands")
+    tmpl_sub = tmpl.add_subparsers(dest="templates_cmd")
+    tmpl_sub.add_parser("list", help="List all templates")
+    tg = tmpl_sub.add_parser("get", help="Get template details")
+    tg.add_argument("id", help="Template ID")
+
+    # records
+    rec = sub.add_parser("records", help="Record commands")
+    rec_sub = rec.add_subparsers(dest="records_cmd")
+
+    rl = rec_sub.add_parser("list", help="List records")
+    rl.add_argument("--page", type=int)
+    rl.add_argument("--per-page", type=int, dest="per_page")
+
+    rg = rec_sub.add_parser("get", help="Get a record")
+    rg.add_argument("id", help="Watermark ID")
+
+    rc = rec_sub.add_parser("create", help="Create a record")
+    rc.add_argument("-t", "--template", default="custom", help="Template ID")
+    rc.add_argument("-i", "--image", required=True, help="Path to image file")
+    rc.add_argument("--set", action="append", metavar="KEY=VALUE", help="Metadata key=value")
+
+    ru = rec_sub.add_parser("update", help="Update record metadata")
+    ru.add_argument("id", help="Watermark ID")
+    ru.add_argument("--set", action="append", metavar="KEY=VALUE", help="Metadata key=value")
+
+    # scan
+    sc = sub.add_parser("scan", help="Scan an image for watermarks")
+    sc.add_argument("image_path", help="Path to image file")
+    sc.add_argument("--corners", help="JSON array of 4 corner points")
+
+    # embed
+    em = sub.add_parser("embed", help="Embed watermark into image")
+    em.add_argument("-i", "--image", required=True, help="Path to image file")
+    em.add_argument("-m", "--message", required=True, help="Message to embed (max 5 chars)")
+    em.add_argument("-o", "--output", help="Save watermarked image to file")
+
+    # keys
+    keys = sub.add_parser("keys", help="API key management")
+    keys_sub = keys.add_subparsers(dest="keys_cmd")
+    kc = keys_sub.add_parser("create", help="Create a new API key")
+    kc.add_argument("--name", default="default", help="Key name")
+    keys_sub.add_parser("list", help="List your API keys")
+    kr = keys_sub.add_parser("revoke", help="Revoke an API key")
+    kr.add_argument("id", help="Key ID")
+
+    return parser
+
+
+# ---------------------------------------------------------------------------
+# Main
+# ---------------------------------------------------------------------------
+
+def main():
+    parser = build_parser()
+    args = parser.parse_args()
+
+    if not args.command:
+        parser.print_help()
+        sys.exit(0)
+
+    dispatch = {
+        "configure": cmd_configure,
+        "health": cmd_health,
+        "scan": cmd_scan,
+        "embed": cmd_embed,
+    }
+
+    if args.command in dispatch:
+        dispatch[args.command](args)
+        return
+
+    if args.command == "templates":
+        if args.templates_cmd == "list":
+            cmd_templates_list(args)
+        elif args.templates_cmd == "get":
+            cmd_templates_get(args)
+        else:
+            print("Usage: sqr templates {list,get}")
+            sys.exit(1)
+        return
+
+    if args.command == "records":
+        cmds = {
+            "list": cmd_records_list,
+            "get": cmd_records_get,
+            "create": cmd_records_create,
+            "update": cmd_records_update,
+        }
+        if args.records_cmd in cmds:
+            cmds[args.records_cmd](args)
+        else:
+            print("Usage: sqr records {list,get,create,update}")
+            sys.exit(1)
+        return
+
+    if args.command == "keys":
+        cmds = {
+            "create": cmd_keys_create,
+            "list": cmd_keys_list,
+            "revoke": cmd_keys_revoke,
+        }
+        if args.keys_cmd in cmds:
+            cmds[args.keys_cmd](args)
+        else:
+            print("Usage: sqr keys {create,list,revoke}")
+            sys.exit(1)
+        return
+
+    parser.print_help()

sqr_cli-0.1.0/sqr_cli.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: sqr-cli
+Version: 0.1.0
+Summary: CLI for the YYS-SQR watermarking & digital product passport API
+License: MIT
+Project-URL: Homepage, https://yys-sqr-render-bsbe.onrender.com
+Project-URL: Documentation, https://yys-sqr-render-bsbe.onrender.com/api/v2/docs
+Requires-Python: >=3.8
+Requires-Dist: requests>=2.28.0

sqr_cli-0.1.0/sqr_cli.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+pyproject.toml
+sqr_cli/__init__.py
+sqr_cli/__main__.py
+sqr_cli/main.py
+sqr_cli.egg-info/PKG-INFO
+sqr_cli.egg-info/SOURCES.txt
+sqr_cli.egg-info/dependency_links.txt
+sqr_cli.egg-info/entry_points.txt
+sqr_cli.egg-info/requires.txt
+sqr_cli.egg-info/top_level.txt

sqr_cli-0.1.0/sqr_cli.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

sqr_cli-0.1.0/sqr_cli.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+sqr = sqr_cli.main:main

sqr_cli-0.1.0/sqr_cli.egg-info/requires.txt

@@ -0,0 +1 @@
+requests>=2.28.0

sqr_cli-0.1.0/sqr_cli.egg-info/top_level.txt

@@ -0,0 +1 @@
+sqr_cli