我有一个班级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
但是,这不起作用,因为objects
和A
之间共享B
,我希望Generic
的每个子类都有自己的对象。
如何在不必在每个子类上键入objects = dict()
的情况下实现此目的?
我倾向于说这是一个需要元类的传统情况(为每个新类添加objects
);是这种情况,还是有更简单的选择?
答案 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
。