提示返回类型与参数相同

Question

我想要一个 function 接受 Pydantic model 并返回它的实例。 我希望 VS Code 能够理解返回的类型并提供自动完成功能。 这不能按预期工作：

T = pydantic.BaseModel

def get_request_data(RequestSchema: typing.Type[T]) -> T:
    try:
        return RequestSchema(**flask.request.get_json())
    except pydantic.ValidationError as exc:
        ...


class PagInputSchema(pydantic.BaseModel):
    dataset_id: int = pydantic.Field(alias='datasetId')

model 字段名称不会自动完成：

Answer 1

这样做：

T = t.TypeVar("T")


def get_request_data(RequestSchema: t.Type[T]) -> T:
    ...