hello-agents/Co-creation-projects/2248652135-Trip-Planner-Pro/backend/app/models/schemas.py

"""数据模型定义"""

from typing import List, Optional, Union
from pydantic import BaseModel, Field, field_validator
from datetime import date


# ============ 请求模型 ============

class TripRequest(BaseModel):
    """旅行规划请求"""
    city: str = Field(..., description="目的地城市", example="北京")
    start_date: str = Field(..., description="开始日期 YYYY-MM-DD", example="2025-06-01")
    end_date: str = Field(..., description="结束日期 YYYY-MM-DD", example="2025-06-03")
    travel_days: int = Field(..., description="旅行天数", ge=1, le=30, example=3)
    transportation: str = Field(..., description="交通方式", example="公共交通")
    accommodation: str = Field(..., description="住宿偏好", example="经济型酒店")
    preferences: List[str] = Field(default=[], description="旅行偏好标签", example=["历史文化", "美食"])
    traveler_group: str = Field(default="", description="出行人群,如:独自旅行/情侣夫妻/朋友结伴/家庭亲子/公司团建/老年旅行/研学旅行", example="家庭亲子")
    free_text_input: Optional[str] = Field(default="", description="额外要求", example="希望多安排一些博物馆")

    class Config:
        json_schema_extra = {
            "example": {
                "city": "北京",
                "start_date": "2025-06-01",
                "end_date": "2025-06-03",
                "travel_days": 3,
                "transportation": "公共交通",
                "accommodation": "经济型酒店",
                "preferences": ["历史文化", "美食"],
                "traveler_group": "家庭亲子",
                "free_text_input": "希望多安排一些博物馆"
            }
        }


class POISearchRequest(BaseModel):
    """POI搜索请求"""
    keywords: str = Field(..., description="搜索关键词", example="故宫")
    city: str = Field(..., description="城市", example="北京")
    citylimit: bool = Field(default=True, description="是否限制在城市范围内")


class RouteRequest(BaseModel):
    """路线规划请求"""
    origin_address: str = Field(..., description="起点地址", example="北京市朝阳区阜通东大街6号")
    destination_address: str = Field(..., description="终点地址", example="北京市海淀区上地十街10号")
    origin_city: Optional[str] = Field(default=None, description="起点城市")
    destination_city: Optional[str] = Field(default=None, description="终点城市")
    route_type: str = Field(default="walking", description="路线类型: walking/driving/transit")


# ============ 响应模型 ============

class Location(BaseModel):
    """地理位置"""
    longitude: float = Field(..., description="经度")
    latitude: float = Field(..., description="纬度")


class Attraction(BaseModel):
    """景点信息"""
    name: str = Field(..., description="景点名称")
    address: str = Field(..., description="地址")
    location: Location = Field(..., description="经纬度坐标")
    visit_duration: int = Field(..., description="建议游览时间(分钟)")
    description: str = Field(..., description="景点描述")
    category: Optional[str] = Field(default="景点", description="景点类别")
    rating: Optional[float] = Field(default=None, description="评分")
    photos: Optional[List[str]] = Field(default_factory=list, description="景点图片URL列表")
    poi_id: Optional[str] = Field(default="", description="POI ID")
    image_url: Optional[str] = Field(default=None, description="图片URL")
    ticket_price: int = Field(default=0, description="门票价格(元)")


class Meal(BaseModel):
    """餐饮信息"""
    type: str = Field(..., description="餐饮类型: breakfast/lunch/dinner/snack")
    name: str = Field(..., description="餐饮名称")
    address: Optional[str] = Field(default=None, description="地址")
    location: Optional[Location] = Field(default=None, description="经纬度坐标")
    description: Optional[str] = Field(default=None, description="描述")
    estimated_cost: int = Field(default=0, description="预估费用(元)")


class Hotel(BaseModel):
    """酒店信息"""
    name: str = Field(..., description="酒店名称")
    address: str = Field(default="", description="酒店地址")
    location: Optional[Location] = Field(default=None, description="酒店位置")
    price_range: str = Field(default="", description="价格范围")
    rating: str = Field(default="", description="评分")
    distance: str = Field(default="", description="距离景点距离")
    type: str = Field(default="", description="酒店类型")
    estimated_cost: int = Field(default=0, description="预估费用(元/晚)")


class TransportSegment(BaseModel):
    """交通段详情"""
    type: str = Field(..., description="交通方式: 步行/公交/地铁/出租车/自驾")
    instruction: str = Field(..., description="详细交通说明,如'从天安门东站乘坐1号线到王府井站'")
    from_name: str = Field(..., description="起点名称,如'酒店'或上一个景点名")
    to_name: str = Field(..., description="终点名称,如景点名或餐厅名")
    departure_time: str = Field(default="", description="出发时间,如'09:00'")
    duration: int = Field(default=0, description="耗时(分钟)")
    distance: int = Field(default=0, description="距离(米)")
    route_detail: Optional[str] = Field(default=None, description="路线详情,如'经过5站·步行300米'")


class DayPlan(BaseModel):
    """单日行程"""
    date: str = Field(..., description="日期 YYYY-MM-DD")
    day_index: int = Field(..., description="第几天(从0开始)")
    description: str = Field(..., description="当日行程描述")
    transportation: str = Field(..., description="交通方式")
    transportation_details: List[TransportSegment] = Field(default=[], description="详细交通分段信息,包含每一段的路线、时间、距离")
    accommodation: str = Field(..., description="住宿")
    hotel: Optional[Hotel] = Field(default=None, description="推荐酒店")
    attractions: List[Attraction] = Field(default=[], description="景点列表")
    meals: List[Meal] = Field(default=[], description="餐饮列表")


class WeatherInfo(BaseModel):
    """天气信息"""
    date: str = Field(..., description="日期 YYYY-MM-DD")
    day_weather: str = Field(default="", description="白天天气")
    night_weather: str = Field(default="", description="夜间天气")
    day_temp: Union[int, str] = Field(default=0, description="白天温度")
    night_temp: Union[int, str] = Field(default=0, description="夜间温度")
    wind_direction: str = Field(default="", description="风向")
    wind_power: str = Field(default="", description="风力")

    @field_validator('day_temp', 'night_temp', mode='before')
    @classmethod
    def parse_temperature(cls, v):
        """解析温度,移除°C等单位"""
        if isinstance(v, str):
            # 移除°C, ℃等单位符号
            v = v.replace('°C', '').replace('℃', '').replace('°', '').strip()
            try:
                return int(v)
            except ValueError:
                return 0
        return v


class Budget(BaseModel):
    """预算信息"""
    total_attractions: int = Field(default=0, description="景点门票总费用")
    total_hotels: int = Field(default=0, description="酒店总费用")
    total_meals: int = Field(default=0, description="餐饮总费用")
    total_transportation: int = Field(default=0, description="交通总费用")
    total: int = Field(default=0, description="总费用")


class TripPlan(BaseModel):
    """旅行计划"""
    city: str = Field(..., description="目的地城市")
    start_date: str = Field(..., description="开始日期")
    end_date: str = Field(..., description="结束日期")
    days: List[DayPlan] = Field(..., description="每日行程")
    weather_info: List[WeatherInfo] = Field(default=[], description="天气信息")
    overall_suggestions: str = Field(..., description="总体建议")
    budget: Optional[Budget] = Field(default=None, description="预算信息")


class TripPlanResponse(BaseModel):
    """旅行计划响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(default="", description="消息")
    data: Optional[TripPlan] = Field(default=None, description="旅行计划数据")


class POIInfo(BaseModel):
    """POI信息"""
    id: str = Field(..., description="POI ID")
    name: str = Field(..., description="名称")
    type: str = Field(..., description="类型")
    address: str = Field(..., description="地址")
    location: Location = Field(..., description="经纬度坐标")
    tel: Optional[str] = Field(default=None, description="电话")


class POISearchResponse(BaseModel):
    """POI搜索响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(default="", description="消息")
    data: List[POIInfo] = Field(default=[], description="POI列表")


class RouteInfo(BaseModel):
    """路线信息"""
    distance: float = Field(..., description="距离(米)")
    duration: int = Field(..., description="时间(秒)")
    route_type: str = Field(..., description="路线类型")
    description: str = Field(..., description="路线描述")


class RouteResponse(BaseModel):
    """路线规划响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(default="", description="消息")
    data: Optional[RouteInfo] = Field(default=None, description="路线信息")


class WeatherResponse(BaseModel):
    """天气查询响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(default="", description="消息")
    data: List[WeatherInfo] = Field(default=[], description="天气信息")


# ============ 错误响应 ============

class ErrorResponse(BaseModel):
    """错误响应"""
    success: bool = Field(default=False, description="是否成功")
    message: str = Field(..., description="错误消息")
    error_code: Optional[str] = Field(default=None, description="错误代码")


# ============ 认证模型 ============

class RegisterRequest(BaseModel):
    """注册请求（密码经RSA公钥加密，Base64编码）"""
    username: str = Field(..., min_length=2, max_length=50, description="用户名")
    encrypted_password: str = Field(..., description="RSA-OAEP加密后的密码(Base64)")


class LoginRequest(BaseModel):
    """登录请求（密码经RSA公钥加密，Base64编码）"""
    username: str = Field(..., description="用户名")
    encrypted_password: str = Field(..., description="RSA-OAEP加密后的密码(Base64)")


# ============ 历史记录模型 ============

class SaveHistoryRequest(BaseModel):
    """保存历史记录请求"""
    city: str = Field(..., description="城市")
    start_date: str = Field(..., description="开始日期")
    end_date: str = Field(..., description="结束日期")
    travel_days: int = Field(..., description="旅行天数")
    preferences: List[str] = Field(default=[], description="偏好标签")
    traveler_group: str = Field(default="", description="出行人群")
    plan_data: dict = Field(..., description="完整行程数据(JSON)")


class HistoryItemResponse(BaseModel):
    """保存历史记录响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(default="", description="消息")
    history_id: int = Field(default=0, description="历史记录ID")


class HistoryListResponse(BaseModel):
    """历史记录列表响应"""
    success: bool = Field(..., description="是否成功")
    records: list = Field(default=[], description="历史记录列表")


class HistoryDetailResponse(BaseModel):
    """历史记录详情响应"""
    success: bool = Field(..., description="是否成功")
    record: Optional[dict] = Field(default=None, description="历史记录详情")


class DeleteResponse(BaseModel):
    """删除响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(default="", description="消息")


# ============ 聊天模型 ============

class ChatMessageSchema(BaseModel):
    """聊天消息"""
    id: int = Field(..., description="消息ID")
    session_id: int = Field(..., description="会话ID")
    role: str = Field(..., description="角色: user/assistant")
    content: str = Field(..., description="消息内容")
    created_at: str = Field(..., description="创建时间")


class ChatSessionSchema(BaseModel):
    """聊天会话"""
    id: int = Field(..., description="会话ID")
    user_id: int = Field(..., description="用户ID")
    title: str = Field(..., description="会话标题")
    created_at: str = Field(default="", description="创建时间")
    updated_at: str = Field(default="", description="更新时间")


class ChatSessionResponse(BaseModel):
    """会话响应"""
    success: bool = Field(..., description="是否成功")
    session: Optional[ChatSessionSchema] = Field(default=None, description="会话信息")


class ChatSessionListResponse(BaseModel):
    """会话列表响应"""
    success: bool = Field(..., description="是否成功")
    sessions: list = Field(default=[], description="会话列表")


class ChatMessagesResponse(BaseModel):
    """消息列表响应"""
    success: bool = Field(..., description="是否成功")
    messages: list = Field(default=[], description="消息列表")


class ChatSendMessageRequest(BaseModel):
    """发送消息请求"""
    content: str = Field(..., description="消息内容", min_length=1)


class ChatDeleteResponse(BaseModel):
    """聊天删除响应"""
    success: bool = Field(..., description="是否成功")
    message: str = Field(..., description="消息")