时间戳

首页 > 解决方案 > 如何获取从同一父类扩展的类中的属性

python - 如何获取从同一父类扩展的类中的属性

问题描述

我定义了一些类

class Animal:
    def do_parent_method():
        pass

class Monkey(Animal):
    pass


class Elephant(Animal):
    pass


@dataclass
class Zoo:
    monkey: Monkey= Monkey()
    elephant: Elephant = Elephant()
    start_time: datetime = None
    name: str = 'Not important at all'

    def data_format(self):
        items = [self.monkey, self.elephant]  # Now I hard code here
        for item in items:
            do_something()

关键是关于如何获取Zoo类中的属性

也许有一天，我们会在代码中添加另一种动物

@dataclass
class Zoo:
    monkey: Monkey= Monkey()
    elephant: Elephant = Elephant()
    start_time: datetime = None
    name: str = 'Not important at all'

    def data_format(self):
        items = [get the attributes that extends from Animal]  # How to do?
        for item in items:
            do_parent_method()

现在我只想items成为一个列表，以便我可以循环它。

或者如果你有另一个好主意对我也有好处。

笔记：

Zoom 类中的所有属性都只有一些 str、datetime、int 类型。所有其他实例将是 Animal 类的子类。

固定的：

不小心输入了“zoo”变成了“zoom”

标签： pythonpython-3.xpython-dataclasses

解决方案

该dataclasses.fields函数可以返回有关类的字段信息，包括每个字段的名称和类型。所以你的list理解可以写成：

items = [getattr(self, field.name) for field in fields(self) if issubclass(field.type, Animal)]

这里的缺陷是它不适用于字符串注释，包括模块使用的所有情况from __future__ import annotations。您可以使用此处的技巧来解析为实际类型，或者您可以无条件地获取所有字段，然后使用isinstance检查过滤它们（验证运行时类型，而不是在运行时可以轻松忽略的注释类型）：

items = [attr for attr in (getattr(self, field.name) for field in fields(self)) if isinstance(attr, Animal)]

推荐阅读