def create_widget(parent,
control_name,
control_value,
trait,
label_class=None,
max_items=0):
""" Method to create the list widget.
Parameters
----------
parent: QWidget (mandatory)
the parent widget
control_name: str (mandatory)
the name of the control we want to create
control_value: list of items (mandatory)
the default control value
trait: Tait (mandatory)
the trait associated to the control
label_class: Qt widget class (optional, default: None)
the label widget will be an instance of this class. Its constructor
will be called using 2 arguments: the label string and the parent
widget.
max_items: int (optional)
display at most this number of items. Defaults to 0: no limit.
Returns
-------
out: 2-uplet
a two element tuple of the form (control widget: ,
associated labels: (a label QLabel, the tools QWidget))
"""
# Get the inner trait: expect only one inner trait
# note: trait.inner_traits might be a method (ListInt) or a tuple
# (List), whereas trait.handler.inner_trait is always a method
if len(trait.handler.inner_traits()) == 1:
inner_trait = trait.handler.inner_traits()[0]
elif len(trait.handler.inner_traits()) == 0:
# maybe a generic list, or a compount trait
if hasattr(trait.handler, 'handlers') \
and len(trait.handler.handlers) > 0 \
and hasattr(trait.handler.handlers[0], 'inner_traits') \
and len(trait.handler.handlers[0].inner_traits()) > 0:
inner_trait = trait.handler.handlers[0].inner_traits()[0]
else:
# can't determine type, fallback to string
inner_trait = traits.Str()
else:
raise Exception(
"Expect only one inner trait in List control. Trait '{0}' "
"inner trait is '{1}'.".format(control_name,
trait.handler.inner_traits()))
if control_value is traits.Undefined:
control_value = []
# Create the list widget: a frame
parent = get_ref(parent)
frame = QtGui.QFrame(parent=parent)
#frame.setFrameShape(QtGui.QFrame.StyledPanel)
frame.setFrameShape(QtGui.QFrame.NoFrame)
# Create tools to interact with the list widget: expand or collapse -
# add a list item - remove a list item
tool_widget = QtGui.QWidget(parent)
layout = QtGui.QHBoxLayout()
layout.addStretch(1)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(2)
tool_widget.setLayout(layout)
# Create the tool buttons
resize_button = QtGui.QToolButton()
add_button = QtGui.QToolButton()
delete_button = QtGui.QToolButton()
layout.addWidget(resize_button)
layout.addWidget(add_button)
layout.addWidget(delete_button)
# Set the tool icons
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/add")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
add_button.setIcon(icon)
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/delete")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
delete_button.setIcon(icon)
icon = QtGui.QIcon()
icon.addPixmap(
QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/nav_down")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
resize_button.setIcon(icon)
resize_button.setFixedSize(30, 22)
add_button.setFixedSize(40, 22)
delete_button.setFixedSize(40, 22)
menu = QtGui.QMenu()
menu.addAction(
'Enter list',
partial(ListControlWidget.enter_list, weak_proxy(parent),
control_name, weak_proxy(frame)))
menu.addAction(
'Load list',
partial(ListControlWidget.load_list, weak_proxy(parent),
control_name, weak_proxy(frame)))
if isinstance(inner_trait.trait_type, traits.File) \
or isinstance(inner_trait.trait_type, traits.Directory):
menu.addAction(
'Select files',
partial(ListControlWidget.select_files, weak_proxy(parent),
control_name, weak_proxy(frame)))
add_button.setMenu(menu)
menu = QtGui.QMenu()
menu.addAction(
'Clear all',
partial(ListControlWidget.clear_all, weak_proxy(parent),
control_name, weak_proxy(frame), trait.trait_type.minlen))
delete_button.setMenu(menu)
# Create a new controller that contains length 'control_value' inner
# trait elements
controller = ListController()
if inner_trait.groups:
del inner_trait.groups
n = max_items
if n == 0:
n = len(control_value)
for cnt, inner_control_values in enumerate(control_value[:n]):
controller.add_trait(str(cnt), inner_trait)
#if inner_trait.groups:
#del trait(str(cnt)).groups
setattr(controller, str(cnt), inner_control_values)
# Create the associated controller widget
controller_widget = ControllerWidget(controller,
parent=frame,
live=True)
controller_widget.setObjectName('inner_controller')
controller_widget.setStyleSheet(
'ControllerWidget#inner_controller { padding: 0px; }')
# Store some parameters in the list widget
frame.inner_trait = inner_trait
frame.trait = trait
frame.controller = controller
frame.controller_widget = controller_widget
frame.connected = False
frame.max_items = max_items
# Add the list controller widget to the list widget
frame.setLayout(controller_widget.layout())
frame.layout().setContentsMargins(0, 0, 0, 0)
frame.setObjectName('inner_frame')
frame.setStyleSheet('QFrame#inner_frame { padding: 0px; }')
# Set some callback on the list control tools
# Resize callback
resize_hook = partial(ListControlWidget.expand_or_collapse,
weak_proxy(frame), weak_proxy(resize_button))
resize_button.clicked.connect(resize_hook)
# Add list item callback
add_hook = partial(ListControlWidget.add_list_item, weak_proxy(parent),
control_name, weak_proxy(frame))
add_button.clicked.connect(add_hook)
# Delete list item callback
delete_hook = partial(ListControlWidget.delete_list_item,
weak_proxy(parent), control_name,
weak_proxy(frame))
delete_button.clicked.connect(delete_hook)
# Create the label associated with the list widget
control_label = trait.label
if control_label is None:
control_label = control_name
if label_class is None:
label_class = QtGui.QLabel
if control_label is not None:
label = label_class(control_label, parent)
else:
label = None
return (frame, (label, tool_widget))
def create_widget(parent, control_name, control_value, trait,
label_class=None):
""" Method to create the dict widget.
Parameters
----------
parent: QWidget (mandatory)
the parent widget
control_name: str (mandatory)
the name of the control we want to create
control_value: dict of items (mandatory)
the default control value
trait: Tait (mandatory)
the trait associated to the control
label_class: Qt widget class (optional, default: None)
the label widget will be an instance of this class. Its constructor
will be called using 2 arguments: the label string and the parent
widget.
Returns
-------
out: 2-uplet
a two element tuple of the form (control widget: ,
associated labels: (a label QLabel, the tools QWidget))
"""
# Get the inner trait: expect only one inner trait
# note: trait.inner_traits might be a method (ListInt) or a tuple
# (List), whereas trait.handler.inner_trait is always a method
if len(trait.handler.inner_traits()) != 2:
raise Exception(
"Expect two inner traits in Dict control. Trait '{0}' "
"inner traits are '{1}'.".format(
control_name, trait.inner_traits))
inner_trait = trait.handler.inner_traits()[1]
# Create the dict widget: a frame
frame = QtGui.QFrame(parent=parent)
frame.setFrameShape(QtGui.QFrame.StyledPanel)
# Create tools to interact with the dict widget: expand or collapse -
# add a dict item - remove a dict item
tool_widget = QtGui.QWidget(parent)
layout = QtGui.QHBoxLayout()
layout.addStretch(1)
tool_widget.setLayout(layout)
# Create the tool buttons
resize_button = QtGui.QToolButton()
add_button = QtGui.QToolButton()
layout.addWidget(resize_button)
layout.addWidget(add_button)
# Set the tool icons
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/add")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
add_button.setIcon(icon)
icon = QtGui.QIcon()
icon.addPixmap(
QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/nav_down")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
resize_button.setIcon(icon)
resize_button.setFixedSize(30, 22)
add_button.setFixedSize(30, 22)
# Create a new controller that contains length 'control_value' inner
# trait elements
controller = DictController()
for name, inner_control_values in six.iteritems(control_value):
controller.add_trait(str(name), inner_trait)
setattr(controller, str(name), inner_control_values)
# Create the associated controller widget
controller_widget = ControllerWidget(controller, parent=frame,
live=True, editable_labels=True)
# Store some parameters in the dict widget
frame.inner_trait = inner_trait
frame.trait = trait
frame.controller = controller
frame.controller_widget = controller_widget
frame.connected = False
# Add the dict controller widget to the dict widget
frame.setLayout(controller_widget.layout())
# Set some callback on the dict control tools
# Resize callback
resize_hook = partial(
DictControlWidget.expand_or_collapse, weak_proxy(frame),
weak_proxy(resize_button))
resize_button.clicked.connect(resize_hook)
# Add dict item callback
add_hook = partial(
DictControlWidget.add_dict_item, parent, control_name, frame)
add_button.clicked.connect(add_hook)
# Create the label associated with the dict widget
control_label = trait.label
if control_label is None:
control_label = control_name
if label_class is None:
label_class = QtGui.QLabel
if control_label is not None:
label = label_class(control_label, parent)
else:
label = None
controller_widget.main_controller_def = (DictControlWidget, parent,
control_name, frame)
return (frame, (label, tool_widget))
def create_widget(parent,
control_name,
control_value,
trait,
label_class=None):
""" Method to create the controller widget.
Parameters
----------
parent: QWidget (mandatory)
the parent widget
control_name: str (mandatory)
the name of the control we want to create
control_value: instance of Controller (mandatory)
the default control value
trait: Tait (mandatory)
the trait associated to the control
label_class: Qt widget class (optional, default: None)
the label widget will be an instance of this class. Its constructor
will be called using 2 arguments: the label string and the parent
widget.
Returns
-------
out: 2-uplet
a two element tuple of the form (control widget: ,
associated labels: (a label QLabel, the tools QWidget))
"""
# Create the list widget: a frame
frame = QtGui.QFrame(parent=parent)
frame.setFrameShape(QtGui.QFrame.StyledPanel)
# Create tools to interact with the list widget: expand or collapse -
# add a list item - remove a list item
tool_widget = QtGui.QWidget(parent)
layout = QtGui.QHBoxLayout()
layout.addStretch(1)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(2)
tool_widget.setLayout(layout)
# Create the tool buttons
resize_button = QtGui.QToolButton()
layout.addWidget(resize_button)
# Set the tool icons
icon = QtGui.QIcon()
icon.addPixmap(
QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/nav_down")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
resize_button.setIcon(icon)
resize_button.setFixedSize(30, 22)
editable_labels = False
if trait.handler.inner_traits():
editable_labels = True
frame.inner_trait = trait.handler.inner_traits()[0]
add_button = QtGui.QToolButton()
delete_button = QtGui.QToolButton()
layout.addWidget(add_button)
# Set the tool icons
icon = QtGui.QIcon()
icon.addPixmap(
QtGui.QPixmap(_fromUtf8(":/soma_widgets_icons/add")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
add_button.setIcon(icon)
add_button.setFixedSize(30, 22)
delete_button.setFixedSize(30, 22)
# Add list item callback
add_hook = partial(ControllerControlWidget.add_item,
weak_proxy(parent), control_name,
weak_proxy(frame))
add_button.clicked.connect(add_hook)
# Create the associated controller widget
controller_widget = ControllerWidget(control_value,
parent=frame,
live=True,
editable_labels=editable_labels)
# Store some parameters in the list widget
frame.trait = trait
frame.controller = control_value
frame.controller_widget = controller_widget
frame.connected = False
# Add the list controller widget to the list widget
frame.setLayout(controller_widget.layout())
# Set some callback on the controller control tools
# Resize callback
resize_hook = partial(ControllerControlWidget.expand_or_collapse,
weak_proxy(frame), weak_proxy(resize_button))
resize_button.clicked.connect(resize_hook)
if getattr(trait, 'expanded') is False:
ControllerControlWidget.set_expanded(frame, resize_button, False)
# Create the label associated with the controller widget
control_label = trait.label
if control_label is None:
control_label = control_name
if label_class is None:
label_class = QtGui.QLabel
if control_label is not None:
label = label_class(control_label, parent)
else:
label = None
return (frame, (label, tool_widget))
