40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from sqlalchemy import Column, String, DateTime,Integer, Boolean
|
|
from sqlalchemy.sql import func
|
|
from pydantic import BaseModel, Field
|
|
from .database import Base
|
|
from typing import Optional
|
|
|
|
# 数据库模型
|
|
class UserDB(Base):
|
|
__tablename__ = "users"
|
|
|
|
userid = Column(Integer, primary_key=True,autoincrement=True, index=True)
|
|
username = Column(String(50))
|
|
phone = Column(String(11), unique=True, index=True)
|
|
is_admin = Column(Boolean, default=False)
|
|
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
|
update_time = Column(DateTime(timezone=True), onupdate=func.now())
|
|
|
|
# Pydantic 模型
|
|
class UserLogin(BaseModel):
|
|
phone: str = Field(..., pattern="^1[3-9]\d{9}$")
|
|
verify_code: str = Field(..., min_length=6, max_length=6)
|
|
|
|
class UserInfo(BaseModel):
|
|
userid: int
|
|
username: str
|
|
phone: str
|
|
is_admin: bool
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class VerifyCodeRequest(BaseModel):
|
|
phone: str = Field(..., pattern="^1[3-9]\d{9}$")
|
|
|
|
class UserUpdate(BaseModel):
|
|
username: Optional[str] = Field(None, min_length=2, max_length=50)
|
|
avatar: Optional[str] = Field(None, max_length=200)
|
|
|
|
class Config:
|
|
extra = "forbid" # 禁止额外字段 |