dy-cli 0.2.0__py3-none-any.whl

dy_cli/commands/init.py

@@ -0,0 +1,200 @@
+"""
+dy init — 新用户引导式初始化。
+
+自动完成: 检查环境 → 安装 Chromium → 配置代理 → 登录。
+"""
+from __future__ import annotations
+
+import os
+import sys
+import shutil
+import subprocess
+
+import click
+from rich.panel import Panel
+
+from dy_cli.utils import config
+from dy_cli.utils.output import success, error, info, warning, console, status
+
+
+@click.command("init", help="🚀 初始化设置 (新用户从这里开始)")
+@click.option("--proxy", default=None, help="代理地址 (如 http://127.0.0.1:7897)")
+@click.option("--no-proxy", is_flag=True, help="不使用代理 (在国内网络)")
+@click.option("--skip-login", is_flag=True, help="跳过登录步骤")
+@click.option("--skip-chromium", is_flag=True, help="跳过 Chromium 安装")
+def init(proxy, no_proxy, skip_login, skip_chromium):
+    """引导新用户完成初始化。"""
+    console.print()
+    console.print(Panel(
+        "[bold]欢迎使用 🎬 dy-cli — 抖音命令行工具[/]\n\n"
+        "接下来将引导你完成初始化设置:\n"
+        "  [dim]1.[/] 检查系统环境\n"
+        "  [dim]2.[/] 安装 Playwright Chromium\n"
+        "  [dim]3.[/] 配置网络\n"
+        "  [dim]4.[/] 登录抖音账号",
+        title="🚀 初始化向导",
+        border_style="blue",
+    ))
+    console.print()
+
+    # ── Step 1: 环境检查 ──────────────────────────────────
+    console.rule("[bold]Step 1/4 — 环境检查[/]")
+    console.print()
+
+    status("系统", f"{sys.platform} {os.uname().machine}")
+    status("Python", f"{sys.version.split()[0]}")
+
+    # Check playwright
+    pw_ok = _check_playwright()
+    if pw_ok:
+        status("Playwright", "✅ 已安装", "green")
+    else:
+        status("Playwright", "⚠️ 未安装", "yellow")
+
+    # Check httpx
+    try:
+        import httpx
+        status("httpx", f"✅ {httpx.__version__}", "green")
+    except ImportError:
+        status("httpx", "⚠️ 未安装", "yellow")
+
+    console.print()
+
+    # ── Step 2: 安装 Chromium ─────────────────────────────
+    console.rule("[bold]Step 2/4 — 安装 Playwright Chromium[/]")
+    console.print()
+
+    if skip_chromium:
+        info("已跳过 Chromium 安装")
+    elif pw_ok and _check_chromium():
+        success("Chromium 已安装")
+    else:
+        info("正在安装 Playwright Chromium (首次运行需要)...")
+        try:
+            subprocess.run(
+                [sys.executable, "-m", "playwright", "install", "chromium"],
+                check=True,
+                capture_output=True,
+            )
+            success("Chromium 安装完成")
+        except subprocess.CalledProcessError as e:
+            warning(f"Chromium 安装失败: {e.stderr.decode()[:200]}")
+            info("请手动运行: playwright install chromium")
+        except FileNotFoundError:
+            warning("playwright 未安装，请先运行: pip install playwright")
+
+    console.print()
+
+    # ── Step 3: 网络配置 ──────────────────────────────────
+    console.rule("[bold]Step 3/4 — 网络配置[/]")
+    console.print()
+
+    cfg = config.load_config()
+
+    if no_proxy:
+        proxy_addr = ""
+        info("不使用代理 (国内网络直连)")
+    elif proxy:
+        proxy_addr = proxy
+        info(f"使用指定代理: {proxy}")
+    else:
+        console.print("  抖音在国内可以直连，海外可能需要代理。")
+        console.print("  如果在[bold]国内[/]，直接回车跳过。")
+        console.print()
+        proxy_addr = click.prompt(
+            "  代理地址",
+            default=cfg["api"].get("proxy", ""),
+            show_default=True,
+        )
+        if proxy_addr.strip().lower() in ("none", "no", "skip", "跳过", "无", ""):
+            proxy_addr = ""
+
+    cfg["api"]["proxy"] = proxy_addr
+    config.save_config(cfg)
+    success("配置已保存")
+    console.print()
+
+    # ── Step 4: 登录 ─────────────────────────────────────
+    console.rule("[bold]Step 4/4 — 登录抖音[/]")
+    console.print()
+
+    if skip_login:
+        info("已跳过登录步骤")
+        info("稍后使用 [bold]dy login[/] 登录")
+    else:
+        from dy_cli.engines.playwright_client import PlaywrightClient, PlaywrightError
+
+        client = PlaywrightClient(headless=False)
+
+        # Check if already logged in
+        if client.cookie_exists() and client.check_login():
+            success("已登录抖音 ✅")
+        else:
+            info("即将打开浏览器，请使用抖音 App 扫码登录...")
+            console.print()
+            console.print(Panel(
+                "[bold]请使用抖音 App 扫码登录:[/]\n\n"
+                "  1. 打开抖音 App\n"
+                "  2. 点击右下角 [bold]我[/]\n"
+                "  3. 点击右上角 [bold]☰[/] → [bold]扫一扫[/]\n"
+                "  4. 扫描浏览器中的二维码\n\n"
+                "[dim]扫码后登录会自动完成，cookies 会持久保存。[/]",
+                title="📱 扫码登录",
+                border_style="green",
+            ))
+            console.print()
+
+            try:
+                ok = client.login()
+                if ok:
+                    success("登录成功! 🎉")
+                else:
+                    warning("登录超时，请稍后重试: dy login")
+            except PlaywrightError as e:
+                error(f"登录失败: {e}")
+                info("稍后使用 [bold]dy login[/] 重试")
+            except Exception as e:
+                error(f"登录失败: {e}")
+                info("请确保已安装 Chromium: playwright install chromium")
+
+    # ── 完成 ─────────────────────────────────────────────
+    console.print()
+    console.rule("[bold green]✅ 初始化完成[/]")
+    console.print()
+    console.print(Panel(
+        "[bold]🎉 你已准备就绪! 以下是常用命令:[/]\n\n"
+        "  [bold cyan]dy search[/] \"关键词\"              搜索视频\n"
+        "  [bold cyan]dy trending[/]                     抖音热榜\n"
+        "  [bold cyan]dy download[/] URL                 无水印下载\n"
+        "  [bold cyan]dy publish[/] -t 标题 -c 描述 -v 视频  发布视频\n"
+        "  [bold cyan]dy detail[/] AWEME_ID              视频详情\n"
+        "  [bold cyan]dy analytics[/]                    数据看板\n"
+        "  [bold cyan]dy me[/]                           查看我的信息\n"
+        "  [bold cyan]dy --help[/]                       查看所有命令\n\n"
+        "[dim]提示: 大部分命令支持 --json-output 输出 JSON 格式[/]",
+        title="📖 快速参考",
+        border_style="cyan",
+    ))
+    console.print()
+
+
+def _check_playwright() -> bool:
+    """检查 playwright 是否已安装。"""
+    try:
+        import playwright
+        return True
+    except ImportError:
+        return False
+
+
+def _check_chromium() -> bool:
+    """检查 Playwright Chromium 是否已安装。"""
+    try:
+        result = subprocess.run(
+            [sys.executable, "-m", "playwright", "install", "--dry-run", "chromium"],
+            capture_output=True,
+            text=True,
+        )
+        return result.returncode == 0
+    except Exception:
+        return False

dy_cli/commands/interact.py

@@ -0,0 +1,140 @@
+"""
+dy like / comment / favorite / follow — 互动命令 (Playwright)。
+"""
+from __future__ import annotations
+
+import click
+
+from dy_cli.engines.playwright_client import PlaywrightClient, PlaywrightError
+from dy_cli.utils.index_cache import resolve_id
+from dy_cli.utils.output import success, error, info, console, print_comments
+
+
+def _resolve(id_str: str) -> str:
+    try:
+        return resolve_id(id_str)
+    except ValueError as e:
+        error(str(e))
+        raise SystemExit(1)
+
+
+def _pw(account=None) -> PlaywrightClient:
+    return PlaywrightClient(account=account, headless=True)
+
+
+@click.command("like", help="点赞视频 (支持短索引: dy like 1)")
+@click.argument("aweme_id")
+@click.option("--unlike", is_flag=True, help="取消点赞")
+@click.option("--account", default=None, help="使用指定账号")
+def like(aweme_id, unlike, account):
+    """点赞或取消点赞。"""
+    aweme_id = _resolve(aweme_id)
+    action = "unlike" if unlike else "like"
+    action_cn = "取消点赞" if unlike else "点赞"
+    info(f"正在{action_cn}: {aweme_id}")
+
+    try:
+        result = _pw(account).interact(aweme_id, action)
+        if result.get("success"):
+            success(f"{action_cn}成功 👍")
+        else:
+            error(f"{action_cn}失败: 未找到按钮")
+            raise SystemExit(1)
+    except PlaywrightError as e:
+        error(f"{action_cn}失败: {e}")
+        raise SystemExit(1)
+
+
+@click.command("favorite", help="收藏视频 (支持短索引: dy fav 1)")
+@click.argument("aweme_id")
+@click.option("--unfavorite", is_flag=True, help="取消收藏")
+@click.option("--account", default=None, help="使用指定账号")
+def favorite(aweme_id, unfavorite, account):
+    """收藏或取消收藏。"""
+    aweme_id = _resolve(aweme_id)
+    action = "unfavorite" if unfavorite else "favorite"
+    action_cn = "取消收藏" if unfavorite else "收藏"
+    info(f"正在{action_cn}: {aweme_id}")
+
+    try:
+        result = _pw(account).interact(aweme_id, action)
+        if result.get("success"):
+            success(f"{action_cn}成功 ⭐")
+        else:
+            error(f"{action_cn}失败: 未找到按钮")
+            raise SystemExit(1)
+    except PlaywrightError as e:
+        error(f"{action_cn}失败: {e}")
+        raise SystemExit(1)
+
+
+@click.command("comment", help="评论视频 (支持短索引: dy comment 1 -c '好看')")
+@click.argument("aweme_id")
+@click.option("--content", "-c", required=True, help="评论内容")
+@click.option("--account", default=None, help="使用指定账号")
+def comment(aweme_id, content, account):
+    """发表评论。"""
+    aweme_id = _resolve(aweme_id)
+    info(f"正在评论: {aweme_id}")
+
+    try:
+        result = _pw(account).interact(aweme_id, "comment", content=content)
+        if result.get("success"):
+            success("评论成功 💬")
+        else:
+            error("评论失败: 未找到输入框")
+            raise SystemExit(1)
+    except PlaywrightError as e:
+        error(f"评论失败: {e}")
+        raise SystemExit(1)
+
+
+@click.command("comments", help="查看视频评论 (支持短索引)")
+@click.argument("aweme_id")
+@click.option("--count", type=int, default=20, help="评论数量")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+def comments(aweme_id, count, account, as_json):
+    """查看视频评论列表。"""
+    aweme_id = _resolve(aweme_id)
+    from dy_cli.engines.api_client import DouyinAPIClient, DouyinAPIError
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        info(f"正在获取评论: {aweme_id}")
+        data = client.get_comments(aweme_id, count=count)
+        comment_list = data.get("comments", [])
+
+        if as_json:
+            from dy_cli.utils.output import print_json
+            print_json(data)
+        else:
+            print_comments(comment_list)
+
+    except DouyinAPIError as e:
+        error(f"获取评论失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()
+
+
+@click.command("follow", help="关注用户")
+@click.argument("sec_user_id")
+@click.option("--unfollow", is_flag=True, help="取消关注")
+@click.option("--account", default=None, help="使用指定账号")
+def follow(sec_user_id, unfollow, account):
+    """关注或取消关注用户。"""
+    action = "unfollow" if unfollow else "follow"
+    action_cn = "取消关注" if unfollow else "关注"
+    info(f"正在{action_cn}用户")
+
+    try:
+        result = _pw(account).interact("", action, sec_user_id=sec_user_id)
+        if result.get("success"):
+            success(f"{action_cn}成功 👥")
+        else:
+            error(f"{action_cn}失败: 未找到按钮")
+            raise SystemExit(1)
+    except PlaywrightError as e:
+        error(f"{action_cn}失败: {e}")
+        raise SystemExit(1)

dy_cli/commands/live.py

@@ -0,0 +1,141 @@
+"""
+dy live — 直播相关命令（抖音特色功能）。
+"""
+from __future__ import annotations
+
+import os
+import subprocess
+import shutil
+
+import click
+
+from dy_cli.engines.api_client import DouyinAPIClient, DouyinAPIError
+from dy_cli.utils import config
+from dy_cli.utils.output import success, error, info, warning, console, print_json, print_live_info
+
+
+@click.group("live", help="📺 直播功能 (查看/录制)")
+def live_group():
+    pass
+
+
+@live_group.command("info", help="查看直播间信息")
+@click.argument("room_id")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+def live_info(room_id, account, as_json):
+    """查看直播间信息（观众数、主播、拉流地址）。"""
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        info(f"正在获取直播间信息: {room_id}")
+        data = client.get_live_info(room_id)
+
+        if as_json:
+            print_json(data)
+        else:
+            print_live_info(data)
+
+            # Show stream URLs if available
+            stream_data = data.get("stream_url", {})
+            if isinstance(stream_data, dict):
+                flv_url = stream_data.get("flv_pull_url", {})
+                hls_url = stream_data.get("hls_pull_url_map", {})
+                if flv_url or hls_url:
+                    console.print()
+                    info("拉流地址:")
+                    for quality, url in (flv_url or hls_url).items():
+                        console.print(f"  [{quality}] {url[:80]}…" if len(url) > 80 else f"  [{quality}] {url}")
+
+    except DouyinAPIError as e:
+        error(f"获取直播信息失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()
+
+
+@live_group.command("record", help="录制直播")
+@click.argument("room_id")
+@click.option("--output", "-o", default=None, help="输出文件路径")
+@click.option("--quality", type=click.Choice(["FULL_HD1", "HD1", "SD1", "SD2"]),
+              default="FULL_HD1", help="画质 (默认最高)")
+@click.option("--account", default=None, help="使用指定账号")
+def live_record(room_id, output, quality, account):
+    """
+    录制直播视频 (需要 ffmpeg)。
+
+    使用 ffmpeg 拉流保存，Ctrl+C 停止录制。
+    """
+    # Check ffmpeg
+    if not shutil.which("ffmpeg"):
+        error("需要安装 ffmpeg: brew install ffmpeg (macOS)")
+        raise SystemExit(1)
+
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        info(f"正在获取直播拉流地址: {room_id}")
+        data = client.get_live_info(room_id)
+
+        # Check if live
+        status_val = data.get("status")
+        if status_val != 2:
+            error("直播间未开播或已结束")
+            raise SystemExit(1)
+
+        # Get stream URL
+        stream_data = data.get("stream_url", {})
+        flv_urls = stream_data.get("flv_pull_url", {})
+        hls_urls = stream_data.get("hls_pull_url_map", {})
+
+        stream_url = None
+        urls = flv_urls or hls_urls
+        if isinstance(urls, dict):
+            # Try preferred quality, fallback to any available
+            stream_url = urls.get(quality) or next(iter(urls.values()), None)
+
+        if not stream_url:
+            error("未获取到拉流地址")
+            raise SystemExit(1)
+
+        # Output file
+        if not output:
+            cfg = config.load_config()
+            dl_dir = cfg["default"].get("download_dir", os.path.expanduser("~/Downloads/douyin"))
+            os.makedirs(dl_dir, exist_ok=True)
+            owner = data.get("owner", {}).get("nickname", room_id)
+            import datetime
+            ts = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+            output = os.path.join(dl_dir, f"live_{owner}_{ts}.mp4")
+
+        info(f"开始录制: {output}")
+        info("按 Ctrl+C 停止录制")
+        console.print()
+
+        # Use ffmpeg to record
+        cmd = [
+            "ffmpeg",
+            "-i", stream_url,
+            "-c", "copy",
+            "-movflags", "+faststart",
+            output,
+        ]
+
+        try:
+            subprocess.run(cmd, check=True)
+        except KeyboardInterrupt:
+            console.print()
+            if os.path.isfile(output):
+                size = os.path.getsize(output)
+                size_str = f"{size / 1024 / 1024:.1f}MB"
+                success(f"录制完成: {output} ({size_str})")
+            else:
+                warning("录制已取消")
+        except subprocess.CalledProcessError as e:
+            error(f"ffmpeg 录制失败: {e}")
+
+    except DouyinAPIError as e:
+        error(f"获取直播信息失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()

dy_cli/commands/profile.py

@@ -0,0 +1,78 @@
+"""
+dy me / profile — 用户信息命令。
+"""
+from __future__ import annotations
+
+import click
+
+from dy_cli.engines.api_client import DouyinAPIClient, DouyinAPIError
+from dy_cli.engines.playwright_client import PlaywrightClient, PlaywrightError
+from dy_cli.utils import config
+from dy_cli.utils.output import (
+    success, error, info, console,
+    print_user_profile, print_json, print_videos,
+)
+
+
+@click.command("me", help="查看自己的账号信息")
+@click.option("--account", default=None, help="账号名")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+def me(account, as_json):
+    """查看当前登录账号信息。"""
+    client = PlaywrightClient(account=account, headless=True)
+
+    if not client.cookie_exists():
+        error("未登录，请先运行: dy login")
+        raise SystemExit(1)
+
+    info("正在检查登录状态...")
+    try:
+        logged_in = client.check_login()
+        if logged_in:
+            success("已登录抖音 ✅")
+            console.print(f"  [bold]Cookie:[/] {client.cookie_file}")
+        else:
+            error("Cookie 已失效，请重新登录: dy login")
+            raise SystemExit(1)
+    except PlaywrightError as e:
+        error(f"检查失败: {e}")
+        raise SystemExit(1)
+
+
+@click.command("profile", help="查看用户主页")
+@click.argument("sec_user_id")
+@click.option("--posts", is_flag=True, help="同时加载作品列表")
+@click.option("--post-count", type=int, default=20, help="作品数量 (默认 20)")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+def profile(sec_user_id, posts, post_count, account, as_json):
+    """查看用户主页信息和作品。"""
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        info(f"正在获取用户资料...")
+        user = client.get_user_profile(sec_user_id)
+
+        if as_json and not posts:
+            print_json(user)
+            return
+
+        print_user_profile(user)
+
+        # Load posts
+        if posts:
+            info("正在获取作品列表...")
+            post_data = client.get_user_posts(sec_user_id, count=post_count)
+            aweme_list = post_data.get("aweme_list", [])
+
+            if as_json:
+                print_json({"user": user, "posts": aweme_list})
+            else:
+                nickname = user.get("nickname", "")
+                print_videos(aweme_list, keyword=f"{nickname} 的作品")
+
+    except DouyinAPIError as e:
+        error(f"获取用户资料失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()

dy_cli/commands/publish.py

@@ -0,0 +1,123 @@
+"""
+dy publish — 发布命令 (Playwright 引擎)。
+"""
+from __future__ import annotations
+
+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
+
+
+@click.command("publish", help="发布视频或图文到抖音")
+@click.option("--title", "-t", required=True, help="标题")
+@click.option("--content", "-c", default=None, help="描述正文")
+@click.option("--content-file", type=click.Path(exists=True), default=None, help="从文件读取描述")
+@click.option("--video", "-v", default=None, help="视频文件路径")
+@click.option("--images", "-i", multiple=True, help="图片路径 (可多个)")
+@click.option("--tags", multiple=True, help="标签 (可多个，如: --tags 旅行 --tags 美食)")
+@click.option("--visibility", type=click.Choice(["公开", "好友可见", "仅自己可见"]),
+              default="公开", help="可见范围")
+@click.option("--schedule", default=None, help="定时发布 (ISO 8601, 如 2026-03-16T10:00:00+08:00)")
+@click.option("--thumbnail", default=None, type=click.Path(exists=True), help="封面图片路径")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--headless", is_flag=True, help="无头模式 (不显示浏览器)")
+@click.option("--dry-run", is_flag=True, help="预览模式，不实际发布")
+def publish(title, content, content_file, video, images, tags, visibility, schedule, thumbnail, account, headless, dry_run):
+    """发布视频或图文。"""
+
+    # Handle content
+    if content_file:
+        with open(content_file, "r", encoding="utf-8") as f:
+            content = f.read().strip()
+    if not content:
+        content = ""
+
+    # Validate media
+    images = list(images)
+    if not images and not video:
+        error("必须提供视频 (--video) 或图片 (--images)")
+        raise SystemExit(1)
+
+    if video and images:
+        error("不能同时提供视频和图片，请选择一种")
+        raise SystemExit(1)
+
+    # Validate files
+    if video and not video.startswith("http") and not os.path.isfile(video):
+        error(f"视频文件不存在: {video}")
+        raise SystemExit(1)
+
+    for img in images:
+        if not img.startswith("http") and not os.path.isfile(img):
+            error(f"图片文件不存在: {img}")
+            raise SystemExit(1)
+
+    tags = list(tags)
+
+    # Dry run
+    if dry_run:
+        console.print()
+        info("📋 发布预览:")
+        console.print(f"  [bold]标题:[/] {title}")
+        console.print(f"  [bold]描述:[/] {content[:100]}{'...' if len(content) > 100 else ''}")
+        if video:
+            console.print(f"  [bold]视频:[/] {video}")
+        else:
+            console.print(f"  [bold]图片:[/] {', '.join(images)}")
+        if tags:
+            console.print(f"  [bold]标签:[/] {', '.join(tags)}")
+        console.print(f"  [bold]可见:[/] {visibility}")
+        if schedule:
+            console.print(f"  [bold]定时:[/] {schedule}")
+        if thumbnail:
+            console.print(f"  [bold]封面:[/] {thumbnail}")
+        console.print()
+        return
+
+    # Publish
+    cfg = config.load_config()
+    use_headless = headless or cfg["playwright"].get("headless", False)
+
+    client = PlaywrightClient(
+        account=account,
+        headless=use_headless,
+        slow_mo=cfg["playwright"].get("slow_mo", 0),
+    )
+
+    if not client.cookie_exists():
+        error("未登录，请先运行: dy login")
+        raise SystemExit(1)
+
+    try:
+        if video:
+            info(f"正在发布视频: {os.path.basename(video)}")
+            result = client.publish_video(
+                title=title,
+                content=content,
+                video_path=os.path.abspath(video),
+                tags=tags or None,
+                visibility=visibility,
+                schedule_at=schedule,
+                thumbnail_path=thumbnail,
+            )
+        else:
+            info(f"正在发布图文 ({len(images)} 张图片)")
+            result = client.publish_image_text(
+                title=title,
+                content=content,
+                images=[os.path.abspath(img) if not img.startswith("http") else img for img in images],
+                tags=tags or None,
+                visibility=visibility,
+                schedule_at=schedule,
+            )
+
+        success("发布成功! 🎉")
+        info("提示: 可用 [bold]dy search[/] 搜索验证发布状态")
+
+    except PlaywrightError as e:
+        error(f"发布失败: {e}")
+        raise SystemExit(1)