travel-agent-cli 0.2.0 → 0.2.2

package/python/cli/commands.py

@@ -0,0 +1,254 @@
+"""旅行 Agent CLI - 参考 Claude Code 设计模式
+
+命令设计原则：
+1. 简洁的命令定义，包含清晰的描述
+2. 支持别名（aliases）
+3. 参数提示（argumentHint）
+4. 即时执行（immediate）vs 延迟执行
+5. 支持交互模式和非交互模式
+6. 输出格式化（使用 rich 库）
+"""
+from dataclasses import dataclass, field
+from typing import Callable, Awaitable, Optional, List, Dict, Any, Union
+from enum import Enum
+
+
+class CommandMode(Enum):
+    """命令执行模式"""
+    INTERACTIVE = "interactive"      # 交互模式（默认）
+    NON_INTERACTIVE = "non-interactive"  # 非交互模式（脚本化）
+    BOTH = "both"                    # 两种模式都支持
+
+
+class CommandType(Enum):
+    """命令类型"""
+    PROMPT = "prompt"        # 触发 LLM 对话
+    LOCAL = "local"          # 本地执行
+    LOCAL_JSX = "local-jsx"  # 本地执行带 UI 组件
+
+
+@dataclass
+class CommandArg:
+    """命令参数定义"""
+    name: str
+    description: str
+    required: bool = False
+    default: Any = None
+    type_hint: str = "TEXT"
+    choices: Optional[List[str]] = None
+
+
+@dataclass
+class Command:
+    """命令定义（参考 Claude Code Command 接口）
+
+    示例:
+        run_command = Command(
+            name="run",
+            aliases=["start", "execute"],
+            description="运行完整工作流",
+            argument_hint="<keyword>",
+            immediate=True,
+            supports_non_interactive=True,
+            handler=run_workflow,
+            args=[
+                CommandArg("keyword", "搜索关键词", default="旅行"),
+                CommandArg("top_n", "推荐数量", type_hint="INTEGER", default=10),
+            ]
+        )
+    """
+    name: str
+    description: str
+    handler: Callable[..., Awaitable[Any]]
+
+    # 命令元数据
+    aliases: List[str] = field(default_factory=list)
+    argument_hint: Optional[str] = None  # 参数提示，如 "[options]" 或 "<prompt>"
+    immediate: bool = False  # 是否立即执行（不等待确认）
+    supports_non_interactive: bool = True  # 是否支持非交互模式
+    hidden: bool = False  # 是否在帮助中隐藏
+
+    # 参数定义
+    args: List[CommandArg] = field(default_factory=list)
+
+    # 执行模式
+    mode: CommandMode = CommandMode.BOTH
+
+    # 短描述（用于帮助列表，长描述的截取版）
+    short_description: Optional[str] = None
+
+    def __post_init__(self):
+        if self.short_description is None:
+            # 截取 description 的前 60 个字符作为短描述
+            self.short_description = self.description[:60] + (
+                "..." if len(self.description) > 60 else ""
+            )
+
+
+class CommandRegistry:
+    """命令注册表（参考 Claude Code 的命令系统）
+
+    使用示例:
+        registry = CommandRegistry()
+
+        # 注册命令
+        @registry.command(
+            name="run",
+            aliases=["start"],
+            description="运行完整工作流",
+            argument_hint="[options]",
+        )
+        async def run_workflow(keyword: str = "旅行"):
+            ...
+
+        # 获取命令
+        cmd = registry.get("run")
+
+        # 列出所有命令
+        for cmd in registry.list_commands():
+            print(f"  {cmd.name}: {cmd.short_description}")
+    """
+
+    def __init__(self):
+        self._commands: Dict[str, Command] = {}
+
+    def command(
+        self,
+        name: str,
+        aliases: Optional[List[str]] = None,
+        description: str = "",
+        argument_hint: Optional[str] = None,
+        immediate: bool = False,
+        supports_non_interactive: bool = True,
+    ):
+        """装饰器：注册命令
+
+        使用示例:
+            @registry.command(
+                name="run",
+                description="运行完整工作流",
+                argument_hint="[options]",
+            )
+            async def run_workflow(args):
+                ...
+        """
+        def decorator(func: Callable[..., Awaitable[Any]]) -> Command:
+            cmd = Command(
+                name=name,
+                description=description,
+                handler=func,
+                aliases=aliases or [],
+                argument_hint=argument_hint,
+                immediate=immediate,
+                supports_non_interactive=supports_non_interactive,
+            )
+            self.register(cmd)
+            return cmd
+        return decorator
+
+    def register(self, cmd: Command) -> None:
+        """注册命令"""
+        # 注册主名称
+        self._commands[cmd.name] = cmd
+
+        # 注册别名
+        for alias in cmd.aliases:
+            if alias in self._commands:
+                raise ValueError(f"命令别名 '{alias}' 已存在")
+            self._commands[alias] = cmd
+
+    def get(self, name: str) -> Optional[Command]:
+        """获取命令"""
+        return self._commands.get(name)
+
+    def list_commands(self, include_hidden: bool = False) -> List[Command]:
+        """列出所有命令"""
+        cmds = list(self._commands.values())
+        # 去重（因为别名指向同一个 Command 对象）
+        seen = set()
+        unique = []
+        for cmd in cmds:
+            if cmd.name not in seen:
+                seen.add(cmd.name)
+                if not cmd.hidden or include_hidden:
+                    unique.append(cmd)
+        return unique
+
+    def has(self, name: str) -> bool:
+        """检查命令是否存在"""
+        return name in self._commands
+
+    def execute(
+        self,
+        name: str,
+        args: Optional[List[str]] = None,
+        **kwargs
+    ) -> Awaitable[Any]:
+        """执行命令"""
+        cmd = self.get(name)
+        if not cmd:
+            raise ValueError(f"未知命令：{name}")
+        return cmd.handler(*args or [], **kwargs)
+
+
+# =============================================================================
+# 输出格式化（参考 Claude Code print.ts）
+# =============================================================================
+
+class OutputFormatter:
+    """输出格式化工具"""
+
+    @staticmethod
+    def format_command_list(commands: List[Command]) -> str:
+        """格式化命令列表输出"""
+        lines = []
+
+        # 分组：常用命令和其他命令
+        common_commands = [c for c in commands if c.immediate]
+        other_commands = [c for c in commands if not c.immediate and not c.hidden]
+
+        if common_commands:
+            lines.append("常用命令:")
+            for cmd in sorted(common_commands, key=lambda x: x.name):
+                hint = f" {cmd.argument_hint}" if cmd.argument_hint else ""
+                lines.append(f"  {cmd.name}{hint}")
+                lines.append(f"    {cmd.short_description}")
+            lines.append("")
+
+        if other_commands:
+            lines.append("其他命令:")
+            for cmd in sorted(other_commands, key=lambda x: x.name):
+                hint = f" {cmd.argument_hint}" if cmd.argument_hint else ""
+                aliases_str = f" (别名：{', '.join(cmd.aliases)})" if cmd.aliases else ""
+                lines.append(f"  {cmd.name}{hint}{aliases_str}")
+                lines.append(f"    {cmd.short_description}")
+
+        return "\n".join(lines)
+
+    @staticmethod
+    def format_command_help(cmd: Command) -> str:
+        """格式化单个命令的帮助信息"""
+        lines = []
+
+        # 命令名称和参数提示
+        header = cmd.name
+        if cmd.argument_hint:
+            header += f" {cmd.argument_hint}"
+        if cmd.aliases:
+            header += f" (别名：{', '.join(cmd.aliases)})"
+
+        lines.append(f"用法：{header}")
+        lines.append("")
+        lines.append(cmd.description)
+        lines.append("")
+
+        # 参数列表
+        if cmd.args:
+            lines.append("参数:")
+            for arg in cmd.args:
+                req = "必填" if arg.required else "可选"
+                default = f" [默认：{arg.default}]" if arg.default is not None else ""
+                lines.append(f"  {arg.name} ({arg.type_hint}, {req}){default}")
+                lines.append(f"    {arg.description}")
+
+        return "\n".join(lines)

package/python/collectors/__init__.py

@@ -0,0 +1,14 @@
+"""数据采集模块"""
+from collectors.xiaohongshu import XiaohongshuCollector
+from collectors.weibo import WeiboCollector
+from collectors.wenlv import WenlvCollector
+from collectors.ota.fliggy import FliggyCollector
+from collectors.ota.ctrip import CtripCollector
+
+__all__ = [
+    "XiaohongshuCollector",
+    "WeiboCollector",
+    "WenlvCollector",
+    "FliggyCollector",
+    "CtripCollector",
+]

package/python/collectors/ota/ctrip.py

@@ -0,0 +1,120 @@
+"""携程数据采集器
+
+爬取携程平台的机票和酒店价格
+参考：知乎 2025 爬虫教程
+"""
+import asyncio
+from typing import List, Optional
+from datetime import datetime
+
+from config.models import FlightInfo, HotelInfo
+from utils.storage import Storage
+from utils.http import get_client
+
+
+class CtripCollector:
+    """携程 OTA 采集器"""
+
+    def __init__(self):
+        self.storage = Storage()
+        self.base_url = "https://www.ctrip.com"
+
+    async def search_flights(
+        self,
+        departure_city: str,
+        arrival_city: str,
+        departure_date: Optional[str] = None,
+    ) -> List[FlightInfo]:
+        """搜索机票价格"""
+        print(f"携程搜索机票：{departure_city} -> {arrival_city}")
+
+        # 返回示例数据
+        return self._get_sample_flights(departure_city, arrival_city)
+
+    async def search_hotels(
+        self,
+        destination: str,
+        check_in: Optional[str] = None,
+        check_out: Optional[str] = None,
+    ) -> List[HotelInfo]:
+        """搜索酒店价格"""
+        print(f"携程搜索酒店：{destination}")
+
+        # 返回示例数据
+        return self._get_sample_hotels(destination)
+
+    def _get_sample_flights(
+        self, departure: str, arrival: str
+    ) -> List[FlightInfo]:
+        """示例航班数据"""
+        samples = [
+            {
+                "id": f"ctrip_flight_{departure}_{arrival}_001",
+                "departure_city": departure,
+                "arrival_city": arrival,
+                "price": 750 + hash(departure + arrival) % 500,
+                "airline": "中国国航",
+                "departure_time": "07:30",
+                "travel_time": "2h 30m",
+            },
+            {
+                "id": f"ctrip_flight_{departure}_{arrival}_002",
+                "departure_city": departure,
+                "arrival_city": arrival,
+                "price": 680 + hash(departure + arrival) % 400,
+                "airline": "海南航空",
+                "departure_time": "12:00",
+                "travel_time": "2h 40m",
+            },
+        ]
+
+        return [
+            FlightInfo(
+                id=s["id"],
+                departure_city=s["departure_city"],
+                arrival_city=s["arrival_city"],
+                price=s["price"],
+                airline=s.get("airline"),
+                departure_time=s.get("departure_time"),
+                travel_time=s.get("travel_time"),
+                platform="ctrip",
+            )
+            for s in samples
+        ]
+
+    def _get_sample_hotels(self, destination: str) -> List[HotelInfo]:
+        """示例酒店数据"""
+        samples = [
+            {
+                "id": f"ctrip_hotel_{destination}_001",
+                "name": f"{destination}希尔顿酒店",
+                "destination": destination,
+                "price_per_night": 600 + hash(destination) % 300,
+                "rating": 4.7,
+                "stars": 5,
+                "url": f"https://www.ctrip.com/hotel/{destination}_001",
+            },
+            {
+                "id": f"ctrip_hotel_{destination}_002",
+                "name": f"{destination}亚朵酒店",
+                "destination": destination,
+                "price_per_night": 350 + hash(destination) % 200,
+                "rating": 4.6,
+                "stars": 4,
+                "url": f"https://www.ctrip.com/hotel/{destination}_002",
+            },
+        ]
+
+        return [
+            HotelInfo(
+                id=s["id"],
+                name=s["name"],
+                destination=s["destination"],
+                price_per_night=s["price_per_night"],
+                rating=s.get("rating"),
+                stars=s.get("stars"),
+                platform="ctrip",
+                url=s.get("url"),
+            )
+            for s in samples
+        ]

package/python/collectors/ota/fliggy.py

@@ -0,0 +1,152 @@
+"""飞猪数据采集器
+
+爬取飞猪平台的机票和酒店价格
+参考：RealDataAPI, 知乎 2025 爬虫教程
+"""
+import asyncio
+from typing import List, Optional, Dict
+from datetime import datetime, timedelta
+
+from config.models import FlightInfo, HotelInfo
+from utils.storage import Storage
+from utils.http import get_client
+
+
+class FliggyCollector:
+    """飞猪 OTA 采集器"""
+
+    def __init__(self):
+        self.storage = Storage()
+        self.base_url = "https://www.fliggy.com"
+
+    async def search_flights(
+        self,
+        departure_city: str,
+        arrival_city: str,
+        departure_date: Optional[str] = None,
+    ) -> List[FlightInfo]:
+        """搜索机票价格
+
+        Args:
+            departure_city: 出发城市
+            arrival_city: 到达城市
+            departure_date: 出发日期，格式 YYYY-MM-DD
+
+        注意：飞猪有严格反爬，需要处理：
+        1. 动态加载（使用 Playwright）
+        2. 验证码
+        3. Cookie 验证
+        """
+        # 简化示例，实际需要集成 Playwright
+        print(f"搜索机票：{departure_city} -> {arrival_city}")
+
+        # 返回示例数据
+        return self._get_sample_flights(departure_city, arrival_city)
+
+    async def search_hotels(
+        self,
+        destination: str,
+        check_in: Optional[str] = None,
+        check_out: Optional[str] = None,
+        price_min: Optional[int] = None,
+        price_max: Optional[int] = None,
+    ) -> List[HotelInfo]:
+        """搜索酒店价格"""
+        print(f"搜索酒店：{destination}")
+
+        # 返回示例数据
+        return self._get_sample_hotels(destination)
+
+    def _get_sample_flights(
+        self, departure: str, arrival: str
+    ) -> List[FlightInfo]:
+        """示例航班数据"""
+        samples = [
+            {
+                "id": f"fliggy_flight_{departure}_{arrival}_001",
+                "departure_city": departure,
+                "arrival_city": arrival,
+                "price": 800 + hash(departure + arrival) % 500,
+                "airline": "中国国航",
+                "departure_time": "08:00",
+                "travel_time": "2h 30m",
+            },
+            {
+                "id": f"fliggy_flight_{departure}_{arrival}_002",
+                "departure_city": departure,
+                "arrival_city": arrival,
+                "price": 650 + hash(departure + arrival) % 400,
+                "airline": "南方航空",
+                "departure_time": "14:30",
+                "travel_time": "2h 45m",
+            },
+            {
+                "id": f"fliggy_flight_{departure}_{arrival}_003",
+                "departure_city": departure,
+                "arrival_city": arrival,
+                "price": 1200 + hash(departure + arrival) % 600,
+                "airline": "东方航空",
+                "departure_time": "19:00",
+                "travel_time": "2h 20m",
+            },
+        ]
+
+        return [
+            FlightInfo(
+                id=s["id"],
+                departure_city=s["departure_city"],
+                arrival_city=s["arrival_city"],
+                price=s["price"],
+                airline=s.get("airline"),
+                departure_time=s.get("departure_time"),
+                travel_time=s.get("travel_time"),
+                platform="fliggy",
+            )
+            for s in samples
+        ]
+
+    def _get_sample_hotels(self, destination: str) -> List[HotelInfo]:
+        """示例酒店数据"""
+        samples = [
+            {
+                "id": f"fliggy_hotel_{destination}_001",
+                "name": f"{destination}国际大酒店",
+                "destination": destination,
+                "price_per_night": 500 + hash(destination) % 300,
+                "rating": 4.8,
+                "stars": 5,
+                "url": f"https://www.fliggy.com/hotel/{destination}_001",
+            },
+            {
+                "id": f"fliggy_hotel_{destination}_002",
+                "name": f"{destination}精品民宿",
+                "destination": destination,
+                "price_per_night": 280 + hash(destination) % 200,
+                "rating": 4.5,
+                "stars": None,
+                "url": f"https://www.fliggy.com/hotel/{destination}_002",
+            },
+            {
+                "id": f"fliggy_hotel_{destination}_003",
+                "name": f"{destination}经济酒店",
+                "destination": destination,
+                "price_per_night": 150 + hash(destination) % 100,
+                "rating": 4.2,
+                "stars": 3,
+                "url": f"https://www.fliggy.com/hotel/{destination}_003",
+            },
+        ]
+
+        return [
+            HotelInfo(
+                id=s["id"],
+                name=s["name"],
+                destination=s["destination"],
+                price_per_night=s["price_per_night"],
+                rating=s.get("rating"),
+                stars=s.get("stars"),
+                platform="fliggy",
+                url=s.get("url"),
+            )
+            for s in samples
+        ]