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

[Anonymous]

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

from typing import TypeVar

T = TypeVar("T", bound=str | int)
VALUES: dict[str, str] = {"N_SIZE": "100", "N_ADDR": "0x100", "S_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]

(может быть, использованиеbound уже неправильно?), а затем вызовите его так:
name = get_parameter("NAME", str)
size = get_parameter("SIZE", int)

Как уже говорилось, приведенный выше код выдает ошибку в Python 3.12 с mypy 1.18.2 в строках с возвратомs:
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? Чтобы разрешить: size = get_parameter("SIZE") иметь тип 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)
• Альтернатива от пользователя «исправить ошибку» в Discord Python:

from __future__ import annotations

from typing import TypeVar, cast

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

def get_parameter(parameter: str, _rtype: type[T]) -> T:
value: object
if parameter.startswith("N"):
value = int(VALUES[parameter], 0)
else:
value = VALUES[parameter]
return cast("T", value)

x = get_parameter("SIZE", str) # x is a str
y = get_parameter("NAME", int) # y is an int
z = get_parameter("ADDR", int) # z is an int to mypy, but at runtime it's a string


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