dy-cli 0.2.0__py3-none-any.whl

dy_cli/commands/search.py

@@ -0,0 +1,131 @@
+"""
+dy search / detail — 搜索和详情命令。
+"""
+from __future__ import annotations
+
+import json
+
+import click
+
+from dy_cli.engines.api_client import DouyinAPIClient, DouyinAPIError
+from dy_cli.utils.index_cache import save_index, resolve_id
+from dy_cli.utils.export import export_data
+from dy_cli.utils.output import (
+    success, error, info, warning, console,
+    print_videos, print_json, print_video_detail, print_comments,
+)
+
+
+SORT_MAP = {
+    "综合": 0,
+    "最多点赞": 1,
+    "最新发布": 2,
+}
+
+TIME_MAP = {
+    "不限": 0,
+    "一天内": 1,
+    "一周内": 7,
+    "半年内": 182,
+}
+
+
+@click.command("search", help="搜索抖音视频")
+@click.argument("keyword")
+@click.option("--sort", type=click.Choice(["综合", "最多点赞", "最新发布"]),
+              default="综合", help="排序方式")
+@click.option("--time", "pub_time", type=click.Choice(["不限", "一天内", "一周内", "半年内"]),
+              default="不限", help="发布时间")
+@click.option("--type", "search_type", type=click.Choice(["general", "video", "user"]),
+              default="general", help="搜索类型")
+@click.option("--count", type=int, default=20, help="结果数量 (默认 20)")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON 格式")
+@click.option("-o", "--output", default=None, help="导出到文件 (.json/.csv/.yaml)")
+def search(keyword, sort, pub_time, search_type, count, account, as_json, output):
+    """搜索抖音视频/用户。"""
+    client = DouyinAPIClient.from_config(account)
+    info(f"正在搜索: {keyword}")
+
+    try:
+        result = client.search(
+            keyword=keyword,
+            sort_type=SORT_MAP.get(sort, 0),
+            publish_time=TIME_MAP.get(pub_time, 0),
+            search_type=search_type,
+            count=count,
+        )
+    except DouyinAPIError as e:
+        error(f"搜索失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()
+
+    if as_json:
+        print_json(result)
+        return
+
+    # Extract video list
+    data_list = result.get("data", [])
+    videos = []
+    for item in data_list:
+        aweme_info = item.get("aweme_info")
+        if aweme_info:
+            videos.append(aweme_info)
+
+    # 缓存索引 — 支持 dy read 1 / dy download 3
+    save_index(videos)
+
+    # 导出
+    if output:
+        export_data(videos, output)
+        return
+
+    print_videos(videos, keyword=keyword)
+
+
+@click.command("detail", help="查看视频详情 (支持短索引: dy detail 1)")
+@click.argument("aweme_id")
+@click.option("--comments", is_flag=True, help="同时加载评论")
+@click.option("--comment-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 detail(aweme_id, comments, comment_count, account, as_json):
+    """查看视频详情和评论。支持短索引 (dy search → dy detail 1)。"""
+    try:
+        aweme_id = resolve_id(aweme_id)
+    except ValueError as e:
+        error(str(e))
+        raise SystemExit(1)
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        info(f"正在获取详情: {aweme_id}")
+        video_detail = client.get_video_detail(aweme_id)
+
+        if as_json and not comments:
+            print_json(video_detail)
+            return
+
+        print_video_detail(video_detail)
+
+        # Load comments if requested
+        if comments:
+            info("正在加载评论...")
+            try:
+                comment_data = client.get_comments(aweme_id, count=comment_count)
+                comment_list = comment_data.get("comments", [])
+
+                if as_json:
+                    print_json({"detail": video_detail, "comments": comment_list})
+                else:
+                    print_comments(comment_list)
+            except DouyinAPIError as e:
+                warning(f"评论加载失败: {e}")
+                info("评论 API 需要签名，可单独用 [bold]dy comments[/] 尝试")
+
+    except DouyinAPIError as e:
+        error(f"获取详情失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()

dy_cli/commands/trending.py

@@ -0,0 +1,82 @@
+"""
+dy trending — 抖音热榜命令（抖音特色功能）。
+"""
+from __future__ import annotations
+
+import time
+
+import click
+
+from dy_cli.engines.api_client import DouyinAPIClient, DouyinAPIError
+from dy_cli.utils.export import export_data
+from dy_cli.utils.output import success, error, info, warning, console, print_trending, print_json
+
+
+@click.command("trending", help="🔥 抖音热榜")
+@click.option("--count", type=int, default=50, help="显示条数 (默认 50)")
+@click.option("--watch", is_flag=True, help="实时刷新模式 (每 5 分钟更新)")
+@click.option("--account", default=None, help="使用指定账号")
+@click.option("--json-output", "as_json", is_flag=True, help="输出 JSON")
+@click.option("-o", "--output", default=None, help="导出到文件 (.json/.csv/.yaml)")
+def trending(count, watch, account, as_json, output):
+    """查看抖音热榜。"""
+    client = DouyinAPIClient.from_config(account)
+
+    try:
+        if watch:
+            _watch_trending(client, count, as_json)
+        else:
+            _show_trending(client, count, as_json, output)
+    except KeyboardInterrupt:
+        info("已退出热榜监控")
+    except DouyinAPIError as e:
+        error(f"获取热榜失败: {e}")
+        raise SystemExit(1)
+    finally:
+        client.close()
+
+
+def _show_trending(client: DouyinAPIClient, count: int, as_json: bool, output: str = None):
+    """显示热榜。"""
+    info("正在获取抖音热榜...")
+    items = client.get_trending()
+
+    if as_json:
+        print_json(items[:count])
+        return
+
+    if output:
+        export_data(items[:count], output)
+        return
+
+    print_trending(items[:count])
+
+    if items:
+        console.print()
+        info(f"共 {len(items)} 条热搜，显示前 {min(count, len(items))} 条")
+
+
+def _watch_trending(client: DouyinAPIClient, count: int, as_json: bool):
+    """实时刷新热榜。"""
+    interval = 300  # 5 minutes
+    info(f"热榜监控模式 (每 {interval // 60} 分钟刷新，Ctrl+C 退出)")
+    console.print()
+
+    while True:
+        try:
+            items = client.get_trending()
+            if as_json:
+                print_json(items[:count])
+            else:
+                console.clear()
+                import datetime
+                now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                console.rule(f"[bold]🔥 抖音热榜 — {now}[/]")
+                print_trending(items[:count])
+                console.print()
+                info(f"下次刷新: {interval // 60} 分钟后 (Ctrl+C 退出)")
+
+            time.sleep(interval)
+        except DouyinAPIError as e:
+            warning(f"刷新失败: {e}, 等待重试...")
+            time.sleep(60)