33

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

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__手动编写该方法还是有一种简单的方法来实现这一点?

4

5 回答 5

33

从强制或检查类型的意义上说,从不遵守数据类属性的类型提示。大多数像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))
于 2019-02-25T10:06:15.913 回答
9

您可以使用以下方法实现此目的__post_init__

import dataclasses

@dataclasses.dataclass
class Test:
    value : int

    def __post_init__(self):
        self.value = int(self.value)

该方法在方法之后__init__调用

https://docs.python.org/3/library/dataclasses.html#post-init-processing

于 2019-02-25T10:08:32.230 回答
9

使用pydantic.validate_arguments很容易实现

只需在数据类中使用validate_arguments装饰器:

from dataclasses import dataclass
from pydantic import validate_arguments


@validate_arguments
@dataclass
class Test:
    value: int

然后尝试你的演示,'str type' 1 将从转换strint

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

如果您传递了真正错误的类型,它将引发异常

>>> test = Test('apple')
Traceback (most recent call last):
...
pydantic.error_wrappers.ValidationError: 1 validation error for Test
value
  value is not a valid integer (type=type_error.integer)
于 2021-02-03T09:49:05.550 回答
1

是的,简单的答案是自己进行转换__init__()。我这样做是因为我想要我的对象frozen=True

对于类型验证,Pydandic 声称可以这样做,但我还没有尝试过:https ://pydantic-docs.helpmanual.io/

于 2019-04-11T18:57:15.083 回答
0

使用 Python dataclasses,另一种方法是使用该__post_init__方法,正如其他答案中所指出的:

@dataclasses.dataclass
class Test:
    value: int

    def __post_init__(self):
        self.value = int(self.value)

或者您可以使用该attrs软件包,它可以让您轻松设置转换器

@attr.define
class Test:
    value: int = attr.field(converter=int)
于 2022-03-01T17:29:53.913 回答