Аннотация динамического типа для специального метода менеджеров моделей Django ⇐ Python

[Anonymous]

Мне нужна помощь с подсказкой типа для метода пользовательского менеджера моделей.
Это мой собственный менеджер и базовая модель. Я наследую эту базовую модель другим моделям, чтобы мне не приходилось снова и снова писать общие поля.

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

class BaseManager(models.Manager):
def get_or_none(self, *args, **kwargs):
try:
return self.get(*args, **kwargs)
except self.model.DoesNotExist:
return None

class BaseModel(models.Model):
id = models.UUIDField(
primary_key=True, default=uuid.uuid4, verbose_name="ID", editable=False
)
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)

objects = BaseManager()  # Set the custom manager

class Meta:
abstract = True

Это пример модели:

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

class MyModel(BaseModel):
category = models.CharField(max_length=10)

А теперь:

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

my_object = MyModel.objects.get_or_none(
category="...",
)

При наведении указателя мыши в IDE аннотация типа выглядит следующим образом:

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

my_object: BaseModel | None = MyModel.objects. get_or_none(...

Но мне нужна такая аннотация типа:

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

my_object: MyModel | None = MyModel.objects. get_or_none(...

Как это сделать? Это работает для методов по умолчанию, таких как get и filter. Но как это сделать для пользовательских методов, таких как get_or_none?
Пожалуйста, помогите мне.
Спасибо

Подробнее здесь: https://stackoverflow.com/questions/793 ... tom-method