forked from Einswilli/KivymdStudio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tree.py
38 lines (31 loc) · 1.05 KB
/
tree.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import sys
from PySide6.QtWidgets import QApplication, QTreeWidget, QTreeWidgetItem
data = {"Project A": ["file_a.py", "file_a.txt", "something.xls"],
"Project B": ["file_b.csv", "photo.jpg"],
"Project C": []}
# tree = QTreeWidget()
# tree.setColumnCount(2)
# tree.setHeaderLabels(["Name", "Type"])
class FolderTree(QTreeWidget):
def __init__(self, parent=None):
super().__init__(parent=parent)
self.setColumnCount(1)
self.setHeaderHidden(True)
self.children=[]
def addchildren(self,child):
items = []
for key, values in data.items():
item = QTreeWidgetItem([key])
for value in values:
ext = value.split(".")[-1].upper()
child = QTreeWidgetItem([value, ext])
item.addChild(child)
items.append(item)
self.children=items
self.insertTopLevelItems(0, self.children)
return self.children
#app = QApplication()
#tree=FolderTree()
#tree.addchildren(data)
#tree.show()
#sys.exit(app.exec())