时间戳

首页 > 解决方案 > 在 python 数据类 __init__ 方法中强制类型转换

python - 在 python 数据类 __init__ 方法中强制类型转换

问题描述

我有以下非常简单的数据类：

import dataclasses

@dataclasses.dataclass
class Test:
    value: int

我创建了一个类的实例，但我使用了一个字符串而不是一个整数：

>>> test = Test('1')
>>> type(test.value)
<class 'str'>

我真正想要的是强制转换为我在类定义中定义的数据类型：

>>> test = Test('1')
>>> type(test.value)
<class 'int'>

我必须__init__手动编写该方法还是有一种简单的方法来实现这一点？

标签： pythonpython-dataclasses

解决方案

从强制或检查类型的意义上说，从不遵守数据类属性的类型提示。大多数像mypy这样的静态类型检查器都应该完成这项工作，Python 不会在运行时执行它，因为它从来没有这样做过。

如果要添加手动类型检查代码，请在__post_init__方法中进行：

@dataclasses.dataclass
class Test:
    value: int

    def __post_init__(self):
        if not isinstance(self.value, int):
            raise ValueError('value not an int')
            # or self.value = int(self.value)

您可以使用dataclasses.fields(self)获取Field指定字段和类型的对象元组，并循环遍历该对象以自动为每个字段执行此操作，而无需单独为每个字段编写它。

def __post_init__(self):
    for field in dataclasses.fields(self):
        value = getattr(self, field.name)
        if not isinstance(value, field.type):
            raise ValueError(f'Expected {field.name} to be {field.type}, '
                             f'got {repr(value)}')
            # or setattr(self, field.name, field.type(value))

推荐阅读