在 Python class 中创建一个增量 ID，子类各自维护自己的 ID 系列

Question

我想创建一个 class，其中 class 的每个 object 都被赋予一个从0开始的唯一顺序 ID。 我还希望每个子类都具有从0开始的顺序 ID，编号不受父类或兄弟类的影响。

使用类似问题的答案，我创建了 class Foo

class Foo:
    id_iter = itertools.count()
    def __init__(self):
        self.id = next(self.id_iter)

Foo的所有实例都将具有顺序id值。 我还想创建一个 class Bar继承自Foo这样

>>> Foo().id
0
>>> Foo().id
1
>>> Foo().id
2
>>> Bar().id  # The first Bar id starts at 0 rather than 3
0

如果我像这样创建Bar ：

class Bar(Foo):
    def __init__(self):
        super(Bar, self).__init__()

上面对Bar().id的调用将返回3 ，因为两个类都使用相同的迭代器。 或者，我可以将id_iter = itertools.count()添加回我的Bar定义并获得我期望的 output。 但是，我正在制作Foo的多个子类，并且不想将该行添加到每个 class； 我希望Foo的所有子类都自动继承 ID 功能，而不必向子类定义添加任何额外内容。 我怎样才能做到这一点？

Answer 1

您不能使用 class 属性执行此操作，因为 class 属性不会被继承。 一种方法可能是使用单独的字典来保持类的计数。 例如：

import itertools


class ClassCounter(object):
    counters = {}

    @classmethod
    def get_counter(cls, class_to_count):
        cls.counters.setdefault(class_to_count, itertools.count())
        return next(cls.counters[class_to_count])


class Foo:
    def __init__(self):
        self.id = ClassCounter.get_counter(self.__class__)


class Bar(Foo):
    def __init__(self):
        super(Bar, self).__init__()


if __name__ == '__main__':
    foo1 = Foo()
    foo2 = Foo()
    foo3 = Foo()
    bar1 = Bar()
    bar2 = Bar()
    print(f"{foo1.id=}, {foo2.id=}, {foo3.id=}, {bar1.id=}, {bar2.id=}")

这应该打印：

foo1.id=0, foo2.id=1, foo3.id=2, bar1.id=0, bar2.id=1