python - 如何强制数据类字段的类型？

Question

这个问题已经有答案了:

Validating detailed types in python dataclasses (4 个回答)

Force type conversion in python dataclass __init__ method (8 个回答)

已关闭 3 年前。

在此代码中:

import dataclasses

@dataclasses.dataclass
class MyClass:
    value: str

obj = MyClass(value=1)

数据类MyClass使用不遵守 value 的值实例化类型。

是否有一种简单的方法(使用装饰器、 dataclass 装饰器或库中的参数)来强制字段的类型，以便我的示例中的最后一行引发 ValueError或类似的东西？以这种方式强制类型有一个主要缺点吗？

Answer 1

您可以声明一个自定义 __post_init__ 方法(请参阅 python's doc )并将所有检查放在那里以强制进行类型检查。该方法可以在父类中声明，以减少更改量。

import dataclasses


@dataclasses.dataclass()
class Parent:
    def __post_init__(self):
        for (name, field_type) in self.__annotations__.items():
            if not isinstance(self.__dict__[name], field_type):
                current_type = type(self.__dict__[name])
                raise TypeError(f"The field `{name}` was assigned by `{current_type}` instead of `{field_type}`")

        print("Check is passed successfully")


@dataclasses.dataclass()
class MyClass(Parent):
    value: str


obj1 = MyClass(value="1")
obj2 = MyClass(value=1)

结果:

Check is passed successfully
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<string>", line 3, in __init__
  File "<stdin>", line 7, in __post_init__
TypeError: The field `value` was assigned by `<class 'int'>` instead of `<class 'str'>`