Python类型：从有效值列表动态创建文本别名

Question

我有一个函数，它验证它的参数，只接受给定有效选项列表中的值。在键入方面，我使用Literal类型别名反映此行为，如下所示：

from typing import Literal


VALID_ARGUMENTS = ['foo', 'bar']

Argument = Literal['foo', 'bar']


def func(argument: 'Argument') -> None:
    if argument not in VALID_ARGUMENTS:
        raise ValueError(
            f'argument must be one of {VALID_ARGUMENTS}'
        )
    # ...

这违反了DRY原则，因为我必须重写文本类型定义中的有效参数列表，即使它已经存储在变量VALID_ARGUMENTS中给定VALID_ARGUMENTS变量，如何动态创建Argument文本类型？

以下事项不起作用：

from typing import Literal, Union, NewType


Argument = Literal[*VALID_ARGUMENTS]  # SyntaxError: invalid syntax

Argument = Literal[VALID_ARGUMENTS]  # Parameters to generic types must be types

Argument = Literal[Union[VALID_ARGUMENTS]]  # TypeError: Union[arg, ...]: each arg must be a type. Got ['foo', 'bar'].

Argument = NewType(
    'Argument',
    Union[
        Literal[valid_argument]
        for valid_argument in VALID_ARGUMENTS
    ]
)  # Expected type 'Type[_T]', got 'list' instead

那么，如何做到这一点呢？或者根本就做不到

Answer 1

如果有人仍在为此寻找解决方案：

typing.Literal[tuple(VALID_ARGUMENTS)]