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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
# (C) Copyright 2004-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
"""
This is a modified version of the Standard_Editors/TreeEditor_demo.py in which
the `Employee` class is now a subclass of HasRequiredTraits. Thus, to create a
new `TreeNode` for `Employee`, to create the associated `Employee` object we
can not simply do `Employee()`. Luckily the `TreeNode` API provides a means for
specifying a callable that returns a created instance when we want to add a new
node. To do so we can simply pass a tuple of the for (klass, prompt, factorry)
as an item the `add` list trait for a TreeNode. klass is the class of object we
want to be addable, prompt is a boolean indicating whether or not we want to
prompt the user to specify traits after we instantiate the object before it is
added to the tree, and factory is the previously described callable that must
return an instance of klass.
"""
from traits.api import HasRequiredTraits, HasTraits, Str, Regex, List, Instance
from traitsui.api import Item, View, TreeEditor, TreeNode
class Employee(HasRequiredTraits):
"""Defines a company employee."""
name = Str('<unknown>')
title = Str()
phone = Regex(regex=r'\d\d\d-\d\d\d\d')
def default_title(self):
self.title = 'Senior Engineer'
def create_default_employee():
return Employee(name="Dilbert", title="Engineer", phone="999-8212")
class Department(HasTraits):
"""Defines a department with employees."""
name = Str('<unknown>')
employees = List(Employee)
class Company(HasTraits):
"""Defines a company with departments and employees."""
name = Str('<unknown>')
departments = List(Department)
employees = List(Employee)
# Create an empty view for objects that have no data to display:
no_view = View()
# Define the TreeEditor used to display the hierarchy:
tree_editor = TreeEditor(
nodes=[
# The first node specified is the top level one
TreeNode(
node_for=[Company],
auto_open=True,
# child nodes are
children='',
label='name', # label with Company name
view=View(['name']),
),
TreeNode(
node_for=[Company],
auto_open=True,
children='departments',
label='=Departments', # constant label
view=no_view,
add=[Department],
),
TreeNode(
node_for=[Company],
auto_open=True,
children='employees',
label='=Employees', # constant label
view=no_view,
add=[(Employee, True, create_default_employee)],
),
TreeNode(
node_for=[Department],
auto_open=True,
children='employees',
label='name', # label with Department name
view=View(['name']),
add=[(Employee, True, create_default_employee)],
),
TreeNode(
node_for=[Employee],
auto_open=True,
label='name', # label with Employee name
view=View(['name', 'title', 'phone']),
),
]
)
class Partner(HasTraits):
"""Defines a business partner."""
name = Str('<unknown>')
company = Instance(Company)
traits_view = View(
Item(name='company', editor=tree_editor, show_label=False),
title='Company Structure',
buttons=['OK'],
resizable=True,
style='custom',
width=0.3,
height=500,
)
# Create an example data structure:
jason = Employee(name='Jason', title='Senior Engineer', phone='536-1057')
mike = Employee(name='Mike', title='Senior Engineer', phone='536-1057')
dave = Employee(name='Dave', title='Senior Developer', phone='536-1057')
martin = Employee(name='Martin', title='Senior Engineer', phone='536-1057')
duncan = Employee(name='Duncan', title='Consultant', phone='526-1057')
# Create the demo:
demo = Partner(
name='Enthought, Inc.',
company=Company(
name='Enthought',
employees=[dave, martin, duncan, jason, mike],
departments=[
Department(name='Business', employees=[jason, mike]),
Department(name='Scientific', employees=[dave, martin, duncan]),
],
),
)
# Run the demo (if invoked from the command line):
if __name__ == '__main__':
demo.configure_traits()
|
