我有一个函数验证其参数以仅接受给定的有效选项列表中的值。在打字方面,我使用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 原则,因为我必须重写 Literal 类型定义中的有效参数列表,即使它已经存储在变量中VALID_ARGUMENTS
. 我怎样才能创建Argument
动态文字类型,给定VALID_ARGUMENTS
多变的?
做以下几件事not work:
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
那么,如何才能做到呢?还是根本做不到?
反过来,构建VALID_ARGUMENTS
from Argument
:
Argument = typing.Literal['foo', 'bar']
VALID_ARGUMENTS: typing.Tuple[Argument, ...] = typing.get_args(Argument)
可以在运行时构建Argument
from VALID_ARGUMENTS
,但这样做与静态分析不兼容,静态分析是类型注释的主要用例。建筑VALID_ARGUMENTS
from Argument
是要走的路。
我用过一个元组VALID_ARGUMENTS
在这里,但如果由于某种原因你确实更喜欢一个列表,你可以得到一个:
VALID_ARGUMENTS: typing.List[Argument] = list(typing.get_args(Argument))
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)