如何在Python Dataclasses中通过单个参数生成多个属性？

问题场景 #

尝试用dataclass通过age字段计算birthyear属性时触发错误，原代码及报错信息如下：

错误代码

from dataclasses import dataclass

@dataclass
class Human:
    name: str
    age: int 
    birthyear:int=2022-age

h1 = Human("john", 35)
print(h1.birthyear)

报错信息

Traceback (most recent call last):
  File "D:\Taha\Programs\Python\projects\UNO\tests.py", line 5, in <module>
    class Human:
  File "D:\Taha\Programs\Python\projects\UNO\tests.py", line 8, in Human
    yearOfBirth :int= 2022-age
NameError: name 'age' is not defined

原因解析 #

dataclass的字段默认值是在类定义阶段计算的，此时age只是类的字段声明，还没有生成实例属性，直接引用会触发NameError。必须在实例初始化完成后再计算派生属性。

解决方案 #

以下两种方式可实现你想要的普通类效果：

方案一：使用__post_init__方法（推荐）

利用dataclass自带的__post_init__钩子，在实例初始化完成后计算birthyear，同时支持可选的自定义birthyear参数：

from dataclasses import dataclass

@dataclass
class Human:
    name: str
    age: int 
    birthyear: int = None

    def __post_init__(self):
        # 未传入birthyear时，用2022减去年龄计算
        if self.birthyear is None:
            self.birthyear = 2022 - self.age

# 测试
h1 = Human("john", 35)
print(h1.birthyear)  # 输出1987
h2 = Human("jane", 25, 1998)
print(h2.birthyear)  # 输出1998（使用自定义值）

方案二：手动重写__init__方法

完全复刻普通类的初始化逻辑，dataclass会优先使用自定义的__init__：

from dataclasses import dataclass

@dataclass
class Human:
    name: str
    age: int 
    birthyear: int = 2022

    def __init__(self, name: str, age: int, birthyear: int = 2022):
        self.name = name
        self.age = age
        self.birthyear = birthyear - age

# 测试
h1 = Human("john", 35)
print(h1.birthyear)  # 输出1987
h2 = Human("jane", 25, 2023)
print(h2.birthyear)  # 输出1998

内容的提问来源于stack exchange，提问作者CodeDude