我想使用打字的get_type_hints
方法来获取参数注释。但是我在 Python3.6.8
a = typing.Optional[int]
type(a)
Out[13]: typing.Union
type(a) == typing.Union
Out[14]: False
type(a) == type(typing.Optional)
Out[23]: False
type(a) == type(typing.Optional[int])
Out[24]: True
repr(type(a))
Out[25]: 'typing.Union'
repr(typing.Union)
Out[26]: 'typing.Union'
看来,除了比较不是Python的typing.Optional
之外,没有通用的方法来判断类型是否为repr
。有hack吗?
P.S。在3.7中有typing._GenericAlias
,它工作得很好。
答案 0 :(得分:3)
我相信这篇文章Check if a field is typing.Optional已回答了这一问题。
我也在下面粘贴了它:
Optional[X]
等效于Union[X, None]
。所以你可以做,
import re
from typing import Optional
from dataclasses import dataclass, fields
@dataclass(frozen=True)
class TestClass:
required_field_1: str
required_field_2: int
optional_field: Optional[str]
def get_optional_fields(klass):
class_fields = fields(klass)
for field in class_fields:
if (
hasattr(field.type, "__args__")
and len(field.type.__args__) == 2
and field.type.__args__[-1] is type(None)
):
# Check if exactly two arguments exists and one of them are None type
yield field.name
print(list(get_optional_fields(TestClass)))