这种情况确实需要重构。您正在使用未设计为容器的对象。取而代之的是,使用一个容器,例如 dict 或从 dict 继承的类。
如果您必须使用当前设置,我同意Blckknght的观点,最有希望的方法似乎是使用 dir。
class CCollection(object):
def get_children_strings(self):
list_of_strings = []
for attr_name in dir(self):
if attr_name not in dir(CCollection()):
attr = getattr(self, attr_name)
if hasattr(attr, 'get_children_strings'):
list_of_strings.extend(["." + attr_name + child_string for child_string in attr.get_children_strings()])
else:
list_of_strings.append("." + attr_name + " = " + str(attr))
return list_of_strings
def print_tree(self, prefix):
print [prefix + s for s in self.get_children_strings()]
那么你就可以
m = CCollection()
m.a = CCollection()
m.a.b = CCollection()
m.a.b.x = 1
m.a.b.y = 2
m.a.c = 3
m.d = 4
m.print_tree("m")
m.a.print_tree("m.a")
m.a.b.print_tree("m.a.b")
并获得输出:
>>> m.print_tree("m")
['m.a.b.x = 1', 'm.a.b.y = 2', 'm.a.c = 3', 'm.d = 4']
>>> m.a.print_tree("m.a")
['m.a.b.x = 1', 'm.a.b.y = 2', 'm.a.c = 3']
>>> m.a.b.print_tree("m.a.b")
['m.a.b.x = 1', 'm.a.b.y = 2']
为了更进一步,您可能希望使用具有树遍历函数的类。如果您有一个获取父节点的函数、没有循环的保证以及一个保存节点名称的类变量,您可以自动生成当前通过函数的prefix
参数传入的信息。print_tree