В качестве конкретного примера у меня есть фабричная функция Factory(), которая создает класс CRUD с помощью метода класса .create(). Функция Factory() принимает два параметра:
- : класс модели SQLAlchemy, представляющий таблицу
Код: Выделить всё
model
- : класс схемы Pydantic
Код: Выделить всё
schema
Код: Выделить всё
from typing import Type, TypeVar
from pydantic import BaseModel
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
class Base(DeclarativeBase):
pass
SqlModel = TypeVar("SqlModel", bound=Base)
PydanticSchema = TypeVar("PydanticSchema", bound=BaseModel)
def Factory(model: Type[SqlModel], schema: Type[PydanticSchema]):
class Foo:
@classmethod
def create(cls, data: schema) -> model:
db_model = model(**data.model_dump())
return db_model
return Foo
class MyModel(Base):
name: Mapped[str] = mapped_column(primary_key=True)
class MySchema(BaseModel):
name: str
MyFactory = Factory(MyModel, MySchema)
bar = MyFactory.create(MySchema(name="bar"))
В примере:
< ul>
[*]С помощью TypeVar и Type я получаю данные типа MySchema@Factory и тип возвращаемого значения MyModel@Factory, но это кажется неправильным, поскольку не связано с исходными типами классов и их методами
< li>Без TypeVar оба типа имеют тип Any.
Как мне ввести это, чтобы при вызове MyFactory = Factory (МояМодель, MySchema), MyFactory.create() имеет правильную аннотацию аргумента MySchema и возвращаемый тип MyModel?
Подробнее здесь: https://stackoverflow.com/questions/791 ... class-meth