Как ввести фабрическую функцию Python, которая возвращает класс с методами типизированного класса ⇐ Python

[Anonymous]

У меня возникли проблемы с вводом фабричной функции Python, которая возвращает класс с правильно типизированными методами класса.
В качестве конкретного примера у меня есть фабричная функция Factory(), которая создает класс CRUD с помощью метода класса .create(). Функция Factory() принимает два параметра:

• model: класс модели SQLAlchemy, представляющий таблицу
• schema: класс схемы Pydantic

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"))

Я хочу использовать эти два класса как в теле метода .create(), так и при его типизации, чтобы при создании экземпляра класса из функции Factory метод . Метод create() содержит классы модели и схемы, которые были переданы в Factory.
В примере:
< ul>
[*]С помощью TypeVar и Type я получаю данные типа MySchema@Factory и тип возвращаемого значения MyModel@Factory, но это кажется неправильным, поскольку не связано с исходными типами классов и их методами
< li>Без TypeVar оба типа имеют тип Any.

Как мне ввести это, чтобы при вызове MyFactory = Factory (МояМодель, MySchema), MyFactory.create() имеет правильную аннотацию аргумента MySchema и тип возвращаемого значения MyModel?
EDIT 2024-11-08 09:43:00 UTC
Я думаю, что мне удалось сделать это с помощью дженериков и двойного ввода TypeVar, но я ни в коем случае не уверен, что это правильно/злоупотребляет система типов, поэтому любая обратная связь будет полезна:
SqlModel = TypeVar("SqlModel", bound=Base)
PydanticSchema = TypeVar("PydanticSchema", bound=BaseModel)

def Factory(model: Type[SqlModel], schema: Type[PydanticSchema]):
FooSqlModel = TypeVar("FooSqlModel", bound=model)
FooSchema = TypeVar("FooSchema", bound=schema)

class Foo(Generic[FooSqlModel, FooSchema]):
@classmethod
def create(cls, data: FooSchema) -> FooSqlModel:
db_model = model(**data.model_dump())
return db_model

return Foo[model, schema]


Подробнее здесь: https://stackoverflow.com/questions/791 ... class-meth