| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- # -*- 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编码的验证码图片')
|
