Python：继承自基础Dataclass的Dataclass，如何将值从base升级到新类？

Question

如何将基础数据类中的值升级为从其继承的值？

示例（Python 3.7.2）

from dataclasses import dataclass

@dataclass
class Person:
    name: str 
    smell: str = "good"    

@dataclass
class Friend(Person):

    # ... more fields

    def say_hi(self):        
        print(f'Hi {self.name}')

friend = Friend(name='Alex')
f1.say_hi()

打印“Hi Alex”

random_stranger = Person(name = 'Bob', smell='OK')

return for random_stranger“Person（name ='Bob'，smell ='OK'）”

如何将random_stranger变成朋友？

Friend(random_stranger)

返回“朋友（姓名=人（姓名='鲍勃'，气味='确定'），气味='好'）”

我想得到“朋友（名字='鲍勃'，气味='确定'）”的结果。

Friend(random_stranger.name, random_stranger.smell)

有效，但我如何避免复制所有字段？

或者我可能无法在从数据类继承的类上使用@dataclass装饰器？

Answer 1

您可能不希望class本身是一个可变属性，而是使用诸如枚举之类的东西来表示这样的状态。根据要求，您可以考虑以下几种模式之一：

class RelationshipStatus(Enum):
    STRANGER = 0
    FRIEND = 1
    PARTNER = 2

@dataclass
class Person(metaclass=ABCMeta):
    full_name: str
    smell: str = "good"
    status: RelationshipStatus = RelationshipStatus.STRANGER

@dataclass
class GreetablePerson(Person):
    nickname: str = ""

    @property
    def greet_name(self):
        if self.status == RelationshipStatus.STRANGER:
            return self.full_name
        else:
            return self.nickname

    def say_hi(self):
        print(f"Hi {self.greet_name}")

if __name__ == '__main__':
    random_stranger = GreetablePerson(full_name="Robert Thirstwilder",
                                      nickname="Bobby")
    random_stranger.status = RelationshipStatus.STRANGER
    random_stranger.say_hi()
    random_stranger.status = RelationshipStatus.FRIEND
    random_stranger.say_hi()

您也可能希望以trait / mixin风格实现此功能。而不是创建一个GreetablePerson，而是创建一个类Greetable，也是抽象的，并使你的具体类继承这两者。

您也可以考虑使用优秀的，后向的，更灵活的attrs包。这还可以让您使用evolve()函数创建一个新对象：

friend = attr.evolve(random_stranger, status=RelationshipStatus.FRIEND)

Answer 2

您要求的是factory method pattern实现的，并且可以使用@classmethod关键字直接在python类中实现。

只需在基类定义中包含一个dataclass工厂方法方法，如下所示：

import dataclasses

@dataclasses.dataclass
class Person:
    name: str
    smell: str = "good"

    @classmethod
    def from_instance(cls, instance):
        return cls(**dataclasses.asdict(instance))

从这个基类继承的任何新数据类现在都可以创建彼此的实例[* 1]，如下所示：

@dataclasses.dataclass
class Friend(Person):
    def say_hi(self):        
        print(f'Hi {self.name}')

random_stranger = Person(name = 'Bob', smell='OK')
friend = Friend.from_instance(random_stranger)
print(friend.say_hi())
# "Hi Bob"

---

[1]：如果您的子类引入没有默认值的新字段，或者您尝试从子类实例创建父类实例，则它将不起作用。

Answer 3

快速破解：

random_stranger = Person(name = 'Bob', smell='OK')
friend = Friend(**random_stranger.__dict__)

这将获取random_stranger的所有属性并将它们传递给Friend构造函数。