Skip to content

Commit 6f54d57

Browse files
committed
Typing fixes for new tree view
1 parent 8b86ad0 commit 6f54d57

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

euporie/core/widgets/tree.py

+5-7
Original file line numberDiff line numberDiff line change
@@ -84,21 +84,19 @@ def format_node(
8484

8585
# Apply mouse_handler to rows with children
8686
if has_children:
87-
row = [(style, text, mouse_handler) for style, text in row]
87+
row = [(style, text, mouse_handler) for (style, text, *_) in row]
8888

8989
result.extend(row)
9090

9191
if is_expanded and has_children:
92-
if isinstance(data, dict):
93-
items = data.items()
94-
else: # list
95-
items = enumerate(data)
92+
if isinstance(data, list):
93+
data = {str(i): v for i, v in enumerate(data)}
9694

97-
for k, v in items:
95+
for k, v in data.items():
9896
new_path = (*path, str(k)) if path else (str(k),)
9997
format_node(v, new_path, indent + 1, k)
10098

101-
format_node(self.data, "", 0, self.title)
99+
format_node(self.data, tuple(), 0, self.title)
102100
return result
103101

104102
def _toggle(

0 commit comments

Comments
 (0)