mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2025-08-18 06:42:51 +08:00
95 lines
2.2 KiB
Python
95 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from datetime import datetime
|
|
|
|
from email_validator import validate_email, EmailNotValidError
|
|
from pydantic import HttpUrl, Field, validator
|
|
|
|
from backend.app.common.enums import StatusType
|
|
from backend.app.schemas.base import SchemaBase
|
|
from backend.app.schemas.dept import GetAllDept
|
|
from backend.app.schemas.role import GetAllRole
|
|
|
|
|
|
class Auth(SchemaBase):
|
|
username: str
|
|
password: str
|
|
|
|
|
|
class AuthLogin(Auth):
|
|
captcha: str
|
|
|
|
|
|
class CreateUser(Auth):
|
|
dept_id: int
|
|
roles: list[int]
|
|
nickname: str
|
|
email: str = Field(..., example='user@example.com')
|
|
|
|
@validator('email')
|
|
def email_validate(cls, v):
|
|
try:
|
|
validate_email(v, check_deliverability=False).email
|
|
except EmailNotValidError:
|
|
raise ValueError('邮箱格式错误')
|
|
return v
|
|
|
|
|
|
class _UserInfoBase(SchemaBase):
|
|
dept_id: int
|
|
username: str
|
|
nickname: str
|
|
email: str = Field(..., example='user@example.com')
|
|
phone: str | None = None
|
|
|
|
@validator('email')
|
|
def email_validate(cls, v):
|
|
try:
|
|
validate_email(v, check_deliverability=False).email
|
|
except EmailNotValidError:
|
|
raise ValueError('邮箱格式错误')
|
|
return v
|
|
|
|
@validator('phone')
|
|
def phone_validate(cls, v):
|
|
if v is not None and not v.isdigit():
|
|
raise ValueError('手机号格式错误')
|
|
return v
|
|
|
|
|
|
class UpdateUser(_UserInfoBase):
|
|
roles: list[int]
|
|
|
|
|
|
class Avatar(SchemaBase):
|
|
url: HttpUrl = Field(..., description='头像 http 地址')
|
|
|
|
|
|
class GetUserInfoNoRelation(_UserInfoBase):
|
|
dept_id: int | None = None
|
|
id: int
|
|
user_uuid: str
|
|
avatar: str | None = None
|
|
status: StatusType = Field(default=StatusType.enable)
|
|
is_superuser: bool
|
|
is_multi_login: bool
|
|
time_joined: datetime = None
|
|
last_login: datetime | None = None
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
class GetAllUserInfo(GetUserInfoNoRelation):
|
|
dept: GetAllDept | None = None
|
|
roles: list[GetAllRole]
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
class ResetPassword(SchemaBase):
|
|
old_password: str
|
|
new_password: str
|
|
confirm_password: str
|