50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
from typing import Optional
|
||
import enum
|
||
from sqlalchemy import Column, Integer, String, DECIMAL, DateTime, Enum
|
||
from sqlalchemy.sql import func
|
||
from pydantic import BaseModel, Field
|
||
from .database import Base
|
||
|
||
class CommunityStatus(str, enum.Enum):
|
||
UNOPEN = "UNOPEN" # 未运营
|
||
OPENING = "OPENING" # 已运营
|
||
|
||
# 数据库模型
|
||
class CommunityDB(Base):
|
||
__tablename__ = "communities"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
name = Column(String(100), nullable=False)
|
||
address = Column(String(200), nullable=False)
|
||
longitude = Column(DECIMAL(9,6), nullable=False) # 经度，精确到小数点后6位
|
||
latitude = Column(DECIMAL(9,6), nullable=False) # 纬度，精确到小数点后6位
|
||
status = Column(Enum(CommunityStatus), nullable=False, default=CommunityStatus.UNOPEN)
|
||
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
||
update_time = Column(DateTime(timezone=True), onupdate=func.now())
|
||
|
||
# Pydantic 模型
|
||
class CommunityCreate(BaseModel):
|
||
name: str = Field(..., max_length=100)
|
||
address: str = Field(..., max_length=200)
|
||
longitude: float = Field(..., ge=-180, le=180)
|
||
latitude: float = Field(..., ge=-90, le=90)
|
||
status: CommunityStatus = Field(default=CommunityStatus.UNOPEN)
|
||
|
||
class CommunityUpdate(BaseModel):
|
||
name: Optional[str] = Field(None, max_length=100)
|
||
address: Optional[str] = Field(None, max_length=200)
|
||
longitude: Optional[float] = Field(None, ge=-180, le=180)
|
||
latitude: Optional[float] = Field(None, ge=-90, le=90)
|
||
status: Optional[CommunityStatus] = None
|
||
|
||
class CommunityInfo(BaseModel):
|
||
id: int
|
||
name: str
|
||
address: str
|
||
latitude: float
|
||
longitude: float
|
||
status: CommunityStatus
|
||
distance: Optional[float] = None # 距离，单位：米
|
||
|
||
class Config:
|
||
from_attributes = True |