dy-cli 0.2.0__py3-none-any.whl

dy_cli/__init__.py

@@ -0,0 +1,3 @@
+"""dy-cli — 抖音命令行工具。"""
+
+__version__ = "0.2.0"

dy_cli/commands/account.py

@@ -0,0 +1,103 @@
+"""
+dy account — 多账号管理命令。
+"""
+from __future__ import annotations
+
+import os
+
+import click
+from rich.table import Table
+from rich import box
+
+from dy_cli.engines.playwright_client import PlaywrightClient
+from dy_cli.utils import config
+from dy_cli.utils.output import success, error, info, console
+
+
+@click.group("account", help="多账号管理")
+def account_group():
+    pass
+
+
+@account_group.command("list", help="列出所有账号")
+def list_accounts():
+    """列出已配置的账号。"""
+    cookies_dir = config.COOKIES_DIR
+    default_account = config.load_config()["default"]["account"]
+
+    if not os.path.isdir(cookies_dir):
+        info("暂无配置账号")
+        info("使用 [bold]dy account add <name>[/] 添加账号")
+        return
+
+    files = [f for f in os.listdir(cookies_dir) if f.endswith(".json")]
+    if not files:
+        info("暂无配置账号")
+        return
+
+    table = Table(title="📱 账号列表", box=box.ROUNDED)
+    table.add_column("名称", style="bold")
+    table.add_column("Cookie 文件")
+    table.add_column("状态")
+    table.add_column("默认", justify="center")
+
+    for f in sorted(files):
+        name = f.replace(".json", "")
+        cookie_path = os.path.join(cookies_dir, f)
+        size = os.path.getsize(cookie_path)
+        status_text = "✅ 有效" if size > 100 else "⚠️ 空"
+        is_default = "⭐" if name == default_account else ""
+        table.add_row(name, cookie_path, status_text, is_default)
+
+    console.print(table)
+
+
+@account_group.command("add", help="添加新账号并登录")
+@click.argument("name")
+def add_account(name):
+    """添加新账号并打开浏览器登录。"""
+    cookie_file = config.get_cookie_file(name)
+    if os.path.isfile(cookie_file):
+        if not click.confirm(f"账号 '{name}' 已存在，是否重新登录?", default=False):
+            return
+
+    info(f"正在为账号 '{name}' 打开登录页面...")
+    client = PlaywrightClient(account=name, headless=False)
+    try:
+        ok = client.login()
+        if ok:
+            success(f"账号 '{name}' 已添加并登录")
+        else:
+            error("登录失败")
+    except Exception as e:
+        error(f"登录失败: {e}")
+        raise SystemExit(1)
+
+
+@account_group.command("remove", help="删除账号")
+@click.argument("name")
+@click.confirmation_option(prompt="确认删除此账号?")
+def remove_account(name):
+    """删除账号 (Cookie 文件)。"""
+    cookie_file = config.get_cookie_file(name)
+    if os.path.isfile(cookie_file):
+        os.remove(cookie_file)
+        success(f"账号 '{name}' 已删除")
+    else:
+        error(f"账号 '{name}' 不存在")
+        raise SystemExit(1)
+
+
+@account_group.command("default", help="设置默认账号")
+@click.argument("name")
+def set_default(name):
+    """设置默认账号。"""
+    cookie_file = config.get_cookie_file(name)
+    if not os.path.isfile(cookie_file):
+        warning_text = f"账号 '{name}' 尚未登录"
+        info(warning_text)
+        if not click.confirm("仍要设为默认?", default=False):
+            return
+
+    config.set_value("default.account", name)
+    success(f"默认账号已设为: {name}")

dy_cli/commands/analytics.py

@@ -0,0 +1,120 @@
+"""
+dy analytics / notifications — 数据分析命令 (Playwright)。
+"""
+from __future__ import annotations
+
+import json
+import os
+
+import click
+
+from dy_cli.engines.playwright_client import PlaywrightClient, PlaywrightError
+from dy_cli.utils import config
+from dy_cli.utils.output import (
+    success, error, info, warning, console,
+    print_analytics, print_json,
+)
+
+
+@click.command("analytics", help="📊 数据看板 (创作者数据分析)")
+@click.option("--csv", "csv_file", default=None, help="导出 CSV 文件路径")
+@click.option("--page-size", type=int, default=10, help="每页条数 (默认 10)")
+@click.option("--account", default=None, help="账号名")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+def analytics(csv_file, page_size, account, as_json):
+    """获取创作者数据看板 (Playwright 引擎)。"""
+    cfg = config.load_config()
+    client = PlaywrightClient(
+        account=account,
+        headless=True,
+    )
+
+    if not client.cookie_exists():
+        error("未登录，请先运行: dy login")
+        raise SystemExit(1)
+
+    info("正在获取数据看板 (Playwright)...")
+
+    try:
+        result = client.get_analytics(page_size=page_size)
+    except PlaywrightError as e:
+        error(f"获取数据失败: {e}")
+        info("请确保已登录: dy status")
+        raise SystemExit(1)
+
+    if as_json:
+        print_json(result)
+    else:
+        print_analytics(result)
+
+    # Export CSV
+    if csv_file:
+        rows = result.get("rows", [])
+        if rows:
+            import csv
+            keys = rows[0].keys()
+            with open(csv_file, "w", newline="", encoding="utf-8-sig") as f:
+                writer = csv.DictWriter(f, fieldnames=keys)
+                writer.writeheader()
+                writer.writerows(rows)
+            success(f"CSV 已导出: {csv_file}")
+        else:
+            warning("无数据可导出")
+
+
+@click.command("notifications", help="🔔 通知消息")
+@click.option("--account", default=None, help="账号名")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+def notifications(account, as_json):
+    """获取通知消息 (Playwright 引擎)。"""
+    cfg = config.load_config()
+    client = PlaywrightClient(
+        account=account,
+        headless=True,
+    )
+
+    if not client.cookie_exists():
+        error("未登录，请先运行: dy login")
+        raise SystemExit(1)
+
+    info("正在获取通知消息...")
+
+    try:
+        result = client.get_notifications()
+    except PlaywrightError as e:
+        error(f"获取通知失败: {e}")
+        raise SystemExit(1)
+
+    if as_json:
+        print_json(result)
+    else:
+        _print_notifications(result)
+
+
+def _print_notifications(data: dict):
+    """格式化输出通知。"""
+    from rich.table import Table
+    from rich import box
+
+    mentions = data.get("mentions", [])
+    if not mentions:
+        info("暂无新通知")
+        return
+
+    table = Table(title="🔔 通知消息", box=box.ROUNDED, show_lines=True)
+    table.add_column("#", style="dim", width=3)
+    table.add_column("类型", width=8)
+    table.add_column("用户", max_width=15)
+    table.add_column("内容", max_width=40, overflow="fold")
+    table.add_column("时间", width=16)
+
+    for i, mention in enumerate(mentions, 1):
+        table.add_row(
+            str(i),
+            mention.get("type", "-"),
+            mention.get("user", "-"),
+            mention.get("content", "-"),
+            mention.get("time", "-"),
+        )
+
+    console.print(table)

dy_cli/commands/auth.py

@@ -0,0 +1,159 @@
+"""
+dy login / logout / status — 认证命令。
+
+支持两种登录方式:
+1. 浏览器 Cookie 自动提取 (默认, 零摩擦)
+2. Playwright 扫码 (--qrcode)
+"""
+from __future__ import annotations
+
+import json
+import os
+
+import click
+
+from dy_cli.engines.playwright_client import PlaywrightClient, PlaywrightError
+from dy_cli.utils import config
+from dy_cli.utils.output import success, error, info, warning, status, console
+
+
+def _extract_browser_cookies(account: str | None = None) -> bool:
+    """从浏览器自动提取抖音 Cookie（需要用户在浏览器中已登录抖音）。"""
+    try:
+        import browser_cookie3 as bc3
+    except ImportError:
+        return False
+
+    # 从多个域名收集 cookie
+    all_cookies: dict[str, dict] = {}
+    browsers = ["chrome", "firefox", "edge", "brave", "opera", "chromium", "safari"]
+    domains = [".douyin.com", "www.douyin.com", "creator.douyin.com"]
+
+    found_browser = None
+    for browser_name in browsers:
+        loader = getattr(bc3, browser_name, None)
+        if not loader:
+            continue
+        for domain in domains:
+            try:
+                jar = loader(domain_name=domain)
+                for c in jar:
+                    if "douyin" in (c.domain or ""):
+                        all_cookies[c.name] = {
+                            "name": c.name,
+                            "value": c.value,
+                            "domain": c.domain,
+                            "path": c.path or "/",
+                        }
+                        found_browser = browser_name
+            except Exception:
+                continue
+        if all_cookies:
+            break
+
+    if not all_cookies:
+        return False
+
+    # 检查是否有关键 session cookie
+    key_names = {"sessionid", "passport_csrf_token", "odin_tt", "sid_guard"}
+    has_session = bool(key_names & set(all_cookies.keys()))
+
+    if not has_session:
+        info(f"从 {found_browser} 提取了 {len(all_cookies)} 个 cookie，但缺少登录态")
+        info("请先在浏览器中登录 douyin.com，然后重试")
+        return False
+
+    # 保存为 Playwright storage_state 格式
+    cookie_file = config.get_cookie_file(account)
+    os.makedirs(os.path.dirname(cookie_file), exist_ok=True)
+    storage = {
+        "cookies": list(all_cookies.values()),
+        "origins": [],
+    }
+    with open(cookie_file, "w", encoding="utf-8") as f:
+        json.dump(storage, f, ensure_ascii=False, indent=2)
+    info(f"从 {found_browser} 提取了 {len(all_cookies)} 个 cookie (含登录态)")
+    return True
+
+
+@click.command("login", help="登录抖音")
+@click.option("--account", default=None, help="账号名")
+@click.option("--browser", is_flag=True, help="从浏览器提取 Cookie (需在浏览器中已登录抖音)")
+def login(account, browser):
+    """登录抖音。默认扫码登录，--browser 从浏览器提取 Cookie。"""
+    cfg = config.load_config()
+
+    # 已登录检查
+    client = PlaywrightClient(account=account, headless=True)
+    if client.cookie_exists():
+        try:
+            if client.check_login():
+                success("已登录抖音")
+                if not click.confirm("是否重新登录?", default=False):
+                    return
+        except Exception:
+            pass
+
+    # 方式 1: 从浏览器提取 Cookie
+    if browser:
+        info("正在从浏览器提取 Cookie...")
+        if _extract_browser_cookies(account):
+            success("登录成功! 🎉 (从浏览器提取)")
+            return
+        else:
+            warning("浏览器 Cookie 提取失败，切换到扫码模式")
+
+    # 方式 2: Playwright 扫码 (默认)
+    info("正在打开浏览器，请使用抖音 App 扫码...")
+    pw_client = PlaywrightClient(
+        account=account,
+        headless=False,
+        slow_mo=cfg["playwright"].get("slow_mo", 0),
+    )
+    try:
+        ok = pw_client.login()
+        if ok:
+            success("登录成功! 🎉")
+        else:
+            error("登录超时或失败")
+            raise SystemExit(1)
+    except PlaywrightError as e:
+        error(f"登录失败: {e}")
+        raise SystemExit(1)
+
+
+@click.command("logout", help="退出登录")
+@click.option("--account", default=None, help="账号名")
+def logout(account):
+    """退出登录（删除 Cookie）。"""
+    client = PlaywrightClient(account=account)
+    if client.logout():
+        success("已退出登录，Cookie 已删除")
+    else:
+        info("未找到登录凭据")
+
+
+@click.command("status", help="查看登录状态")
+@click.option("--account", default=None, help="账号名")
+def auth_status(account):
+    """检查登录状态。"""
+    console.print()
+    client = PlaywrightClient(account=account)
+
+    if not client.cookie_exists():
+        status("登录状态", "未登录 (无 Cookie 文件)", "red")
+        info("使用 [bold]dy login[/] 登录")
+    else:
+        info("正在验证 Cookie...")
+        try:
+            logged_in = client.check_login()
+            if logged_in:
+                status("登录状态", "已登录", "green")
+                status("Cookie", client.cookie_file, "dim")
+            else:
+                status("登录状态", "Cookie 已失效", "yellow")
+                info("使用 [bold]dy login[/] 重新登录")
+        except Exception as e:
+            status("登录状态", f"检查失败: {e}", "red")
+
+    console.print()

dy_cli/commands/config_cmd.py

@@ -0,0 +1,67 @@
+"""
+dy config — 配置管理命令。
+"""
+from __future__ import annotations
+
+import click
+
+from dy_cli.utils import config
+from dy_cli.utils.output import success, error, info, console, print_json
+
+
+@click.group("config", help="配置管理")
+def config_group():
+    pass
+
+
+@config_group.command("show", help="显示当前配置")
+def show():
+    """显示所有配置。"""
+    cfg = config.load_config()
+    print_json(cfg)
+    info(f"配置文件: {config.CONFIG_FILE}")
+
+
+@config_group.command("set", help="设置配置项")
+@click.argument("key")
+@click.argument("value")
+def set_config(key, value):
+    """
+    设置配置项。
+
+    KEY 格式: api.proxy, playwright.headless, default.engine
+
+    示例:
+        dy config set api.proxy http://127.0.0.1:7897
+        dy config set api.timeout 60
+        dy config set playwright.headless true
+        dy config set default.engine api
+        dy config set default.download_dir ~/Videos/douyin
+    """
+    # Type inference
+    if value.lower() in ("true", "false"):
+        value = value.lower() == "true"
+    elif value.isdigit():
+        value = int(value)
+
+    config.set_value(key, value)
+    success(f"已设置 {key} = {value}")
+
+
+@config_group.command("get", help="获取配置项")
+@click.argument("key")
+def get_config(key):
+    """获取单个配置项。"""
+    value = config.get(key)
+    if value is None:
+        error(f"配置项不存在: {key}")
+        raise SystemExit(1)
+    console.print(f"{key} = {value}")
+
+
+@config_group.command("reset", help="重置为默认配置")
+@click.confirmation_option(prompt="确认重置所有配置?")
+def reset():
+    """重置为默认配置。"""
+    config.save_config(config.DEFAULT_CONFIG)
+    success("配置已重置为默认值")

dy_cli/commands/download.py

@@ -0,0 +1,212 @@
+"""
+dy download — 无水印下载命令（抖音特色功能）。
+"""
+from __future__ import annotations
+
+import os
+import re
+
+import click
+from rich.progress import Progress, SpinnerColumn, BarColumn, TextColumn, DownloadColumn, TransferSpeedColumn
+
+from dy_cli.engines.api_client import DouyinAPIClient, DouyinAPIError
+from dy_cli.utils import config
+from dy_cli.utils.index_cache import resolve_id
+from dy_cli.utils.output import success, error, info, warning, console
+
+
+@click.command("download", help="下载抖音视频/图片 (无水印, 支持短索引/批量)")
+@click.argument("url_or_id")
+@click.option("--output-dir", "-o", default=None, help="保存目录 (默认 ~/Downloads/douyin)")
+@click.option("--music", is_flag=True, help="同时下载背景音乐")
+@click.option("--limit", type=int, default=0, help="批量下载: 用户作品数量 (需配合 --user)")
+@click.option("--user", is_flag=True, help="批量下载该用户的全部作品 (URL_OR_ID 为 sec_user_id)")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--json-output", "as_json", is_flag=True, help="仅输出下载链接 (JSON)")
+def download(url_or_id, output_dir, music, limit, user, account, as_json):
+    """
+    下载抖音视频/图片（无水印）。支持短索引和批量下载。
+
+    单个下载:
+      dy dl 1                          (搜索后用短索引)
+      dy dl https://v.douyin.com/xxx   (分享链接)
+      dy dl 1234567890                 (视频 ID)
+
+    批量下载用户作品:
+      dy dl SEC_USER_ID --user --limit 20
+    """
+    cfg = config.load_config()
+    output_dir = output_dir or cfg["default"].get("download_dir", os.path.expanduser("~/Downloads/douyin"))
+    os.makedirs(output_dir, exist_ok=True)
+
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        # 批量下载用户作品
+        if user:
+            _batch_download_user(client, url_or_id, output_dir, music, limit or 20, as_json)
+            return
+
+        # Resolve aweme_id (支持短索引)
+        try:
+            url_or_id = resolve_id(url_or_id)
+        except ValueError as e:
+            error(str(e))
+            raise SystemExit(1)
+        if url_or_id.isdigit():
+            aweme_id = url_or_id
+        else:
+            info("正在解析分享链接...")
+            aweme_id = client.resolve_share_url(url_or_id)
+
+        info(f"视频 ID: {aweme_id}")
+
+        # Get download info
+        info("正在获取下载链接...")
+        dl_info = client.get_download_url(aweme_id)
+
+        if as_json:
+            from dy_cli.utils.output import print_json
+            print_json(dl_info)
+            return
+
+        desc = dl_info.get("desc", "untitled")
+        author = dl_info.get("author", "unknown")
+
+        # Sanitize filename
+        safe_name = re.sub(r'[\\/:*?"<>|\n\r]', '_', desc)[:50].strip('_') or aweme_id
+        prefix = f"{author}_{safe_name}"
+
+        downloaded_files = []
+
+        # Download video
+        video_url = dl_info.get("video_url")
+        if video_url:
+            video_path = os.path.join(output_dir, f"{prefix}.mp4")
+            info(f"正在下载视频...")
+            _download_with_progress(client, video_url, video_path)
+            downloaded_files.append(video_path)
+
+        # Download images (for image posts)
+        image_urls = dl_info.get("images")
+        if image_urls:
+            for idx, img_url in enumerate(image_urls, 1):
+                img_path = os.path.join(output_dir, f"{prefix}_{idx}.jpg")
+                info(f"正在下载图片 {idx}/{len(image_urls)}...")
+                _download_with_progress(client, img_url, img_path)
+                downloaded_files.append(img_path)
+
+        # Download music
+        if music:
+            music_url = dl_info.get("music_url")
+            if music_url:
+                music_path = os.path.join(output_dir, f"{prefix}_music.mp3")
+                info(f"正在下载音乐...")
+                _download_with_progress(client, music_url, music_path)
+                downloaded_files.append(music_path)
+            else:
+                warning("未找到背景音乐")
+
+        # Summary
+        if downloaded_files:
+            console.print()
+            success(f"下载完成! ({len(downloaded_files)} 个文件)")
+            for f in downloaded_files:
+                size = os.path.getsize(f)
+                size_str = f"{size / 1024 / 1024:.1f}MB" if size > 1024 * 1024 else f"{size / 1024:.0f}KB"
+                console.print(f"  📁 {f} ({size_str})")
+        else:
+            warning("未找到可下载的内容")
+
+    except DouyinAPIError as e:
+        error(f"下载失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()
+
+
+def _batch_download_user(
+    client: DouyinAPIClient,
+    sec_user_id: str,
+    output_dir: str,
+    music: bool,
+    limit: int,
+    as_json: bool,
+):
+    """批量下载用户作品。"""
+    info(f"正在获取用户作品列表 (limit={limit})...")
+    try:
+        profile = client.get_user_profile(sec_user_id)
+        nickname = profile.get("nickname", sec_user_id)
+        info(f"用户: {nickname}")
+    except Exception:
+        nickname = sec_user_id
+
+    user_dir = os.path.join(output_dir, re.sub(r'[\\/:*?"<>|\n\r]', '_', nickname))
+    os.makedirs(user_dir, exist_ok=True)
+
+    posts = client.get_user_posts(sec_user_id, count=min(limit, 20))
+    aweme_list = posts.get("aweme_list", [])
+
+    if not aweme_list:
+        warning("未找到作品")
+        return
+
+    info(f"找到 {len(aweme_list)} 个作品，开始下载...")
+    downloaded = 0
+    import time
+
+    for i, aweme in enumerate(aweme_list[:limit], 1):
+        aweme_id = aweme.get("aweme_id", "")
+        desc = aweme.get("desc", "untitled")
+        safe = re.sub(r'[\\/:*?"<>|\n\r]', '_', desc)[:40].strip('_') or aweme_id
+
+        try:
+            dl_info = client.get_download_url(aweme_id)
+            video_url = dl_info.get("video_url")
+            if video_url:
+                path = os.path.join(user_dir, f"{i:03d}_{safe}.mp4")
+                if os.path.exists(path):
+                    info(f"[{i}/{len(aweme_list)}] 已存在，跳过: {safe[:30]}")
+                    continue
+                info(f"[{i}/{len(aweme_list)}] {safe[:30]}...")
+                _download_with_progress(client, video_url, path)
+                downloaded += 1
+
+            # Images
+            images = dl_info.get("images")
+            if images:
+                for idx, img_url in enumerate(images, 1):
+                    path = os.path.join(user_dir, f"{i:03d}_{safe}_{idx}.jpg")
+                    client.download_file(img_url, path)
+                downloaded += 1
+
+            time.sleep(1)  # Rate limit
+        except Exception as e:
+            warning(f"[{i}] 下载失败: {e}")
+            continue
+
+    console.print()
+    success(f"批量下载完成! {downloaded}/{len(aweme_list)} 个作品")
+    console.print(f"  📁 {user_dir}")
+
+
+def _download_with_progress(client: DouyinAPIClient, url: str, output_path: str):
+    """带进度条的下载。"""
+    with Progress(
+        SpinnerColumn(),
+        TextColumn("[bold blue]{task.description}"),
+        BarColumn(),
+        DownloadColumn(),
+        TransferSpeedColumn(),
+        console=console,
+    ) as progress:
+        task = progress.add_task(os.path.basename(output_path), total=None)
+
+        def on_progress(downloaded: int, total: int):
+            if total > 0:
+                progress.update(task, total=total, completed=downloaded)
+            else:
+                progress.update(task, completed=downloaded)
+
+        client.download_file(url, output_path, progress_callback=on_progress)