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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
"""
Copyright (c) 2015, Jairus Martin.
Distributed under the terms of the MIT License.
The full license is in the file COPYING.txt, distributed with this software.
Created on Jun 3, 2015
"""
from atom.api import Bool, ContainerList, ForwardTyped, Int, Property, Typed, observe
from enaml.core.declarative import d_
from enamlx.widgets.abstract_item import (
AbstractWidgetItem,
ProxyAbstractWidgetItem,
ProxyAbstractWidgetItemGroup,
)
from enamlx.widgets.abstract_item_view import AbstractItemView, ProxyAbstractItemView
class ProxyTreeView(ProxyAbstractItemView):
declaration = ForwardTyped(lambda: TreeView)
class ProxyTreeViewColumn(ProxyAbstractWidgetItemGroup):
declaration = ForwardTyped(lambda: TreeViewColumn)
def set_column(self, column):
raise NotImplementedError
class ProxyTreeViewItem(ProxyAbstractWidgetItem):
declaration = ForwardTyped(lambda: TreeViewItem)
def refresh_model(self, schange):
raise NotImplementedError
class TreeView(AbstractItemView):
#: Proxy widget
proxy = Typed(ProxyTreeView)
#: Show root node
show_root = d_(Bool(True))
@observe("show_root")
def _update_proxy(self, change):
"""An observer which sends state change to the proxy."""
# The superclass handler implementation is sufficient.
super(TreeView, self)._update_proxy(change)
def child_added(self, child):
super(TreeView, self).child_added(child)
self._update_rows()
def child_removed(self, child):
super(TreeView, self).child_removed(child)
self._update_rows()
def _update_rows(self):
for r, item in enumerate(self._items):
item.row = r
class TreeViewItem(AbstractWidgetItem):
#: Proxy reference
proxy = Typed(ProxyTreeViewItem)
#: The child items
items = d_(ContainerList(default=[]))
#: First visible row
visible_row = d_(Int(0))
#: Number of rows visible
visible_rows = d_(Int(100))
#: First visible column
visible_column = d_(Int(0))
#: Number of columns visible
visible_columns = d_(Int(1))
def _get_items(self):
"""Items should be a list of child TreeViewItems excluding
columns.
"""
return [c for c in self.children if isinstance(c, TreeViewItem)]
def _get_columns(self):
"""List of child TreeViewColumns including
this item as the first column
"""
return [self] + [c for c in self.children if isinstance(c, TreeViewColumn)]
#: Columns
_columns = Property(lambda self: self._get_columns(), cached=True)
def child_added(self, child):
super(TreeViewItem, self).child_added(child)
self.get_member("_columns").reset(self)
self._update_rows()
def child_removed(self, child):
super(TreeViewItem, self).child_removed(child)
self.get_member("_columns").reset(self)
self._update_rows()
def _update_rows(self):
"""Update the row and column numbers of child items."""
for row, item in enumerate(self._items):
item.row = row # Row is the Parent item
item.column = 0
for column, item in enumerate(self._columns):
item.row = self.row # Row is the Parent item
item.column = column
class TreeViewColumn(AbstractWidgetItem):
"""Use this to build a table by defining the columns."""
#: Proxy reference
proxy = Typed(ProxyTreeViewColumn)
|
