在 Python 3.6 中运行时根据联合类型检查变量
问题描述
我正在尝试编写一个函数装饰器,它使用 Python 3.6 类型提示来检查参数字典是否尊重类型提示,如果没有明确说明问题,则引发错误,以用于 HTTP API.
I'm trying to write a function decorator that uses Python 3.6 type hints to check that a dictionary of arguments respects the type hints and if not raise an error with a clear description of the problem, to be used for HTTP APIs.
问题是当函数有一个使用 Union 类型的参数时,我无法在运行时检查变量.
The problem is that when the function has a parameter using the Union type I can't check a variable against it at runtime.
比如我有这个功能
from typing import Union
def bark(myname: str, descr: Union[int, str], mynum: int = 3) -> str:
return descr + myname * mynum
我能做到:
isinstance('Arnold', bark.__annotations__['myname'])
但不是:
isinstance(3, bark.__annotations__['descr'])
因为 Union 不能与 isinstance 或 issubclass 一起使用.
Because Union cannot be used with isinstance or issubclass.
我找不到使用类型对象检查它的方法.我尝试自己实施检查,但是当 bark.__annotations__['descr'] 在 REPL 中显示为 typing.Union[int, str] 我不能在运行时访问类型列表,如果不使用检查 bark.__annotations__['descr'].__repr__() 的丑陋技巧.
I couldn't find a way to check it using the type object.
I tried to implement the check by myself but while bark.__annotations__['descr'] is shown as typing.Union[int, str] in the REPL I can't access the list of the types at runtime, if not using the ugly hack of examining bark.__annotations__['descr'].__repr__().
是否有适当的方法来访问这些信息?还是故意让它在运行时不易访问?
Is there a proper way to access this information? Or is it deliberately intended to not be easily accessible at runtime?
解决方案
你可以使用 Union 的 __args__ 属性,它包含一个 tuple可能的内容:
You could use the __args__ attribute of Union which holds a tuple of the "possible contents:
>>> from typing import Union
>>> x = Union[int, str]
>>> x.__args__
(int, str)
>>> isinstance(3, x.__args__)
True
>>> isinstance('a', x.__args__)
True
__args__ 参数没有记录,因此它可能被认为是弄乱了实现细节",但它似乎比解析 repr 更好.
The __args__ argument is not documented so it could be considered "messing with implementation details" but it seems like a better way than parsing the repr.
相关文章