106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
from pydantic import BaseModel, Field, field_validator
|
|
from datetime import datetime, date
|
|
from typing import Optional, List
|
|
|
|
|
|
# ============ 习惯分组 Schemas ============
|
|
|
|
class HabitGroupBase(BaseModel):
|
|
"""习惯分组基础模型"""
|
|
name: str = Field(..., max_length=100)
|
|
color: str = Field(default="#FFB7C5", max_length=20)
|
|
icon: str = Field(default="flag", max_length=50)
|
|
sort_order: int = Field(default=0)
|
|
|
|
|
|
class HabitGroupCreate(HabitGroupBase):
|
|
"""创建习惯分组请求模型"""
|
|
pass
|
|
|
|
|
|
class HabitGroupUpdate(BaseModel):
|
|
"""更新习惯分组请求模型"""
|
|
name: Optional[str] = Field(None, max_length=100)
|
|
color: Optional[str] = Field(None, max_length=20)
|
|
icon: Optional[str] = Field(None, max_length=50)
|
|
sort_order: Optional[int] = None
|
|
|
|
|
|
class HabitGroupResponse(HabitGroupBase):
|
|
"""习惯分组响应模型"""
|
|
id: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# ============ 习惯 Schemas ============
|
|
|
|
class HabitBase(BaseModel):
|
|
"""习惯基础模型"""
|
|
name: str = Field(..., max_length=200)
|
|
description: Optional[str] = None
|
|
group_id: Optional[int] = None
|
|
target_count: int = Field(default=1, ge=1)
|
|
frequency: str = Field(default="daily", pattern="^(daily|weekly)$")
|
|
active_days: Optional[str] = None
|
|
|
|
|
|
class HabitCreate(HabitBase):
|
|
"""创建习惯请求模型"""
|
|
pass
|
|
|
|
|
|
class HabitUpdate(BaseModel):
|
|
"""更新习惯请求模型"""
|
|
name: Optional[str] = Field(None, max_length=200)
|
|
description: Optional[str] = None
|
|
group_id: Optional[int] = None
|
|
target_count: Optional[int] = Field(None, ge=1)
|
|
frequency: Optional[str] = Field(None, pattern="^(daily|weekly)$")
|
|
active_days: Optional[str] = None
|
|
|
|
@property
|
|
def clearable_fields(self) -> set:
|
|
return {"description", "group_id", "active_days"}
|
|
|
|
|
|
class HabitResponse(HabitBase):
|
|
"""习惯响应模型"""
|
|
id: int
|
|
is_archived: bool
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
group: Optional[HabitGroupResponse] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# ============ 打卡 Schemas ============
|
|
|
|
class CheckinCreate(BaseModel):
|
|
"""打卡请求模型"""
|
|
count: Optional[int] = Field(default=1, ge=1)
|
|
|
|
|
|
class CheckinResponse(BaseModel):
|
|
"""打卡记录响应模型"""
|
|
id: int
|
|
habit_id: int
|
|
checkin_date: date
|
|
count: int
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class HabitStatsResponse(BaseModel):
|
|
"""习惯统计响应模型"""
|
|
total_days: int = 0
|
|
current_streak: int = 0
|
|
longest_streak: int = 0
|
|
today_count: int = 0
|
|
today_completed: bool = False
|