Skip to content

Commit 36b7e20

Browse files
committed
fix: 引入泛型修复as_query和as_form装饰模型文档丢失的问题
1 parent ce16b19 commit 36b7e20

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

ruoyi-fastapi-backend/module_admin/annotation/pydantic_annotation.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,13 @@
22
from fastapi import Form, Query
33
from pydantic import BaseModel
44
from pydantic.fields import FieldInfo
5-
from typing import Type
5+
from typing import Type, TypeVar
66

77

8-
def as_query(cls: Type[BaseModel]):
8+
BaseModelVar = TypeVar('BaseModelVar', bound=BaseModel)
9+
10+
11+
def as_query(cls: Type[BaseModelVar]) -> Type[BaseModelVar]:
912
"""
1013
pydantic模型查询参数装饰器,将pydantic模型用于接收查询参数
1114
"""
@@ -43,7 +46,7 @@ async def as_query_func(**data):
4346
return cls
4447

4548

46-
def as_form(cls: Type[BaseModel]):
49+
def as_form(cls: Type[BaseModelVar]) -> Type[BaseModelVar]:
4750
"""
4851
pydantic模型表单参数装饰器,将pydantic模型用于接收表单参数
4952
"""

0 commit comments

Comments
 (0)