62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from datetime import datetime
|
|
from pydantic import ConfigDict, Field, BaseModel, model_validator
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from ..user.model import UserModel
|
|
|
|
|
|
class AuthSchema(BaseModel):
|
|
"""权限认证模型"""
|
|
model_config = ConfigDict(arbitrary_types_allowed=True)
|
|
|
|
user: UserModel | None = Field(default=None, description='用户信息')
|
|
check_data_scope: bool = Field(default=True, description='是否检查数据权限')
|
|
db: AsyncSession = Field(description='数据库会话')
|
|
|
|
|
|
class JWTPayloadSchema(BaseModel):
|
|
"""JWT载荷模型"""
|
|
sub: str = Field(..., description='用户登录信息')
|
|
is_refresh: bool = Field(default=False, description='是否刷新token')
|
|
exp: datetime | int = Field(..., description='过期时间')
|
|
|
|
@model_validator(mode='after')
|
|
def validate_fields(self):
|
|
if not self.sub or len(self.sub.strip()) == 0:
|
|
raise ValueError("会话编号不能为空")
|
|
return self
|
|
|
|
|
|
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)
|
|
|
|
enable: bool = Field(default=True, description='是否启用验证码')
|
|
key: str = Field(..., min_length=1, description='验证码唯一标识')
|
|
img_base: str = Field(..., min_length=1, description='Base64编码的验证码图片')
|
|
|
|
|
|
|