Есть ли способ типизировать методы модели? ⇐ Python

[Anonymous]

У меня есть маленькая модель. например, это выглядит примерно так:

Код: Выделить всё

class GroupEntity(Base):
id: UUID = UUIDField(unique=True, primary_key=True, default=uuid4())
user_id: UUID = ForeignKeyField(User, backref='groups')
title: str = CharField(max_length=256)
type: GroupTypeEnum = CharField(choices=GROUP_TYPES)
parent_id: UUID = UUIDField(null=True)
price: int = SmallIntegerField(null=True)
created_at: float = BigIntegerField(default=time.time())

class Meta:
table_name = TableNamesEnum.GROUP_ENTITY.value

ну, теперь я хочу создать экземпляр этой модели в своей функции

Код: Выделить всё

    group_entity: Group = Group.create(
title='best group',
type=GroupTypeEnum.Channel,
user_id=user_entity.id
)

это работает нормально, но у меня нет никаких подсказок о полях, которые мне нужно передать.
я проверил метод создания базы (базовая модель крошки) и я видел там

(метод) def create(**query: Any) -> Any
поэтому я думаю, что для этого не существует существующего решения проблема

Я пытался переопределить создать метод типа

Код: Выделить всё

class GroupEntity(Base):
id: UUID = UUIDField(unique=True, primary_key=True, default=uuid4())
user_id: UUID = ForeignKeyField(User, backref='groups')
title: str = CharField(max_length=256)
type: GroupTypeEnum = CharField(choices=GROUP_TYPES)
parent_id: UUID = UUIDField(null=True)
price: int = SmallIntegerField(null=True)
created_at: float = BigIntegerField(default=time.time())

class Meta:
table_name = TableNamesEnum.GROUP_ENTITY.value

@classmethod
def create(cls, user_id: UUID, title: str, type: GroupTypeEnum,
parent_id: Optional[UUID] = None, price: Optional[int] = None) -> 'GroupEntity':
return super().create(
user_id=user_id,
title=title,
type=type,
parent_id=parent_id,
price=price
)

это сработало хорошо, но я не уверен в этом решении, потому что страшно переопределять метод create (и другие, не типизированные методы) в каждой модели
я надеюсь есть более чистое решение

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