from pydantic import BaseModel
from typing import List

# 定义请求体
class RecommendRequest(BaseModel):
    city_uuid: str              # 城市id
    product_code: str           # 卷烟编码
    cust_code_list: List[str]   # 核心零售户ID列表
    cultivacation_id: str       # 投放策略id
    limit_cycle_name: str       # 投放周期名称
    
class ReportRequest(BaseModel):
    cultivacation_id: str       # 投放策略id
    
class EvalReportRequest(BaseModel):
    city_uuid: str              # 城市id
    product_code: str           # 卷烟编码
    cultivacation_id: str       # 投放策略id
    # limit_cycle_name: str       # 投放周期名称
    start_time: str              # 投放开始时间
    end_time: str               # 投放结束时间