artdam-cli 0.2.1__tar.gz → 0.2.3__tar.gz

{artdam_cli-0.2.1 → artdam_cli-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: artdam-cli
-Version: 0.2.1
+Version: 0.2.3
 Summary: ArtDAM 数字资产管理平台命令行工具
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -25,8 +25,11 @@ pip install artdam-cli
 ## 快速开始
 
 ```bash
-# 登录（首次使用，服务器地址已内置）
-artdam login --username <用户名> --password <密码>
+# 登录（默认走 SSO，自动打开浏览器完成授权）
+artdam login
+
+# 用账号密码登录
+artdam login --password --username <用户名>
 
 # 查看可访问的项目列表（获取项目 ID）
 artdam projects
@@ -70,21 +73,37 @@ artdam project show      # 查看当前默认项目
 
 登录 ArtDAM，保存 token 到本地 `~/.artdam/config.json`。
 
+**默认走 SSO 浏览器授权**（推荐）：
+
 ```bash
-artdam login --username <用户名> --password <密码>
+artdam login
+# 自动打开浏览器 → 完成公司 SSO 授权 → token 写入本地配置
 ```
 
+**账号密码登录**（仅管理员 / 服务账号）：
+
+```bash
+artdam login --password --username <用户名>
+# 回车后提示输入密码（不回显）
+```
+
+| 参数 | 说明 |
+|---|---|
+| `--url` | 服务器地址（默认已内置，无需填写） |
+| `--password` | 切换为账号密码模式（不加此 flag 默认走 SSO） |
+| `--username` | 用户名（仅 `--password` 模式） |
+
 #### `artdam search`
 
 搜索资产，返回匹配列表。
 
 ```bash
-artdam search "头盔"                        # 使用默认项目
-artdam search "头盔" --project 1            # 临时指定项目
-artdam search "头盔" --limit 50
-artdam search "头盔" --type image/png
-artdam search "头盔" --folder 10
-artdam search "头盔" --json                 # 输出原始 JSON（AI Agent 用）
+artdam search "场景"                        # 使用默认项目
+artdam search "场景" --project 1            # 临时指定项目
+artdam search "场景" --limit 50
+artdam search "场景" --type image/png
+artdam search "场景" --folder 10
+artdam search "场景" --json                 # 输出原始 JSON（AI Agent 用）
 ```
 
 | 参数 | 说明 |
@@ -225,8 +244,8 @@ artdam folders --json
 ### 典型 Agent 工作流
 
 ```bash
-# 1. 登录
-artdam login --username <用户名> --password <密码>
+# 1. 登录（SSO 浏览器授权，完成后自动保存 token）
+artdam login
 
 # 2. 查询项目列表，选择目标项目并设为默认
 artdam projects --json
@@ -263,7 +282,7 @@ artdam update <asset_id> --description "AI 生成的场景描述"
 
 ```json
 {
-  "base_url": "https://artdam.dsworks.cn",
+  "base_url": "https://artdam.xindong.com",
   "token": "eyJ...",
   "default_project": 1
 }

{artdam_cli-0.2.1 → artdam_cli-0.2.3}/README.md

@@ -15,8 +15,11 @@ pip install artdam-cli
 ## 快速开始
 
 ```bash
-# 登录（首次使用，服务器地址已内置）
-artdam login --username <用户名> --password <密码>
+# 登录（默认走 SSO，自动打开浏览器完成授权）
+artdam login
+
+# 用账号密码登录
+artdam login --password --username <用户名>
 
 # 查看可访问的项目列表（获取项目 ID）
 artdam projects
@@ -60,21 +63,37 @@ artdam project show      # 查看当前默认项目
 
 登录 ArtDAM，保存 token 到本地 `~/.artdam/config.json`。
 
+**默认走 SSO 浏览器授权**（推荐）：
+
 ```bash
-artdam login --username <用户名> --password <密码>
+artdam login
+# 自动打开浏览器 → 完成公司 SSO 授权 → token 写入本地配置
 ```
 
+**账号密码登录**（仅管理员 / 服务账号）：
+
+```bash
+artdam login --password --username <用户名>
+# 回车后提示输入密码（不回显）
+```
+
+| 参数 | 说明 |
+|---|---|
+| `--url` | 服务器地址（默认已内置，无需填写） |
+| `--password` | 切换为账号密码模式（不加此 flag 默认走 SSO） |
+| `--username` | 用户名（仅 `--password` 模式） |
+
 #### `artdam search`
 
 搜索资产，返回匹配列表。
 
 ```bash
-artdam search "头盔"                        # 使用默认项目
-artdam search "头盔" --project 1            # 临时指定项目
-artdam search "头盔" --limit 50
-artdam search "头盔" --type image/png
-artdam search "头盔" --folder 10
-artdam search "头盔" --json                 # 输出原始 JSON（AI Agent 用）
+artdam search "场景"                        # 使用默认项目
+artdam search "场景" --project 1            # 临时指定项目
+artdam search "场景" --limit 50
+artdam search "场景" --type image/png
+artdam search "场景" --folder 10
+artdam search "场景" --json                 # 输出原始 JSON（AI Agent 用）
 ```
 
 | 参数 | 说明 |
@@ -215,8 +234,8 @@ artdam folders --json
 ### 典型 Agent 工作流
 
 ```bash
-# 1. 登录
-artdam login --username <用户名> --password <密码>
+# 1. 登录（SSO 浏览器授权，完成后自动保存 token）
+artdam login
 
 # 2. 查询项目列表，选择目标项目并设为默认
 artdam projects --json
@@ -253,7 +272,7 @@ artdam update <asset_id> --description "AI 生成的场景描述"
 
 ```json
 {
-  "base_url": "https://artdam.dsworks.cn",
+  "base_url": "https://artdam.xindong.com",
   "token": "eyJ...",
   "default_project": 1
 }

artdam_cli-0.2.3/artdam_cli/commands/login.py

@@ -0,0 +1,112 @@
+import threading
+import webbrowser
+from http.server import BaseHTTPRequestHandler, HTTPServer
+from urllib.parse import parse_qs, urlparse
+
+import click
+import httpx
+
+from artdam_cli import config
+
+_SUCCESS_HTML = """<!DOCTYPE html>
+<html><head><meta charset="utf-8"><title>ArtDAM CLI</title>
+<style>body{font-family:sans-serif;text-align:center;padding:80px;background:#0b1117;color:#edf2f7}
+h2{color:#7eb893}p{color:#b7c2ce}</style></head>
+<body><h2>&#x2705; SSO 登录成功</h2>
+<p>可以关闭此页面，返回终端继续操作。</p>
+</body></html>""".encode()
+
+
+def _sso_login(url: str) -> None:
+    """Browser Callback Flow：启动本地 HTTP server，打开浏览器走完整 CAS SSO 流程。"""
+    token_holder: dict = {"value": None}
+    done = threading.Event()
+
+    class _Handler(BaseHTTPRequestHandler):
+        def do_GET(self):
+            params = parse_qs(urlparse(self.path).query)
+            token = params.get("token", [None])[0]
+            if token:
+                token_holder["value"] = token
+                self.send_response(200)
+                self.send_header("Content-Type", "text/html; charset=utf-8")
+                self.end_headers()
+                self.wfile.write(_SUCCESS_HTML)
+                done.set()
+            else:
+                self.send_response(400)
+                self.end_headers()
+
+        def log_message(self, *_):  # 抑制默认 server 日志
+            pass
+
+    # 绑定 port=0 让 OS 分配可用端口，避免 find-then-bind 竞争
+    server = HTTPServer(("127.0.0.1", 0), _Handler)
+    port = server.server_address[1]
+
+    t = threading.Thread(target=server.serve_forever, daemon=True)
+    t.start()
+
+    login_url = f"{url}/api/auth/sso/login?cli_port={port}"
+    click.echo("正在打开浏览器进行 SSO 登录...")
+    if not webbrowser.open(login_url):
+        click.echo(f"无法自动打开浏览器，请手动访问：\n  {login_url}")
+
+    click.echo("等待 SSO 授权（120 秒超时）...")
+    if not done.wait(timeout=120):
+        server.shutdown()
+        raise SystemExit("登录超时（120 秒），请重试")
+
+    server.shutdown()
+    config.save({"base_url": url, "token": token_holder["value"]})
+    click.echo("✓ SSO 登录成功，配置已保存到 ~/.artdam/config.json")
+
+
+def _password_login(url: str, username: str | None, passwd: str | None) -> None:
+    """账号密码登录（仅 admin / 服务账号使用）。"""
+    if not username:
+        username = click.prompt("用户名")
+    if not passwd:
+        passwd = click.prompt("密码", hide_input=True)
+    try:
+        r = httpx.post(
+            f"{url}/api/auth/login",
+            json={"username": username, "password": passwd},
+            timeout=10,
+        )
+    except httpx.ConnectError:
+        raise SystemExit(f"无法连接到 {url}，请检查地址")
+    if r.status_code != 200:
+        raise SystemExit(f"登录失败: {r.json().get('detail', r.text)}")
+    token = r.json().get("access_token")
+    if not token:
+        raise SystemExit("登录响应中未找到 token")
+    config.save({"base_url": url, "token": token})
+    click.echo("✓ 登录成功，配置已保存到 ~/.artdam/config.json")
+
+
+@click.command()
+@click.option(
+    "--url",
+    default="https://artdam.xindong.com",
+    help="服务器地址，默认: https://artdam.xindong.com",
+)
+@click.option(
+    "--password",
+    "use_password",
+    is_flag=True,
+    help="使用账号密码登录（仅 admin/服务账号，默认走 SSO）",
+)
+@click.option("--username", default=None, help="用户名（仅 --password 模式）")
+@click.option(
+    "--passwd", default=None, hide_input=True, help="密码（仅 --password 模式）"
+)
+def login(
+    url: str, use_password: bool, username: str | None, passwd: str | None
+) -> None:
+    """登录 ArtDAM。默认走 SSO 浏览器授权，加 --password 使用账号密码（仅管理员）。"""
+    url = url.rstrip("/")
+    if use_password:
+        _password_login(url, username, passwd)
+    else:
+        _sso_login(url)

{artdam_cli-0.2.1 → artdam_cli-0.2.3}/artdam_cli/commands/projects.py

@@ -1,41 +1,41 @@
-import json
-
-import click
-from rich.console import Console
-from rich.table import Table
-
-from artdam_cli import client
-
-console = Console()
-
-
-@click.command()
-@click.option("--json", "as_json", is_flag=True, help="输出原始 JSON（AI Agent 用）")
-def projects(as_json: bool) -> None:
-    """列出当前用户可访问的所有项目。"""
-    data = client.get("/api/projects/")
-    items = data if isinstance(data, list) else []
-
-    if as_json:
-        click.echo(json.dumps(items, ensure_ascii=False, indent=2))
-        return
-
-    if not items:
-        console.print("[yellow]暂无可访问的项目[/yellow]")
-        return
-
-    table = Table(title=f"项目列表（共 {len(items)} 个）")
-    table.add_column("ID", style="dim", width=6)
-    table.add_column("代码", width=10)
-    table.add_column("名称", min_width=16)
-    table.add_column("描述", min_width=20)
-
-    for item in items:
-        table.add_row(
-            str(item["id"]),
-            item.get("code", ""),
-            item.get("name", ""),
-            item.get("description") or "—",
-        )
-
-    console.print(table)
+import json
+
+import click
+from rich.console import Console
+from rich.table import Table
+
+from artdam_cli import client
+
+console = Console()
+
+
+@click.command()
+@click.option("--json", "as_json", is_flag=True, help="输出原始 JSON（AI Agent 用）")
+def projects(as_json: bool) -> None:
+    """列出当前用户可访问的所有项目。"""
+    data = client.get("/api/projects/")
+    items = data if isinstance(data, list) else []
+
+    if as_json:
+        click.echo(json.dumps(items, ensure_ascii=False, indent=2))
+        return
+
+    if not items:
+        console.print("[yellow]暂无可访问的项目[/yellow]")
+        return
+
+    table = Table(title=f"项目列表（共 {len(items)} 个）")
+    table.add_column("ID", style="dim", width=6)
+    table.add_column("代码", width=10)
+    table.add_column("名称", min_width=16)
+    table.add_column("描述", min_width=20)
+
+    for item in items:
+        table.add_row(
+            str(item["id"]),
+            item.get("code", ""),
+            item.get("name", ""),
+            item.get("description") or "—",
+        )
+
+    console.print(table)

{artdam_cli-0.2.1 → artdam_cli-0.2.3}/artdam_cli/main.py

@@ -1,5 +1,13 @@
+import io
+import sys
+
 import click
 
+# Windows 终端默认 GBK，强制 UTF-8 避免中文乱码
+if hasattr(sys.stdout, "buffer") and sys.stdout.encoding.lower() not in ("utf-8", "utf-8-sig"):
+    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", errors="replace")
+    sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding="utf-8", errors="replace")
+
 from artdam_cli.commands.login import login
 from artdam_cli.commands.search import search
 from artdam_cli.commands.get import get

{artdam_cli-0.2.1 → artdam_cli-0.2.3}/artdam_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: artdam-cli
-Version: 0.2.1
+Version: 0.2.3
 Summary: ArtDAM 数字资产管理平台命令行工具
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -25,8 +25,11 @@ pip install artdam-cli
 ## 快速开始
 
 ```bash
-# 登录（首次使用，服务器地址已内置）
-artdam login --username <用户名> --password <密码>
+# 登录（默认走 SSO，自动打开浏览器完成授权）
+artdam login
+
+# 用账号密码登录
+artdam login --password --username <用户名>
 
 # 查看可访问的项目列表（获取项目 ID）
 artdam projects
@@ -70,21 +73,37 @@ artdam project show      # 查看当前默认项目
 
 登录 ArtDAM，保存 token 到本地 `~/.artdam/config.json`。
 
+**默认走 SSO 浏览器授权**（推荐）：
+
 ```bash
-artdam login --username <用户名> --password <密码>
+artdam login
+# 自动打开浏览器 → 完成公司 SSO 授权 → token 写入本地配置
 ```
 
+**账号密码登录**（仅管理员 / 服务账号）：
+
+```bash
+artdam login --password --username <用户名>
+# 回车后提示输入密码（不回显）
+```
+
+| 参数 | 说明 |
+|---|---|
+| `--url` | 服务器地址（默认已内置，无需填写） |
+| `--password` | 切换为账号密码模式（不加此 flag 默认走 SSO） |
+| `--username` | 用户名（仅 `--password` 模式） |
+
 #### `artdam search`
 
 搜索资产，返回匹配列表。
 
 ```bash
-artdam search "头盔"                        # 使用默认项目
-artdam search "头盔" --project 1            # 临时指定项目
-artdam search "头盔" --limit 50
-artdam search "头盔" --type image/png
-artdam search "头盔" --folder 10
-artdam search "头盔" --json                 # 输出原始 JSON（AI Agent 用）
+artdam search "场景"                        # 使用默认项目
+artdam search "场景" --project 1            # 临时指定项目
+artdam search "场景" --limit 50
+artdam search "场景" --type image/png
+artdam search "场景" --folder 10
+artdam search "场景" --json                 # 输出原始 JSON（AI Agent 用）
 ```
 
 | 参数 | 说明 |
@@ -225,8 +244,8 @@ artdam folders --json
 ### 典型 Agent 工作流
 
 ```bash
-# 1. 登录
-artdam login --username <用户名> --password <密码>
+# 1. 登录（SSO 浏览器授权，完成后自动保存 token）
+artdam login
 
 # 2. 查询项目列表，选择目标项目并设为默认
 artdam projects --json
@@ -263,7 +282,7 @@ artdam update <asset_id> --description "AI 生成的场景描述"
 
 ```json
 {
-  "base_url": "https://artdam.dsworks.cn",
+  "base_url": "https://artdam.xindong.com",
   "token": "eyJ...",
   "default_project": 1
 }

{artdam_cli-0.2.1 → artdam_cli-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "artdam-cli"
-version = "0.2.1"
+version = "0.2.3"
 requires-python = ">=3.10"
 description = "ArtDAM 数字资产管理平台命令行工具"
 readme = "README.md"

artdam_cli-0.2.1/artdam_cli/commands/login.py

@@ -1,31 +0,0 @@
-import click
-import httpx
-
-from artdam_cli import config
-
-
-@click.command()
-@click.option("--url", default="https://artdam.dsworks.cn", help="服务器地址，默认: https://artdam.dsworks.cn")
-@click.option("--username", prompt="用户名")
-@click.option("--password", prompt="密码", hide_input=True)
-def login(url: str, username: str, password: str) -> None:
-    """登录 ArtDAM，保存 token 到本地。"""
-    url = url.rstrip("/")
-    try:
-        r = httpx.post(
-            f"{url}/api/auth/login",
-            json={"username": username, "password": password},
-            timeout=10,
-        )
-    except httpx.ConnectError:
-        raise SystemExit(f"无法连接到 {url}，请检查地址")
-
-    if r.status_code != 200:
-        raise SystemExit(f"登录失败: {r.json().get('detail', r.text)}")
-
-    token = r.json().get("access_token")
-    if not token:
-        raise SystemExit("登录响应中未找到 token")
-
-    config.save({"base_url": url, "token": token})
-    click.echo(f"✓ 登录成功，配置已保存到 ~/.artdam/config.json")