artdam-cli 0.1.0__tar.gz

artdam_cli-0.1.0/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: artdam-cli
+Version: 0.1.0
+Requires-Python: >=3.10
+Requires-Dist: click>=8.1
+Requires-Dist: httpx>=0.27
+Requires-Dist: rich>=13.0

artdam_cli-0.1.0/artdam_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

artdam_cli-0.1.0/artdam_cli/client.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+import httpx
+
+from artdam_cli import config
+
+
+def _make_client() -> httpx.Client:
+    base_url = config.require("base_url")
+    token = config.require("token")
+    return httpx.Client(
+        base_url=base_url,
+        headers={"Authorization": f"Bearer {token}"},
+        timeout=30,
+    )
+
+
+def get(path: str, params: dict | None = None) -> dict:
+    with _make_client() as c:
+        r = c.get(path, params=params)
+        _raise(r)
+        return r.json()
+
+
+def download_file(path: str, dest: str) -> int:
+    base_url = config.require("base_url")
+    token = config.require("token")
+    with httpx.stream(
+        "GET",
+        f"{base_url}{path}",
+        headers={"Authorization": f"Bearer {token}"},
+        timeout=120,
+        follow_redirects=True,
+    ) as r:
+        _raise(r)
+        total = 0
+        with open(dest, "wb") as f:
+            for chunk in r.iter_bytes(65536):
+                f.write(chunk)
+                total += len(chunk)
+    return total
+
+
+def _raise(r: httpx.Response) -> None:
+    if r.status_code == 401:
+        raise SystemExit("Token 无效或已过期，请重新运行 artdam login")
+    if r.status_code == 404:
+        raise SystemExit("资源不存在")
+    if r.is_error:
+        raise SystemExit(f"请求失败 {r.status_code}: {r.text[:200]}")

artdam_cli-0.1.0/artdam_cli/commands/download.py

@@ -0,0 +1,28 @@
+import os
+
+import click
+from rich.console import Console
+from rich.progress import Progress
+
+from artdam_cli import client
+
+console = Console()
+
+
+@click.command()
+@click.argument("asset_id", type=int)
+@click.option("--out", "-o", default=".", help="下载目录（默认当前目录）")
+def download(asset_id: int, out: str) -> None:
+    """下载资产文件。"""
+    data = client.get(f"/api/assets/{asset_id}")
+    filename = data.get("filename", f"asset_{asset_id}")
+    dest = os.path.join(out, filename)
+    os.makedirs(out, exist_ok=True)
+
+    with Progress() as progress:
+        task = progress.add_task(f"下载 {filename}...", total=None)
+        total = client.download_file(f"/api/assets/{asset_id}/file", dest)
+        progress.update(task, completed=total, total=total)
+
+    size_str = f"{total / 1024 / 1024:.1f} MB"
+    console.print(f"✓ 已下载到 [green]{dest}[/green]（{size_str}）")

artdam_cli-0.1.0/artdam_cli/commands/folders.py

@@ -0,0 +1,37 @@
+import json
+
+import click
+from rich.console import Console
+from rich.tree import Tree
+
+from artdam_cli import client
+
+console = Console()
+
+
+def _build_tree(node: Tree, items: list[dict], parent_id: int | None) -> None:
+    children = [i for i in items if i.get("parent_id") == parent_id]
+    for child in children:
+        branch = node.add(f"[cyan]{child['name']}[/cyan] [dim](id={child['id']})[/dim]")
+        _build_tree(branch, items, child["id"])
+
+
+@click.command()
+@click.option("--project", "-p", required=True, type=int, help="项目 ID")
+@click.option("--json", "as_json", is_flag=True, help="输出原始 JSON")
+def folders(project: int, as_json: bool) -> None:
+    """查看文件夹结构。"""
+    data = client.get("/api/collections/", params={"project_id": project})
+    items = data if isinstance(data, list) else data.get("items", [])
+
+    if as_json:
+        click.echo(json.dumps(items, ensure_ascii=False, indent=2))
+        return
+
+    if not items:
+        console.print("[yellow]暂无文件夹[/yellow]")
+        return
+
+    tree = Tree(f"[bold]项目 {project} 文件夹[/bold]")
+    _build_tree(tree, items, None)
+    console.print(tree)

artdam_cli-0.1.0/artdam_cli/commands/get.py

@@ -0,0 +1,39 @@
+import json
+
+import click
+from rich.console import Console
+from rich.panel import Panel
+from rich.text import Text
+
+from artdam_cli import client
+
+console = Console()
+
+
+@click.command()
+@click.argument("asset_id", type=int)
+@click.option("--json", "as_json", is_flag=True, help="输出原始 JSON（AI Agent 用）")
+def get(asset_id: int, as_json: bool) -> None:
+    """查看资产详情。"""
+    data = client.get(f"/api/assets/{asset_id}")
+
+    if as_json:
+        click.echo(json.dumps(data, ensure_ascii=False, indent=2))
+        return
+
+    tags = ", ".join(t["name"] for t in data.get("tags", []))
+    size = data.get("file_size") or 0
+    size_str = f"{size / 1024 / 1024:.1f} MB" if size > 0 else "—"
+
+    lines = [
+        f"[bold]ID[/bold]          {data['id']}",
+        f"[bold]文件名[/bold]      {data.get('filename', '—')}",
+        f"[bold]类型[/bold]        {data.get('mime_type', '—')}",
+        f"[bold]大小[/bold]        {size_str}",
+        f"[bold]标签[/bold]        {tags or '—'}",
+        f"[bold]AI 描述[/bold]     {data.get('ai_description') or '—'}",
+        f"[bold]上传时间[/bold]    {data.get('created_at', '—')}",
+        f"[bold]路径[/bold]        {data.get('original_path', '—')}",
+    ]
+
+    console.print(Panel("\n".join(lines), title=f"资产 #{asset_id}"))

artdam_cli-0.1.0/artdam_cli/commands/login.py

@@ -0,0 +1,31 @@
+import click
+import httpx
+
+from artdam_cli import config
+
+
+@click.command()
+@click.option("--url", prompt="ArtDAM 地址", help="例: http://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")

artdam_cli-0.1.0/artdam_cli/commands/search.py

@@ -0,0 +1,61 @@
+import json
+
+import click
+from rich.console import Console
+from rich.table import Table
+
+from artdam_cli import client
+
+console = Console()
+
+
+@click.command()
+@click.argument("keyword")
+@click.option("--project", "-p", required=True, type=int, help="项目 ID")
+@click.option("--limit", "-n", default=20, show_default=True, help="返回数量上限")
+@click.option("--type", "mime", default=None, help="文件类型，例: image/png")
+@click.option("--folder", default=None, type=int, help="限定文件夹 ID")
+@click.option("--json", "as_json", is_flag=True, help="输出原始 JSON（AI Agent 用）")
+def search(keyword: str, project: int, limit: int, mime: str | None, folder: int | None, as_json: bool) -> None:
+    """搜索资产。"""
+    params: dict = {
+        "project_id": project,
+        "q": keyword,
+        "page_size": limit,
+    }
+    if mime:
+        params["mime_type"] = mime
+    if folder:
+        params["folder_id"] = folder
+
+    data = client.get("/api/assets/", params=params)
+    items = data.get("items", [])
+
+    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"搜索「{keyword}」— 共 {data.get('total', len(items))} 条")
+    table.add_column("ID", style="dim", width=6)
+    table.add_column("文件名", min_width=20)
+    table.add_column("类型", width=10)
+    table.add_column("大小", width=10)
+    table.add_column("标签", min_width=20)
+
+    for item in items:
+        tags = ", ".join(t["name"] for t in item.get("tags", []))
+        size = item.get("file_size") or 0
+        size_str = f"{size / 1024 / 1024:.1f} MB" if size > 0 else "—"
+        table.add_row(
+            str(item["id"]),
+            item.get("filename", ""),
+            (item.get("mime_type") or "").split("/")[-1],
+            size_str,
+            tags or "—",
+        )
+
+    console.print(table)

artdam_cli-0.1.0/artdam_cli/commands/tags.py

@@ -0,0 +1,46 @@
+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("--project", "-p", required=True, type=int, help="项目 ID")
+@click.option("--search", "-s", default=None, help="按名称过滤")
+@click.option("--json", "as_json", is_flag=True, help="输出原始 JSON")
+def tags(project: int, search: str | None, as_json: bool) -> None:
+    """列出项目标签。"""
+    params: dict = {"project_id": project}
+    if search:
+        data = client.get("/api/tags/search", params={"project_id": project, "q": search})
+    else:
+        data = client.get("/api/tags/", params=params)
+
+    items = data if isinstance(data, list) else data.get("items", [])
+
+    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"项目 {project} 标签列表（{len(items)} 个）")
+    table.add_column("ID", style="dim", width=6)
+    table.add_column("标签名", min_width=20)
+    table.add_column("资产数", width=8)
+
+    for item in items:
+        table.add_row(
+            str(item["id"]),
+            item.get("name", ""),
+            str(item.get("asset_count", "—")),
+        )
+
+    console.print(table)

artdam_cli-0.1.0/artdam_cli/config.py

@@ -0,0 +1,28 @@
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+_CONFIG_PATH = Path.home() / ".artdam" / "config.json"
+
+
+def load() -> dict:
+    if not _CONFIG_PATH.exists():
+        return {}
+    return json.loads(_CONFIG_PATH.read_text(encoding="utf-8"))
+
+
+def save(data: dict) -> None:
+    _CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
+    _CONFIG_PATH.write_text(json.dumps(data, indent=2), encoding="utf-8")
+
+
+def get(key: str) -> str | None:
+    return load().get(key)
+
+
+def require(key: str) -> str:
+    value = get(key)
+    if not value:
+        raise SystemExit(f"未配置 {key}，请先运行: artdam login")
+    return value

artdam_cli-0.1.0/artdam_cli/main.py

@@ -0,0 +1,22 @@
+import click
+
+from artdam_cli.commands.login import login
+from artdam_cli.commands.search import search
+from artdam_cli.commands.get import get
+from artdam_cli.commands.tags import tags
+from artdam_cli.commands.folders import folders
+from artdam_cli.commands.download import download
+
+
+@click.group()
+@click.version_option(package_name="artdam-cli")
+def cli() -> None:
+    """ArtDAM 命令行工具 — 搜索、查看、下载数字资产。"""
+
+
+cli.add_command(login)
+cli.add_command(search)
+cli.add_command(get)
+cli.add_command(tags)
+cli.add_command(folders)
+cli.add_command(download)

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

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: artdam-cli
+Version: 0.1.0
+Requires-Python: >=3.10
+Requires-Dist: click>=8.1
+Requires-Dist: httpx>=0.27
+Requires-Dist: rich>=13.0

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

@@ -0,0 +1,18 @@
+pyproject.toml
+artdam_cli/__init__.py
+artdam_cli/client.py
+artdam_cli/config.py
+artdam_cli/main.py
+artdam_cli.egg-info/PKG-INFO
+artdam_cli.egg-info/SOURCES.txt
+artdam_cli.egg-info/dependency_links.txt
+artdam_cli.egg-info/entry_points.txt
+artdam_cli.egg-info/requires.txt
+artdam_cli.egg-info/top_level.txt
+artdam_cli/commands/__init__.py
+artdam_cli/commands/download.py
+artdam_cli/commands/folders.py
+artdam_cli/commands/get.py
+artdam_cli/commands/login.py
+artdam_cli/commands/search.py
+artdam_cli/commands/tags.py

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

@@ -0,0 +1 @@
+

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

@@ -0,0 +1,2 @@
+[console_scripts]
+artdam = artdam_cli.main:cli

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

@@ -0,0 +1,3 @@
+click>=8.1
+httpx>=0.27
+rich>=13.0

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

@@ -0,0 +1 @@
+artdam_cli

artdam_cli-0.1.0/pyproject.toml

@@ -0,0 +1,20 @@
+[build-system]
+requires = ["setuptools>=42"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "artdam-cli"
+version = "0.1.0"
+requires-python = ">=3.10"
+dependencies = [
+    "click>=8.1",
+    "httpx>=0.27",
+    "rich>=13.0",
+]
+
+[project.scripts]
+artdam = "artdam_cli.main:cli"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["artdam_cli*"]

artdam_cli-0.1.0/setup.cfg

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