Типизация основных методов QuerySet

Допустим, у меня есть модель Event, я хочу, чтобы Event.objects.filter(blablabla) возвращала QuerySet[Event], а не просто QuerySet.

Для этой цели я создал Manager. Мне нужен менеджер в отдельном файле, так как у меня много менеджеров. Поэтому идеально было бы сделать что-то вроде 

class EventManager(models.Manager):
    def filter(self, *args, **kwargs) -> QuerySet[Event]:
        return super().filter(*args, **kwargs)

Но я не могу импортировать модель Event, так как это будет круговой импорт (Event.objects - это EventManager).

Как я могу предоставить подсказки типов для функций filter и get?

Вам необходимо использовать константу TYPE_CHECKING из модуля typing. Константа будет False во время выполнения, но предполагается True программами проверки типов.

from __future__ import annotations
from typing import TYPE_CHECKING
if TYPE_CHECKING:
    import Event

class EventManager(models.Manager):
    def filter(self, *args, **kwargs) -> QuerySet[Event]:
        return super().filter(*args, **kwargs)
Вернуться на верх