refactor(auth): 优化认证的代码风格

This commit is contained in:
yuan
2025-09-08 16:50:27 +08:00
parent a1dc54a662
commit 255f9ef245
@@ -2,7 +2,7 @@
from typing import Optional, Union from typing import Optional, Union
from datetime import datetime from datetime import datetime
from pydantic import ConfigDict, Field, BaseModel, field_validator, model_validator from pydantic import ConfigDict, Field, BaseModel, model_validator
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
@@ -25,10 +25,10 @@ class JWTPayloadSchema(BaseModel):
exp: Union[datetime, int] = Field(..., description='过期时间') exp: Union[datetime, int] = Field(..., description='过期时间')
@model_validator(mode='after') @model_validator(mode='after')
def validate_fields(cls, data): def validate_fields(self):
if not data.sub or len(data.sub.strip()) == 0: if not self.sub or len(self.sub.strip()) == 0:
raise ValueError("会话编号不能为空") raise ValueError("会话编号不能为空")
return data return self
class JWTOutSchema(BaseModel): class JWTOutSchema(BaseModel):