检查类型是否为 Python 中的联合类型

Question

我定义了一个数据类：

import dataclasses

@dataclasses.dataclass
class MyClass:
    attr1: int | None
    attr2: str | None

我可以循环遍历我的属性类型：

for field in dataclasses.fields(MyClass):
    fieldname = field.name
    fieldtype = field.type 

但是我如何检查'str'类型是否在'fieldtype'中或获取联合类型中的类型列表？

Answer 1

您可以使用联合类型的 __args__ 属性。

>>> import dataclasses
>>> import types
>>> 
>>> @dataclasses.dataclass
... class MyClass:
...     attr1: int | None
...     attr2: str | None
... 
>>> for field in dataclasses.fields(MyClass):
...     fieldname = field.name
...     fieldtype = field.type 
...     has_str_exists = False
...     if isinstance(fieldtype, types.UnionType):
...         fieldtype = fieldtype.__args__
...         has_str_exists = str in fieldtype
...     print(f"{fieldname=} {fieldtype=} {has_str_exists=}")
... 
fieldname='attr1' fieldtype=(<class 'int'>, <class 'NoneType'>) has_str_exists=False
fieldname='attr2' fieldtype=(<class 'str'>, <class 'NoneType'>) has_str_exists=True