初始化属性的功能

时间:2019-01-21 13:42:14

标签: python-3.x namedtuple python-dataclasses

我需要在初始化属性之前调用一个函数,最好使用python3.7数据类,也可以命名为tuple。

这段代码做了类似的事情,属性名称只被输入了一次,但是非常混乱,我不知道如何支持默认值。我敢肯定,有一种更好,更pythonic的方法可以做到这一点。

from collections import namedtuple

person = {
    'name': (lambda value: value.upper()),
    'age': (lambda value: value*2),
}



class Person(namedtuple("Person", person.keys())):
    def __new__(cls, **kwargs):
        return super().__new__(cls, **{k:person[k](v) for k, v in kwargs.items() if v is not None and k in person})


l = Person(**{'name': 'gesiel', 'age': 34})
print(l)

1 个答案:

答案 0 :(得分:1)

最简单的方法是让常规dataclass带有帮助函数,该帮助函数在构建Person对象之前进行转换。

@dataclass
class Person:
    age: int
    name: str

def make_person(age, name):
    return Person(age*2, name.upper())

person = make_person(100, 'patrick')
print(person.age, person.name)
# 200 PATRICK

您的另一个选择是使用Post-init processing来更改值。

@dataclass
class Person:
    age: int
    name: str
    def __post_init__(self):
        self.age  = self.age * 2
        self.name = self.name.upper()

person = Person(100, 'patrick')
print(person.age, person.name)
# 200 PATRICK