cmd.py

#!/usr/bin/env python3
"""图片搜索 CLI入口"""

COMMAND_NAME = "image_search"
COMMAND_DESC = "图片搜索1688商品"

import os
import sys
import argparse

sys.path.insert(0, os.path.normpath(os.path.join(os.path.dirname(__file__), '..', '..')))

from _auth import get_ak_from_env
from _output import (
    append_next_actions_markdown,
    # [DISABLED] 可视化商品墙已暂时禁用
    # append_visual_hint_markdown,
    # build_products_view_html,
    format_products_table,
    print_error,
    print_output,
    # write_products_html_file,
)

from capabilities.image_search.service import image_search


def main():
    ak_id, _ = get_ak_from_env()
    if not ak_id:
        print_output(False,
                     "❌ AK 未就绪。请先执行 `cli.py get_ak` 自动获取 AK；如自动获取失败，再执行 `cli.py configure YOUR_AK` 手动配置。\n\n获取 AK: https://clawhub.1688.com/",
                     {"data": {}, "action": "run_get_ak"})
        return

    parser = argparse.ArgumentParser(description="图片搜索 - 以图搜图找同款商品")
    parser.add_argument("--image", "-i", required=True, help="图片本地路径或 URL")
    parser.add_argument("--platform", "-p", default="1688", help="目标平台，默认 1688")
    parser.add_argument("--limit", "-l", type=int, default=10, help="返回数量，默认 10")
    parser.add_argument("--sort", "-s", default=None, 
                        choices=["price_asc", "price_desc", "sold_desc", "yx_desc"],
                        help="排序方式：price_asc(价格低→高)、price_desc(价格高→低)、sold_desc(销量高→低)、yx_desc(严选指数高→低)")
    parser.add_argument("--score-level", default="high",
                        choices=["high", "medium", "low"],
                        help="相关性档位：high(高)、medium(中)、low(低)，默认 high")
    parser.add_argument("--purchase-amount", type=int, default=1,
                        help="采购件数，默认 1")
    parser.add_argument("--tags", default="4306497",
                        help="TC标（品池标签），英文逗号分隔，默认 4306497")
    parser.add_argument("--ic-tags", default=None,
                        help="IC标（品池标签），英文逗号分隔")
    args = parser.parse_args()

    try:
        image_input = args.image

        # 区分 URL 和本地文件路径
        if image_input.startswith(("http://", "https://")):
            # URL 直接传给 service 层处理
            image_path = image_input
        else:
            # 本地路径：转为绝对路径并验证存在性
            image_path = os.path.abspath(image_input)
            if not os.path.exists(image_path):
                print_output(False,
                             f"❌ 图片路径无效：`{image_input}`\n\n展开后路径为：`{image_path}`\n\n请确认文件存在，建议使用绝对路径。",
                             {"data": {}})
                return

        result = image_search(
            image_path=image_path,
            platform=args.platform,
            limit=args.limit,
            sort_type=args.sort,
            score_level=args.score_level,
            purchase_amount=args.purchase_amount,
            tags=args.tags,
            ic_tags=args.ic_tags,
        )

        # 构建输出消息
        total = result.get("total_results", 0)
        products = result.get("similar_products", [])
        
        if total > 0:
            header = f"✅ 找到 {total} 个匹配商品"
            message = format_products_table(products, header)
            # [DISABLED] 可视化商品墙已暂时禁用
            # sub = ""
            # if result.get("source_image"):
            #     sub = f"参考图：{result.get('source_image')}"
            # html = build_products_view_html(
            #     products,
            #     page_title="1688 以图搜图结果",
            #     subtitle=sub,
            # )
            # html_path = write_products_html_file(html) if html else None
            # if html_path:
            #     message = append_visual_hint_markdown(message, html_path)
            #     result = dict(result)
            #     result["visual_html_path"] = html_path
            message = append_next_actions_markdown(message, total)
        else:
            message = "未找到匹配商品"
        
        print_output(True, message, {"data": result})
    except Exception as e:
        print_error(e, {"data": {}})


if __name__ == "__main__":
    main()