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
|
#------------------------------------------------------------------------------
#
# Copyright (c) 2016, Enthought, Inc.
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in enthought/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!
#
# Author: Enthought, Inc.
#
#------------------------------------------------------------------------------
from pyface.qt import QtCore, QtGui
from traits.api import Any, List, Str, provides
from pyface.constant import CANCEL
from pyface.i_single_choice_dialog import ISingleChoiceDialog, MSingleChoiceDialog
from .dialog import Dialog, _RESULT_MAP
@provides(ISingleChoiceDialog)
class SingleChoiceDialog(MSingleChoiceDialog, Dialog):
""" A dialog that allows the user to chose a single item from a list.
Note that due to limitations of the QInputDialog class, the cancel trait
is ignored, and the list of displayed strings must be unique.
"""
#### 'ISingleChoiceDialog' interface ######################################
#: List of objects to choose from.
choices = List(Any)
#: The object chosen, if any.
choice = Any
#: An optional attribute to use for the name of each object in the dialog.
name_attribute = Str
#: The message to display to the user.
message = Str
def set_dialog_choice(self, choice):
if self.control is not None:
if self.name_attribute != '':
choice = getattr(choice, self.name_attribute)
self.control.setTextValue(str(choice))
###########################################################################
# Protected 'IDialog' interface.
###########################################################################
def _create_contents(self, parent):
""" Creates the window contents. """
# In this case, Qt does it all for us in 'QInputDialog'
pass
def _show_modal(self):
self.control.setWindowModality(QtCore.Qt.ApplicationModal)
retval = self.control.exec_()
if self.control is None:
# dialog window closed, treat as Cancel, nullify choice
retval = CANCEL
else:
retval = _RESULT_MAP[retval]
return retval
###########################################################################
# 'IWindow' interface.
###########################################################################
def close(self):
""" Closes the window. """
# Get the chosen object.
if self.control is not None and self.return_code != CANCEL:
text = self.control.textValue()
choices = self._choice_strings()
if text in choices:
idx = self._choice_strings().index(text)
self.choice = self.choices[idx]
else:
self.choice = None
# Let the window close as normal.
super(SingleChoiceDialog, self).close()
###########################################################################
# Protected 'IWidget' interface.
###########################################################################
def _create_control(self, parent):
""" Create the toolkit-specific control that represents the window. """
dialog = QtGui.QInputDialog(parent)
dialog.setOption(QtGui.QInputDialog.UseListViewForComboBoxItems, True)
dialog.setWindowTitle(self.title)
dialog.setLabelText(self.message)
# initialize choices: set initial value to first choice
choices = self._choice_strings()
dialog.setComboBoxItems(choices)
dialog.setTextValue(choices[0])
if self.size != (-1, -1):
self.resize(*self.size)
if self.position != (-1, -1):
self.move(*self.position)
return dialog
|
