使用python数据类作为用户输入的DTO的最佳实践

Question

我想使用蟒蛇数据类作为数据容器，它可以传递给程序中的其他类。具体来说，我想在数据类中存储所有默认值，但可以通过用户输入在我的主类中更改值。随后，更新的数据对象应该被传递给所有其他使用它执行计算的类。此外，数据类已经利用post_init准备了一些数据供以后使用。

目前我的解决方案示意性地是

from dataclasses import dataclass


@dataclass
class DataContainer:
    dat1: int = 1
    dat2: str = 'something'
    scale: float = 1.5

    def __post_init__(self):
        self._scaled_var = self.alter_scale(self.scale)

    @staticmethod
    def alter_scale(scale_in):
        return scale_in * 2 / 3


class SubClassA:
    def __init__(self, Data):
        self.dc = Data

    def do_it_function(self, x):
        return x ** 2 + self.dc.dat1


class Main:
    def __init__(self, Data=DataContainer, **kwargs):
        self.dc = Data(**kwargs)
        self.di = SubClassA(self.dc)

问题是，我不喜欢把**kwargs传递给我的数据类，因为对我来说这感觉像是糟糕的练习。然而，它的行为完全像我想要的...

解决这个问题的干净方法是什么？

Answer 1

虽然我不完全理解您试图做什么，但一种方法可能是将所有现有模型类转换为数据类，并使用InitVar和field等辅助函数定义属性，如下所示。

from dataclasses import dataclass, InitVar, field


@dataclass
class DataContainer:
    dat1: int = 1
    dat2: str = 'something'
    scale: float = 1.5

    def __post_init__(self):
        self._scaled_var = self.alter_scale(self.scale)

    @staticmethod
    def alter_scale(scale_in):
        return scale_in * 2 / 3


@dataclass
class SubClassA:
    dc: DataContainer

    def do_it_function(self, x):
        return x ** 2 + self.dc.dat1


@dataclass
class Main:
    di: SubClassA = field(init=False)
    data: InitVar[DataContainer] = DataContainer()

    def __post_init__(self, data: DataContainer):
        self.di = SubClassA(data)

    @property
    def dc(self):
        return self.di.dc


m = Main()
print(m)
print()

data = DataContainer(123, 'hello world!')
m2 = Main(data)
print(m2.di)
print(m2.dc)

输出：

Main(di=SubClassA(dc=DataContainer(dat1=1, dat2='something', scale=1.5)))

SubClassA(dc=DataContainer(dat1=123, dat2='hello world!', scale=1.5))
DataContainer(dat1=123, dat2='hello world!', scale=1.5)