Подсказки типов Python для декораторов классов со ссылками на себя

mypy имеет проблемы, поскольку переменные типа и спецификации параметров присутствуют только в аннотации возврата (а не в аргументах). К сожалению, я не знаю, как заставить его работать в каждом крайнем случае, особенно если вы аннотируете с его помощью методы (экземпляра или класса). Но для обычных функций (по крайней мере, это сработало в большинстве моих случаев использования) может оказаться полезным создать общий протокол декоратора и установить его в качестве типа возвращаемого значения для вашего декоратора - аналогично этому:

class Decorator(Protocol, Generic[CountType, Param, RetType]):
    __call__(self, __self__: CountType, *args: Param.args, **kwargs: Param.kwargs) -> RetType:
        ...


class FunctionCount(Generic[CountType]):
    def __init__(self, count_dict: dict[str, int]) -> None:
        self.count_dict = count_dict

    @staticmethod
    def count(
        f_name: str | None = None,
    ) -> Decorator[CountType, Param, RetType]:
        def decorator(
            func: Callable[Concatenate[CountType, Param], RetType],
        ) -> Callable[Concatenate[CountType, Param], RetType]:
            def wrapper(
                self: CountType, *args: Param.args, **kwargs: Param.kwargs
            ) -> RetType:
                function_name = f_name or f"{self.__class__.__name__}.{func.__name__}"
                if function_name not in self.count_dict:
                    self.count_dict[function_name] = 0
                self.count_dict[function_name] += 1

                return func(self, *args, **kwargs)

            return wrapper

        return decorator

Я думаю, что это действительно должен быть ответ, а не комментарий.

Ваш класс не должен быть универсальным в CountType — метод count является универсальным по своей конструкции, но не сам класс. Вы пытаетесь написать «статический метод, который работает с некоторым вызываемым объектом, первый аргумент которого является подтипом FunctionCount», а не «класс, который может применять свой статический метод только к вызываемому объекту с помощью...», верно? Тогда универсальным должен быть сам статический метод, а не класс! Сравните со следующим:

from typing import ParamSpec, Callable, TypeVar, Concatenate

Param = ParamSpec("Param")
RetType = TypeVar("RetType")
CountType = TypeVar("CountType", bound = "FunctionCount")

class FunctionCount:
    def __init__(self, count_dict: dict[str, int]) -> None:
        self.count_dict = count_dict

    @staticmethod
    def count(
        f_name: str | None = None,
    ) -> Callable[
        [Callable[Concatenate[CountType, Param], RetType]],
        Callable[Concatenate[CountType, Param], RetType],
    ]:
        def decorator(
            func: Callable[Concatenate[CountType, Param], RetType],
        ) -> Callable[Concatenate[CountType, Param], RetType]:
            def wrapper(
                self: CountType, /, *args: Param.args, **kwargs: Param.kwargs
            ) -> RetType:
                function_name = f_name or f"{self.__class__.__name__}.{func.__name__}"
                if function_name not in self.count_dict:
                    self.count_dict[function_name] = 0
                self.count_dict[function_name] += 1

                return func(self, *args, **kwargs)

            return wrapper
            
        return decorator

Это проходит mypy --strict (площадка mypy , площадка пирайт).

Кроме того, запускайте mypy --strict, когда вы сталкиваетесь с некоторыми mypy ошибками, которые содержат Never (или просто что-то «странное», что вам трудно понять) — это может указывать на другие места, где вам что-то mypy не нравится. mypy --strict выдает кучу ошибок в исходном коде, одна из самых важных — отсутствие общего аргумента для подкласса (см. здесь).