mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-27 06:41:12 +00:00
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Optional, Union
|
|
from datetime import datetime
|
|
from pydantic import ConfigDict, Field, BaseModel, model_validator
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.api.v1.schemas.system.user_schema import UserOutSchema
|
|
|
|
|
|
class AuthSchema(BaseModel):
|
|
"""权限认证模型"""
|
|
model_config = ConfigDict(arbitrary_types_allowed=True)
|
|
|
|
user: Optional[UserOutSchema] = Field(default=None, description='用户信息')
|
|
check_data_scope: bool = Field(default=True, description='是否检查数据权限')
|
|
db: AsyncSession = Field(default=None, description='数据库会话')
|
|
|
|
|
|
class JWTPayloadSchema(BaseModel):
|
|
"""JWT载荷模型"""
|
|
sub: str = Field(..., description='用户ID')
|
|
is_refresh: bool = Field(default=False, description='是否刷新token')
|
|
exp: Union[datetime, int] = Field(..., description='过期时间')
|
|
|
|
@model_validator(mode='after')
|
|
def validate_fields(cls, data):
|
|
if not data.sub or len(data.sub.strip()) == 0:
|
|
raise ValueError("用户ID不能为空")
|
|
return data
|
|
|
|
|
|
class JWTOutSchema(BaseModel):
|
|
"""JWT响应模型"""
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
access_token: str = Field(..., min_length=1, description='访问token')
|
|
refresh_token: str = Field(..., min_length=1, description='刷新token')
|
|
token_type: str = Field(default='Bearer', description='token类型')
|
|
expires_in: int = Field(..., gt=0, description='过期时间(秒)')
|
|
|
|
|
|
class RefreshTokenPayloadSchema(BaseModel):
|
|
"""刷新Token载荷模型"""
|
|
refresh_token: str = Field(..., min_length=1, description='刷新token')
|
|
|
|
|
|
class LogoutPayloadSchema(BaseModel):
|
|
"""退出登录载荷模型"""
|
|
token: str = Field(..., min_length=1, description='token')
|
|
|
|
|
|
class CaptchaOutSchema(BaseModel):
|
|
"""验证码响应模型"""
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
key: str = Field(..., min_length=1, description='验证码唯一标识')
|
|
img_base: str = Field(..., min_length=1, description='Base64编码的验证码图片')
|