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

например, если у меня есть сигнал user_update, который ожидает, что sender будет экземпляром User и будет иметь ровно два kwargs: time: int, audit: str, я могу подкласс Signal, чтобы обеспечить это следующим образом:

class UserUpdateSignal(Signal):
  class Receiver(Protocol):
    def __call__(sender: User, /, time: int, audit: str):
      ...

  def send(sender: User, /, time: int, audit: str):
    # super call

  def connect(receiver: Receiver):
    # super call

что приводит к желаемому поведению при проверке типов:

user_update.send(user, time=34, audit = "user_initiated") # OK

@user_update.connect # OK
def receiver(sender: User, /, time: int, audit: str):
    ...

user_update.send("sender") # typing error - signature mismatch

@user_update.connect # typing error - signature mismatch
def receiver(sender: str):
    ...

Проблемы с этим подходом заключаются в следующем:

• это очень многословно, для нескольких десятков сигналов у меня были бы сотни строк кода
• на самом деле он не связывает тип подписи send с типом подписи connect - их можно обновлять независимо, проверка типов пройдет, но код будет аварийно завершать работу при запуске

Идеальный подход заключался бы в применении подписи, определенной один раз, как к send, так и к connect — возможно, с помощью дженериков. На данный момент я попробовал несколько подходов:

Позиционные аргументы только с ParamSpec

Я могу добиться желаемого поведения, используя только

class TypedSignal(Generic[P], Signal):
    def send(self, *args: P.args, **kwargs: P.kwargs):
        super().send(*args, **kwargs)

    def connect(self, receiver: Callable[P, None]):
        return super().connect(receiver=receiver)


user_update = TypedSignal[[User, str]]()

Этот тип правильно проверяет позиционные аргументы, но не поддерживает kwargs из-за ограничений Callable. Мне нужна kwargs поддержка, поскольку blinker использует kwargs для каждого прошлого аргумента sender.

Другие попытки

Использование TypeVar и TypeVarTuple

Я могу добиться подсказки типа для sender arg довольно просто, используя дженерики:

T = TypeVar("T")

class TypedSignal(Generic[T], Signal):
    def send(self, sender: Type[T], **kwargs):
        super(TypedSignal, self).send(sender)

    def connect(self, receiver: Callable[[Type[T], ...], None]) -> Callable:
        return super(TypedSignal, self).connect(receiver)

# used as
my_signal = TypedSignal[MyClass]()

сложнее всего, когда я хочу добавить проверку типов для kwargs. Подход, который я пытаюсь реализовать, заключается в использовании вариативного обобщения и Unpack вот так:

T = TypeVar("T")
KW = TypeVarTuple("KW")

class TypedSignal(Generic[T, Unpack[KW]], Signal):
    def send(self, sender: Type[T], **kwargs: Unpack[Type[KW]]):
        super(TypedSignal, self).send(sender)

    def connect(self, receiver: Callable[[Type[T], Unpack[Type[KW]]], None]) -> Callable:
        return super(TypedSignal, self).connect(receiver)

но mypy жалуется: error: Unpack item in ** argument must be a TypedDict, что кажется странным, потому что эта ошибка выдается даже без использования обобщенного кода, не говоря уже о том, когда передается TypedDict.

Использование ParamSpec и протокола

P = ParamSpec("P")

class TypedSignal(Generic[P], Signal):
    def send(self, *args: P.args, **kwargs: P.kwargs) -> None:
        super().send(*args, **kwargs)

    def connect(self, receiver: Callable[P, None]):
        return super().connect(receiver=receiver)


class Receiver(Protocol):
    def __call__(self, sender: MyClass) -> None:
        pass

update = TypedSignal[Receiver]()


@update.connect
def my_func(sender: MyClass) -> None:
    pass

update.send(MyClass())

но mypy, похоже, оборачивает протокол, поэтому ожидает функцию, которая принимает протокол, и выдает следующие ошибки:

 error: Argument 1 to "connect" of "TypedSignal" has incompatible type "Callable[[MyClass], None]"; expected "Callable[[Receiver], None]"  [arg-type]
 error: Argument 1 to "send" of "TypedSignal" has incompatible type "MyClass"; expected "Receiver"  [arg-type]

Краткое содержание

Есть ли более простой способ сделать это? Возможно ли это при текущей типизации Python?

версия mypy — 1.9.0 — пробовал с более ранними версиями, и она полностью разбилась.