Make子类有自己的class属性

时间:2016-01-11 18:04:48

标签: python

我有一个班级Generic

class Generic:
    raw_data = []
    objects = dict()

和具体课程

class A(Generic):
    raw_data = module.somethingA

class B(Generic):
    raw_data = module.somethingB

我想将每个raw_data填充到班级的每个objects词典中。 为此,我正在跑步:

for object_type in (A, B):
     for data in object_type.raw_data:
         new_object = object_type(*data)
         object_type.objects[id(new_object)] = new_object

但是,这不起作用,因为objectsA之间共享B,我希望Generic的每个子类都有自己的对象。

如何在不必在每个子类上键入objects = dict()的情况下实现此目的?

我倾向于说这是一个需要元类的传统情况(为每个新类添加objects);是这种情况,还是有更简单的选择?

2 个答案:

答案 0 :(得分:2)

使用元类或使用类装饰器。

类装饰器可以简单地创建属性:

def add_objects(cls):
    cls.objects = {}
    return cls

@add_objects
class A(generic):
    raw_data = module.somethingA

然而,这并没有真正添加任何东西;您只需将一行(objects = {})替换为另一行(@add_objects)。

您可以在循环中添加对象:

for object_type in (A, B):
     if 'objects' not in vars(object_type):
         object_type.objects = {}
     for data in object_type.raw_data:
         new_object = object_type(*data)
         object_type.objects[id(new_object)] = new_object

或复制它(读取属性可能会检索父类属性或直接属性,这里没关系):

for object_type in (A, B):
     object_type.objects = object_type.objects.copy()
     for data in object_type.raw_data:
         new_object = object_type(*data)
         object_type.objects[id(new_object)] = new_object

或从头开始创建字典:

for object_type in (A, B):
     object_type.object = {
         id(new_object): new_object
         for data in object_type.raw_data
         for new_object in (object_type(*data),)}

答案 1 :(得分:1)

我认为这里不需要元类。为什么不在填充循环中的每个子类之前复制父类对象?

for object_type in (A, B):
     # copy Generic.objects onto object_type.objects here
     object_type.objects = Generic.objects.copy()
     for data in object_type.raw_data:
         new_object = object_type(*data)
         object_type.objects[id(new_object)] = new_object

此外,如果需要,您可以修改为使用super和/或deepcopy