Как использовать аргумент типа в качестве возвращаемого типа? ⇐ Python

[Anonymous]

Можно ли сообщить средству проверки типа тип возвращаемого значения, указав входной аргумент, например:

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

from __future__ import annotations

from typing import TypeVar

T = TypeVar("T", bound=str | int)
VALUES: dict[str, str] = {"SIZE": "100", "ADDR": "0x100", "NAME": "potato"}

def get_parameter(parameter: str, _rtype: type[T]) -> T:
"""Get the parameter"""
if parameter.startswith("N"):
return int(VALUES[parameter], 0)
return VALUES[parameter]

Это приводит к ошибке в Python 3.12 с mypy 1.18.2 в строках с возвратами:

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

error: Incompatible return value type (got "int", expected "T")  [return-value]
error: Incompatible return value type (got "str", expected "T")  [return-value]

См. также этот пример игровой площадки mypy.
Вопросы:

• Можно ли это сделать и как?
• Может ли это поддерживаться в Python 3.10?
• Может ли входной аргумент rtype иметь значение по умолчанию, скажем, int?

Альтернативы:

• Сайты вызова (их много) могут утверждать тип после вызова функции, но это слишком многословно:

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

  name = get_parameter("NAME")
  assert isinstance(foo, str)
  

• Предоставьте типизированные перегрузки:

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

  from typing import overload
  @overload  # this is making the default
  def get_parameter(n: str, rtype: type[str] = ...) -> str:
  ...
  @overload
  def get_parameter(n: str, rtype: type[int]) -> int:
  ...
  def get_parameter(n: str, rtype: type[str] | type[int] = str) -> str | int:
  return rtype(n)
  

Подробнее здесь: https://stackoverflow.com/questions/798 ... eturn-type