78 lines
3.3 KiB
Python
78 lines
3.3 KiB
Python
from sqlalchemy import Column, Integer, String, ForeignKey, DateTime, Boolean, Enum
|
||
from sqlalchemy.sql import func
|
||
from sqlalchemy.dialects.mysql import DECIMAL
|
||
from pydantic import BaseModel, Field
|
||
from .database import Base
|
||
from typing import Optional
|
||
from app.models.user import Gender # 复用用户模型中的性别枚举
|
||
import enum
|
||
|
||
# 地址类型枚举
|
||
class AddressType(str, enum.Enum):
|
||
PICKUP = "PICKUP" # 代取地址
|
||
COMMON = "COMMON" # 通用地址
|
||
|
||
# 数据库模型
|
||
class AddressDB(Base):
|
||
__tablename__ = "delivery_addresses"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
user_id = Column(Integer, ForeignKey("users.userid"), index=True)
|
||
community_id = Column(Integer, index=True)
|
||
community_name = Column(String(100), nullable=True)
|
||
community_building_id = Column(Integer, ForeignKey("community_buildings.id"), nullable=True)
|
||
community_building_name = Column(String(100), nullable=True)
|
||
address_detail = Column(String(200))
|
||
name = Column(String(50))
|
||
phone = Column(String(11))
|
||
gender = Column(Enum(Gender), nullable=False, default=Gender.UNKNOWN)
|
||
is_default = Column(Boolean, default=False)
|
||
address_type = Column(Enum(AddressType), nullable=False, default=AddressType.PICKUP) # 地址类型
|
||
longitude = Column(DECIMAL(9,6), nullable=True) # 经度，精确到小数点后6位
|
||
latitude = Column(DECIMAL(9,6), nullable=True) # 纬度，精确到小数点后6位
|
||
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
||
update_time = Column(DateTime(timezone=True), onupdate=func.now())
|
||
|
||
# Pydantic 模型
|
||
class AddressCreate(BaseModel):
|
||
community_id: Optional[int] = None
|
||
community_name: Optional[str] = None
|
||
community_building_id: Optional[int] = None
|
||
address_detail: str = Field(..., max_length=200)
|
||
name: str = Field(..., max_length=50)
|
||
phone: str = Field(..., pattern="^1[3-9]\d{9}$")
|
||
gender: Gender = Gender.UNKNOWN
|
||
is_default: bool = True
|
||
address_type: AddressType = AddressType.PICKUP # 地址类型，默认为代取
|
||
longitude: Optional[float] = Field(None, ge=-180, le=180) # 经度
|
||
latitude: Optional[float] = Field(None, ge=-90, le=90) # 纬度
|
||
|
||
class AddressUpdate(BaseModel):
|
||
community_id: Optional[int] = None
|
||
community_building_id: Optional[int] = None
|
||
address_detail: Optional[str] = Field(None, max_length=200)
|
||
name: Optional[str] = Field(None, max_length=50)
|
||
phone: Optional[str] = Field(None, pattern="^1[3-9]\d{9}$")
|
||
gender: Optional[Gender] = None
|
||
is_default: Optional[bool] = None
|
||
address_type: Optional[AddressType] = AddressType.PICKUP # 地址类型
|
||
longitude: Optional[float] = Field(None, ge=-180, le=180) # 经度
|
||
latitude: Optional[float] = Field(None, ge=-90, le=90) # 纬度
|
||
|
||
class AddressInfo(BaseModel):
|
||
id: int
|
||
community_id: Optional[int] = None
|
||
community_name: Optional[str] = None
|
||
community_building_id: Optional[int] = None
|
||
community_building_name: Optional[str] = None
|
||
address_detail: str
|
||
name: str
|
||
phone: str
|
||
gender: Gender
|
||
is_default: bool
|
||
address_type: Optional[AddressType] = AddressType.PICKUP # 地址类型
|
||
longitude: Optional[float] = None # 经度
|
||
latitude: Optional[float] = None # 纬度
|
||
|
||
class Config:
|
||
from_attributes = True |