56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
from sqlalchemy import Column, Integer, ForeignKey, DateTime, UniqueConstraint
|
||
from sqlalchemy.sql import func
|
||
from pydantic import BaseModel, Field
|
||
from typing import Optional, List
|
||
from datetime import datetime
|
||
from .database import Base
|
||
from .timeperiod import TimePeriodInfo
|
||
|
||
# 数据库模型
|
||
class CommunityTimePeriodDB(Base):
|
||
"""社区配送时段关联表"""
|
||
__tablename__ = "community_time_periods"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
community_id = Column(Integer, ForeignKey("communities.id"), nullable=False, index=True)
|
||
time_period_id = Column(Integer, ForeignKey("time_periods.id"), nullable=False)
|
||
capacity = Column(Integer, default=0, nullable=False) # 运力（订单量），0表示无限制
|
||
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
||
update_time = Column(DateTime(timezone=True), onupdate=func.now())
|
||
|
||
# 确保每个社区的每个时段只有一条记录
|
||
__table_args__ = (
|
||
UniqueConstraint('community_id', 'time_period_id', name='uix_community_time_period'),
|
||
)
|
||
|
||
# Pydantic 模型
|
||
class CommunityTimePeriodCreate(BaseModel):
|
||
community_id: int = Field(..., description="社区ID")
|
||
time_period_id: int = Field(..., description="配送时段ID")
|
||
capacity: int = Field(0, description="运力（订单量），0表示无限制")
|
||
|
||
class CommunityTimePeriodUpdate(BaseModel):
|
||
capacity: int = Field(..., description="运力（订单量），0表示无限制")
|
||
|
||
class CommunityTimePeriodInfo(BaseModel):
|
||
id: int
|
||
community_id: int
|
||
time_period_id: int
|
||
capacity: int
|
||
create_time: datetime
|
||
update_time: Optional[datetime] = None
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
class CommunityTimePeriodWithDetail(CommunityTimePeriodInfo):
|
||
time_period: Optional[TimePeriodInfo] = None
|
||
|
||
class CommunityTimePeriodWithCapacity(BaseModel):
|
||
community_id: int
|
||
time_period_id: int
|
||
capacity: int
|
||
|
||
class CommunityTimePeriodBatchCreate(BaseModel):
|
||
community_id: int = Field(..., description="社区ID")
|
||
time_period_capacity_list: List[CommunityTimePeriodWithCapacity] = Field(..., description="配送时段ID列表") |