dy-cli 0.2.0__py3-none-any.whl

dy_cli/main.py

@@ -0,0 +1,144 @@
+"""
+dy — 抖音命令行工具主入口。
+
+Usage:
+    dy search "关键词"                     搜索视频
+    dy trending                            抖音热榜
+    dy download URL                        无水印下载
+    dy publish -t 标题 -c 描述 -v 视频     发布视频
+    dy detail AWEME_ID                     视频详情
+    dy comments AWEME_ID                   查看评论
+    dy like AWEME_ID                       点赞
+    dy comment AWEME_ID -c "内容"          评论
+    dy favorite AWEME_ID                   收藏
+    dy follow SEC_USER_ID                  关注
+    dy live info ROOM_ID                   直播信息
+    dy live record ROOM_ID                 录制直播
+    dy analytics                           数据看板
+    dy notifications                       通知消息
+    dy me                                  我的信息
+    dy profile SEC_USER_ID                 用户主页
+    dy login                               登录
+    dy logout                              退出登录
+    dy status                              登录状态
+    dy account list|add|remove|default     多账号管理
+    dy config show|set|get|reset           配置管理
+"""
+from __future__ import annotations
+
+import click
+
+from dy_cli import __version__
+
+
+BANNER = r"""
+  ╔═══════════════════════════════╗
+  ║   🎬 dy-cli v{version}         ║
+  ║   抖音命令行工具              ║
+  ╚═══════════════════════════════╝
+""".format(version=__version__)
+
+
+class AliasGroup(click.Group):
+    """支持命令别名的 Click Group。"""
+
+    ALIASES = {
+        "pub": "publish",
+        "s": "search",
+        "dl": "download",
+        "t": "trending",
+        "r": "detail",
+        "read": "detail",
+        "fav": "favorite",
+        "noti": "notifications",
+        "stat": "status",
+        "acc": "account",
+        "cfg": "config",
+    }
+
+    def get_command(self, ctx, cmd_name):
+        resolved = self.ALIASES.get(cmd_name, cmd_name)
+        return super().get_command(ctx, resolved)
+
+    def format_help(self, ctx, formatter):
+        formatter.write(BANNER)
+        super().format_help(ctx, formatter)
+
+
+@click.group(cls=AliasGroup, invoke_without_command=True)
+@click.version_option(version=__version__, prog_name="dy-cli")
+@click.pass_context
+def cli(ctx):
+    """🎬 抖音命令行工具 — 搜索、下载、发布、互动、热榜、直播、数据分析"""
+    if ctx.invoked_subcommand is None:
+        click.echo(ctx.get_help())
+
+
+# ------------------------------------------------------------------
+# 注册所有命令
+# ------------------------------------------------------------------
+
+# 初始化
+from dy_cli.commands.init import init
+cli.add_command(init)
+
+# 认证
+from dy_cli.commands.auth import login, logout, auth_status
+cli.add_command(login)
+cli.add_command(logout)
+cli.add_command(auth_status, "status")
+
+# 搜索 & 详情
+from dy_cli.commands.search import search, detail
+cli.add_command(search)
+cli.add_command(detail)
+
+# 下载
+from dy_cli.commands.download import download
+cli.add_command(download)
+
+# 发布
+from dy_cli.commands.publish import publish
+cli.add_command(publish)
+
+# 互动
+from dy_cli.commands.interact import like, favorite, comment, comments, follow
+cli.add_command(like)
+cli.add_command(favorite)
+cli.add_command(comment)
+cli.add_command(comments)
+cli.add_command(follow)
+
+# 热榜
+from dy_cli.commands.trending import trending
+cli.add_command(trending)
+
+# 直播
+from dy_cli.commands.live import live_group
+cli.add_command(live_group, "live")
+
+# 数据分析
+from dy_cli.commands.analytics import analytics, notifications
+cli.add_command(analytics)
+cli.add_command(notifications)
+
+# 用户
+from dy_cli.commands.profile import me, profile
+cli.add_command(me)
+cli.add_command(profile)
+
+# 账号管理
+from dy_cli.commands.account import account_group
+cli.add_command(account_group, "account")
+
+# 配置管理
+from dy_cli.commands.config_cmd import config_group
+cli.add_command(config_group, "config")
+
+
+def main():
+    cli()
+
+
+if __name__ == "__main__":
+    main()

dy_cli/utils/config.py

@@ -0,0 +1,99 @@
+"""
+全局配置管理 — ~/.dy/config.json
+"""
+from __future__ import annotations
+
+import json
+import os
+from typing import Any
+
+CONFIG_DIR = os.path.expanduser("~/.dy")
+CONFIG_FILE = os.path.join(CONFIG_DIR, "config.json")
+COOKIES_DIR = os.path.join(CONFIG_DIR, "cookies")
+
+DEFAULT_CONFIG: dict[str, Any] = {
+    "api": {
+        "cookie_file": os.path.join(COOKIES_DIR, "default.json"),
+        "proxy": "",
+        "timeout": 30,
+    },
+    "playwright": {
+        "headless": False,
+        "chromium_path": "",
+        "slow_mo": 0,
+    },
+    "default": {
+        "account": "default",
+        "engine": "auto",       # auto | api | playwright
+        "output": "table",      # table | json
+        "download_dir": os.path.expanduser("~/Downloads/douyin"),
+    },
+}
+
+
+def _ensure_dir():
+    os.makedirs(CONFIG_DIR, exist_ok=True)
+    os.makedirs(COOKIES_DIR, exist_ok=True)
+
+
+def load_config() -> dict[str, Any]:
+    """加载配置文件，不存在则返回默认配置。"""
+    if os.path.exists(CONFIG_FILE):
+        try:
+            with open(CONFIG_FILE, "r", encoding="utf-8") as f:
+                user_cfg = json.load(f)
+            return _deep_merge(DEFAULT_CONFIG, user_cfg)
+        except Exception:
+            pass
+    return dict(DEFAULT_CONFIG)
+
+
+def save_config(cfg: dict[str, Any]):
+    """保存配置到文件。"""
+    _ensure_dir()
+    with open(CONFIG_FILE, "w", encoding="utf-8") as f:
+        json.dump(cfg, f, ensure_ascii=False, indent=2)
+
+
+def get(key_path: str, default: Any = None) -> Any:
+    """获取嵌套配置值。key_path 格式: 'api.proxy', 'default.engine'"""
+    cfg = load_config()
+    keys = key_path.split(".")
+    current = cfg
+    for k in keys:
+        if isinstance(current, dict) and k in current:
+            current = current[k]
+        else:
+            return default
+    return current
+
+
+def set_value(key_path: str, value: Any):
+    """设置嵌套配置值。"""
+    cfg = load_config()
+    keys = key_path.split(".")
+    current = cfg
+    for k in keys[:-1]:
+        if k not in current or not isinstance(current[k], dict):
+            current[k] = {}
+        current = current[k]
+    current[keys[-1]] = value
+    save_config(cfg)
+
+
+def get_cookie_file(account: str | None = None) -> str:
+    """获取指定账号的 Cookie 文件路径。"""
+    _ensure_dir()
+    account = account or load_config()["default"]["account"]
+    return os.path.join(COOKIES_DIR, f"{account}.json")
+
+
+def _deep_merge(base: dict, override: dict) -> dict:
+    """递归合并字典。"""
+    result = dict(base)
+    for k, v in override.items():
+        if k in result and isinstance(result[k], dict) and isinstance(v, dict):
+            result[k] = _deep_merge(result[k], v)
+        else:
+            result[k] = v
+    return result

dy_cli/utils/envelope.py

@@ -0,0 +1,49 @@
+"""
+统一输出信封 — Agent 友好的结构化输出。
+
+所有 --json / --yaml 输出使用此信封格式:
+  成功: {ok: true, schema_version: "1", data: ...}
+  失败: {ok: false, schema_version: "1", error: {code: ..., message: ...}}
+"""
+from __future__ import annotations
+
+import json
+import sys
+from typing import Any
+
+SCHEMA_VERSION = "1"
+
+
+def success_envelope(data: Any) -> dict:
+    return {"ok": True, "schema_version": SCHEMA_VERSION, "data": data}
+
+
+def error_envelope(code: str, message: str) -> dict:
+    return {
+        "ok": False,
+        "schema_version": SCHEMA_VERSION,
+        "error": {"code": code, "message": message},
+    }
+
+
+def emit(envelope: dict, fmt: str = "auto") -> None:
+    """输出信封到 stdout。
+
+    fmt: "json" | "yaml" | "auto"
+    auto: TTY → 不输出(由 Rich 处理), 非 TTY → yaml
+    """
+    if fmt == "auto":
+        if not sys.stdout.isatty():
+            fmt = "yaml"
+        else:
+            return  # TTY 模式由 Rich 处理
+
+    if fmt == "yaml":
+        try:
+            import yaml
+            sys.stdout.write(yaml.dump(envelope, allow_unicode=True, default_flow_style=False, sort_keys=False))
+        except ImportError:
+            # fallback to json if pyyaml not installed
+            sys.stdout.write(json.dumps(envelope, ensure_ascii=False, indent=2) + "\n")
+    else:
+        sys.stdout.write(json.dumps(envelope, ensure_ascii=False, indent=2) + "\n")

dy_cli/utils/export.py

@@ -0,0 +1,68 @@
+"""
+导出工具 — 将列表数据导出为 JSON / CSV 文件。
+"""
+from __future__ import annotations
+
+import csv
+import json
+import os
+from typing import Any
+
+from dy_cli.utils.output import success, info
+
+
+def export_data(data: list[dict], output_path: str) -> None:
+    """根据文件扩展名导出数据。"""
+    os.makedirs(os.path.dirname(output_path) or ".", exist_ok=True)
+    ext = os.path.splitext(output_path)[1].lower()
+
+    if ext == ".csv":
+        _export_csv(data, output_path)
+    elif ext in (".json", ".jsonl"):
+        _export_json(data, output_path)
+    elif ext in (".yaml", ".yml"):
+        _export_yaml(data, output_path)
+    else:
+        _export_json(data, output_path)
+
+    success(f"已导出 {len(data)} 条到 {output_path}")
+
+
+def _export_json(data: list[dict], path: str) -> None:
+    with open(path, "w", encoding="utf-8") as f:
+        json.dump(data, f, ensure_ascii=False, indent=2)
+
+
+def _export_csv(data: list[dict], path: str) -> None:
+    if not data:
+        return
+    # Flatten nested dicts for CSV
+    flat = [_flatten(item) for item in data]
+    keys = list(flat[0].keys())
+    with open(path, "w", newline="", encoding="utf-8-sig") as f:
+        writer = csv.DictWriter(f, fieldnames=keys, extrasaction="ignore")
+        writer.writeheader()
+        writer.writerows(flat)
+
+
+def _export_yaml(data: list[dict], path: str) -> None:
+    try:
+        import yaml
+        with open(path, "w", encoding="utf-8") as f:
+            yaml.dump(data, f, allow_unicode=True, default_flow_style=False)
+    except ImportError:
+        _export_json(data, path)
+
+
+def _flatten(d: dict, prefix: str = "") -> dict:
+    """将嵌套 dict 展平为单层 (用于 CSV)。"""
+    items: dict[str, Any] = {}
+    for k, v in d.items():
+        key = f"{prefix}{k}" if not prefix else f"{prefix}.{k}"
+        if isinstance(v, dict):
+            items.update(_flatten(v, key))
+        elif isinstance(v, list):
+            items[key] = str(v)[:100]
+        else:
+            items[key] = v
+    return items

dy_cli/utils/index_cache.py

@@ -0,0 +1,83 @@
+"""
+短索引导航 — 列表命令自动缓存结果，后续命令用数字引用。
+
+用法:
+  dy search "美食"     →  缓存搜索结果
+  dy read 1            →  读取第 1 条
+  dy download 3        →  下载第 3 条
+  dy detail 2          →  查看第 2 条详情
+"""
+from __future__ import annotations
+
+import json
+import os
+from typing import Any
+
+from dy_cli.utils.config import CONFIG_DIR
+
+INDEX_FILE = os.path.join(CONFIG_DIR, "index_cache.json")
+
+
+def save_index(items: list[dict[str, str]]) -> None:
+    """保存列表结果的索引。每条记录需要 aweme_id。"""
+    os.makedirs(os.path.dirname(INDEX_FILE), exist_ok=True)
+    entries = []
+    for item in items:
+        aweme_id = item.get("aweme_id", "")
+        if aweme_id:
+            entries.append({
+                "aweme_id": str(aweme_id),
+                "desc": item.get("desc", "")[:60],
+                "author": item.get("author", {}).get("nickname", "") if isinstance(item.get("author"), dict) else str(item.get("author", "")),
+                "sec_uid": item.get("author", {}).get("sec_uid", "") if isinstance(item.get("author"), dict) else "",
+            })
+    with open(INDEX_FILE, "w", encoding="utf-8") as f:
+        json.dump(entries, f, ensure_ascii=False, indent=2)
+
+
+def get_by_index(index: int) -> dict[str, str] | None:
+    """用 1-based 索引获取缓存条目。"""
+    if index <= 0:
+        return None
+    if not os.path.isfile(INDEX_FILE):
+        return None
+    try:
+        with open(INDEX_FILE, "r", encoding="utf-8") as f:
+            data = json.load(f)
+        if not isinstance(data, list) or index > len(data):
+            return None
+        return data[index - 1]
+    except (json.JSONDecodeError, OSError):
+        return None
+
+
+def resolve_id(id_or_index: str) -> str:
+    """解析 aweme_id: 短数字(≤999)视为索引，长数字视为 ID，含字母/URL 原样返回。"""
+    if not id_or_index.isdigit():
+        return id_or_index
+
+    n = int(id_or_index)
+    # 短索引: 1-999; aweme_id 通常 > 15 位
+    if n <= 999:
+        entry = get_by_index(n)
+        if entry:
+            return entry["aweme_id"]
+        # 索引不存在时给出明确提示，不要把 "1" 当 aweme_id
+        count = get_index_count()
+        if count == 0:
+            raise ValueError("没有缓存的搜索结果，请先执行 dy search")
+        raise ValueError(f"索引 {n} 超出范围 (共 {count} 条)")
+
+    return id_or_index
+
+
+def get_index_count() -> int:
+    """获取缓存的条目数量。"""
+    if not os.path.isfile(INDEX_FILE):
+        return 0
+    try:
+        with open(INDEX_FILE, "r", encoding="utf-8") as f:
+            data = json.load(f)
+        return len(data) if isinstance(data, list) else 0
+    except (json.JSONDecodeError, OSError):
+        return 0