Несколько дней назад я искал такую тему, но все, что я мог найти - как заполнить дерево, а не как его извлечь. Поэтому я пишу некоторые. Может быть, кто-то еще найдет это полезным. Спасибо предыдущему автору, я использовал его код с некоторыми улучшениями. Это загрузит вашу коллекцию (это может быть список, словарь или кортеж со многими дочерними элементами)
def load_tree(self, d):
self.fill_widget(self.tree, d)
def fill_widget(self, widget, value):
widget.clear()
self.fill_item(widget.invisibleRootItem(), value)
def fill_item(self, item, value):
def new_item(parent, text):
child = QTreeWidgetItem([str(text)])
if text not in ("[dict]", "[list]", "[tuple]"):
child.setFlags(child.flags() | Qt.ItemIsEditable)
parent.addChild(child)
child.setExpanded(True)
return child
if isinstance(value, dict):
new_parent = new_item(item, f"[{value.__class__.__name__}]")
for elem_k, elem_v in value.items():
sub_parent = new_item(new_parent, elem_k)
self.fill_item(sub_parent, elem_v)
elif isinstance(value, (tuple, list)):
new_parent = new_item(item, f"[{value.__class__.__name__}]")
for val in value:
self.fill_item(new_parent, val)
else:
new_item(item, f"{value}")
И извлеките код немного сложнее. Если кто-то может сделать его более причудливым - пожалуйста.
def get_dict(self):
result = None
def unpack(to_unpack, key, source=None):
for child_index in range(to_unpack.childCount()):
child = to_unpack.child(child_index)
child_text = child.text(0)
try:
child_text = float(child_text)
except ValueError:
try:
child_text = int(child_text)
except ValueError:
pass
if source is None:
core = result
else:
core = source
if key == "[dict]":
core.update({child_text: None})
if child.childCount() > 0:
unpack(child, child_text, core)
elif key == "[list]" or key == "[tuple]":
if child_text == "[dict]":
core.append({})
elif child_text == "[list]" or child_text == "[tuple]":
core.append([])
else:
core.append(child_text)
if child.childCount() > 0:
unpack(child, child_text, core[child_index])
else:
if child_text == "[dict]":
core.update({key: {}})
elif child_text == "[list]" or child_text == "[tuple]":
core.update({key: []})
else:
core.update({key: child_text})
if child.childCount() > 0:
unpack(child, child_text, core[key])
for index in range(self.tree.topLevelItemCount()):
parent = self.tree.topLevelItem(index)
element_text = parent.text(0)
if element_text == "[dict]":
result = {}
unpack(parent, element_text)
elif element_text == "[list]" or element_text == "[tuple]":
result = []
unpack(parent, element_text)
else:
result = element_text
return result
Где self.tree - объект QTreeWidget в вашем окне.
person
Windy
schedule
31.08.2018
QTreeWidget, вам действительно следует использоватьQTreeView, поддерживаемый моделью. В противном случае вам придется постоянно заново заполнять все дерево, чтобы отражать изменения. - person anonymous schedule 07.09.2017