class SelectDialog(QDialog):
'''
This dialog creates an input mask for a string list and return selected entries.
'''
def __init__(self,
items=list(),
buttons=QDialogButtonBox.Cancel | QDialogButtonBox.Ok,
exclusive=False,
preselect_all=False,
title='',
description='',
icon='',
parent=None,
select_if_single=True,
checkitem1='',
checkitem2='',
closein=0):
'''
Creates an input dialog.
@param items: a list with strings
@type items: C{list()}
'''
QDialog.__init__(self, parent=parent)
self.setObjectName(' - '.join(['SelectDialog', utf8(items)]))
self.verticalLayout = QVBoxLayout(self)
self.verticalLayout.setObjectName("verticalLayout")
self.verticalLayout.setContentsMargins(1, 1, 1, 1)
# add filter row
self.filter_frame = QFrame(self)
filterLayout = QHBoxLayout(self.filter_frame)
filterLayout.setContentsMargins(1, 1, 1, 1)
label = QLabel("Filter:", self.filter_frame)
self.filter_field = EnchancedLineEdit(self.filter_frame)
filterLayout.addWidget(label)
filterLayout.addWidget(self.filter_field)
self.filter_field.textChanged.connect(self._on_filter_changed)
self.verticalLayout.addWidget(self.filter_frame)
if description:
self.description_frame = QFrame(self)
descriptionLayout = QHBoxLayout(self.description_frame)
# descriptionLayout.setContentsMargins(1, 1, 1, 1)
if icon:
self.icon_label = QLabel(self.description_frame)
self.icon_label.setSizePolicy(QSizePolicy.Fixed,
QSizePolicy.Fixed)
self.icon_label.setPixmap(
QPixmap(icon).scaled(30, 30, Qt.KeepAspectRatio))
descriptionLayout.addWidget(self.icon_label)
self.description_label = QLabel(self.description_frame)
self.description_label.setWordWrap(True)
self.description_label.setText(description)
descriptionLayout.addWidget(self.description_label)
self.verticalLayout.addWidget(self.description_frame)
# create area for the parameter
self.content = MainBox(self)
if items:
self.scroll_area = QScrollArea(self)
self.scroll_area.setFocusPolicy(Qt.NoFocus)
self.scroll_area.setObjectName("scroll_area")
self.scroll_area.setWidgetResizable(True)
self.scroll_area.setWidget(self.content)
self.verticalLayout.addWidget(self.scroll_area)
self.checkitem1 = checkitem1
self.checkitem1_result = False
self.checkitem2 = checkitem2
self.checkitem2_result = False
# add select all option
if not exclusive and items:
self._ignore_next_toggle = False
self.select_all_checkbox = QCheckBox('all entries')
self.select_all_checkbox.setTristate(True)
self.select_all_checkbox.stateChanged.connect(
self._on_select_all_checkbox_stateChanged)
self.verticalLayout.addWidget(self.select_all_checkbox)
self.content.toggled.connect(self._on_main_toggle)
if self.checkitem1:
self.checkitem1_checkbox = QCheckBox(self.checkitem1)
self.checkitem1_checkbox.stateChanged.connect(
self._on_select_checkitem1_checkbox_stateChanged)
self.verticalLayout.addWidget(self.checkitem1_checkbox)
if self.checkitem2:
self.checkitem2_checkbox = QCheckBox(self.checkitem2)
self.checkitem2_checkbox.stateChanged.connect(
self._on_select_checkitem2_checkbox_stateChanged)
self.verticalLayout.addWidget(self.checkitem2_checkbox)
if not items:
spacerItem = QSpacerItem(1, 1, QSizePolicy.Expanding,
QSizePolicy.Expanding)
self.verticalLayout.addItem(spacerItem)
self._close_timer = None
self._closein = closein - 1
if closein > 0:
self.closein_label = QLabel("OK in %d sec..." % closein)
self.closein_label.setAlignment(Qt.AlignRight)
self.verticalLayout.addWidget(self.closein_label)
self._close_timer = threading.Timer(1.0, self._on_close_timer)
self._close_timer.start()
# create buttons
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.setObjectName("buttonBox")
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(buttons)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.verticalLayout.addWidget(self.buttonBox)
# set the input fields
if items:
self.content.createFieldsFromValues(items, exclusive)
if (select_if_single and len(items) == 1) or preselect_all:
self.select_all_checkbox.setCheckState(Qt.Checked)
if not items or len(items) < 7:
self.filter_frame.setVisible(False)
# print '=============== create', self.objectName()
#
# def __del__(self):
# print "************ destroy", self.objectName()
def _on_main_toggle(self, state):
self.cancel_autoclose()
self._ignore_next_toggle = state != self.select_all_checkbox.checkState(
)
self.select_all_checkbox.setCheckState(state)
def _on_select_all_checkbox_stateChanged(self, state):
self.cancel_autoclose()
if not self._ignore_next_toggle:
self.content.setState(state)
self._ignore_next_toggle = False
def _on_select_checkitem1_checkbox_stateChanged(self, state):
self.cancel_autoclose()
if state == Qt.Checked:
self.checkitem1_result = True
elif state == Qt.Unchecked:
self.checkitem1_result = False
def _on_select_checkitem2_checkbox_stateChanged(self, state):
self.cancel_autoclose()
if state == Qt.Checked:
self.checkitem2_result = True
elif state == Qt.Unchecked:
self.checkitem2_result = False
def _on_filter_changed(self):
self.content.filter(self.filter_field.text())
def _on_close_timer(self):
self.closein_label.setText("OK in %d sec..." % self._closein)
if self._closein == 0:
self.buttonBox.accepted.emit()
return
self._closein -= 1
self._close_timer = threading.Timer(1.0, self._on_close_timer)
self._close_timer.start()
def cancel_autoclose(self):
if self._close_timer is not None:
self._close_timer.cancel()
self.closein_label.setVisible(False)
def getSelected(self):
return self.content.getSelected()
@staticmethod
def getValue(title,
description='',
items=list(),
exclusive=False,
preselect_all=False,
icon='',
parent=None,
select_if_single=True,
checkitem1='',
checkitem2='',
closein=0):
selectDia = SelectDialog(items,
exclusive=exclusive,
preselect_all=preselect_all,
description=description,
icon=icon,
parent=parent,
select_if_single=select_if_single,
checkitem1=checkitem1,
checkitem2=checkitem2,
closein=closein)
selectDia.setWindowTitle(title)
selectDia.resize(480, 256)
if selectDia.exec_():
if selectDia.checkitem2:
return selectDia.getSelected(
), True, selectDia.checkitem1_result, selectDia.checkitem2_result
if selectDia.checkitem1:
return selectDia.getSelected(
), True, selectDia.checkitem1_result
return selectDia.getSelected(), True
if selectDia.checkitem2:
return list(), False, False, False
if selectDia.checkitem1:
return list(), False, False
return list(), False
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%% close handling %%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
def accept(self):
self.cancel_autoclose()
self.setResult(QDialog.Accepted)
self.hide()
def reject(self):
self.cancel_autoclose()
self.setResult(QDialog.Rejected)
self.hide()
def hideEvent(self, event):
self.close()
def closeEvent(self, event):
'''
Test the open files for changes and save this if needed.
'''
self.cancel_autoclose()
self.setAttribute(Qt.WA_DeleteOnClose, True)
QDialog.closeEvent(self, event)
class MessageBox(QDialog):
NoIcon = 0
Information = 1
Warning = 2
Critical = 3
Question = 4
NoButton = 0
Ok = 1 # An "OK" button defined with the AcceptRole .
Open = 2 # A "Open" button defined with the AcceptRole .
Save = 4 # A "Save" button defined with the AcceptRole .
Cancel = 8 # A "Cancel" button defined with the RejectRole .
Close = 16 # A "Close" button defined with the RejectRole .
Discard = 32 # A "Discard" or "Don't Save" button, depending on the platform, defined with the DestructiveRole .
Apply = 64 # An "Apply" button defined with the ApplyRole .
Reset = 128 # A "Reset" button defined with the ResetRole .
RestoreDefaults = 256 # A "Restore Defaults" button defined with the ResetRole .
Help = 512 # A "Help" button defined with the HelpRole .
SaveAll = 1024 # A "Save All" button defined with the AcceptRole .
Yes = 2048 # A "Yes" button defined with the YesRole .
YesToAll = 4096 # A "Yes to All" button defined with the YesRole .
No = 8192 # A "No" button defined with the NoRole .
NoToAll = 16384 # A "No to All" button defined with the NoRole .
Abort = 32768 # An "Abort" button defined with the RejectRole .
Retry = 65536 # A "Retry" button defined with the AcceptRole .
Ignore = 131072 # An "Ignore" button defined with the AcceptRole .
Avoid = 262144 # An "'Don't show again'" button defined with the HelpRole, returns a default AcceptButton .
def __init__(self,
icon,
title,
text,
detailed_text="",
buttons=Cancel | Ok,
parent=None):
QDialog.__init__(self, parent=parent)
self.setWindowFlags(self.windowFlags() & ~Qt.WindowTitleHint)
self.setWindowFlags(self.windowFlags()
& ~Qt.WindowContextHelpButtonHint
& ~Qt.WindowMinimizeButtonHint)
self.setObjectName('MessageBox')
self._use_checkbox = True
self.text = text
self.verticalLayout = QVBoxLayout(self)
self.verticalLayout.setObjectName("verticalLayout")
self.verticalLayout.setContentsMargins(1, 1, 1, 1)
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setObjectName("horizontalLayout")
self.horizontalLayout.setContentsMargins(1, 1, 1, 1)
# create icon
pixmap = None
if icon == self.NoIcon:
pass
elif icon == self.Question:
pixmap = QPixmap(
QImage(":icons/question.png").scaled(56, 56,
Qt.IgnoreAspectRatio,
Qt.SmoothTransformation))
elif icon == self.Information:
pixmap = QPixmap(
QImage(":icons/info.png").scaled(56, 56, Qt.IgnoreAspectRatio,
Qt.SmoothTransformation))
elif icon == self.Warning:
pixmap = QPixmap(
QImage(":icons/warning.png").scaled(56, 56,
Qt.IgnoreAspectRatio,
Qt.SmoothTransformation))
elif icon == self.Critical:
pixmap = QPixmap(
QImage(":icons/critical.png").scaled(56, 56,
Qt.IgnoreAspectRatio,
Qt.SmoothTransformation))
spacerItem = QSpacerItem(10, 60, QSizePolicy.Minimum,
QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem)
self.icon_label = QLabel()
if pixmap is not None:
self.icon_label.setPixmap(pixmap)
self.icon_label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.horizontalLayout.addWidget(self.icon_label)
spacerItem = QSpacerItem(10, 60, QSizePolicy.Minimum,
QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem)
# add message
self.message_label = QLabel(text)
self.message_label.setWordWrap(True)
self.message_label.setScaledContents(True)
self.message_label.setOpenExternalLinks(True)
self.horizontalLayout.addWidget(self.message_label)
self.verticalLayout.addLayout(self.horizontalLayout)
# create buttons
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.setObjectName("buttonBox")
self.buttonBox.setOrientation(Qt.Horizontal)
self._accept_button = None
self._reject_button = None
self._buttons = buttons
self._create_buttons(buttons)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.verticalLayout.addWidget(self.buttonBox)
if detailed_text:
self.btn_show_details = QPushButton(self.tr('Details...'))
self.btn_show_details.setCheckable(True)
self.btn_show_details.setChecked(True)
self.btn_show_details.toggled.connect(self.on_toggled_details)
self.buttonBox.addButton(self.btn_show_details,
QDialogButtonBox.ActionRole)
# create area for detailed text
self.textEdit = textEdit = QTextEdit(self)
textEdit.setObjectName("textEdit")
textEdit.setReadOnly(True)
textEdit.setText(detailed_text)
# textEdit.setVisible(False)
self.verticalLayout.addWidget(self.textEdit)
self.resize(480, self.verticalLayout.totalSizeHint().height())
buttons_in_box = self.buttonBox.buttons()
if buttons_in_box:
self.buttonBox.buttons()[0].setFocus()
def setAcceptButton(self, button):
'''
Sets the button with given ID to accept button if more then one button with AcceptRole was added to this dialog.
Adds the buttton to the box if is not already in.
:param int button: button id
'''
if not button & self._buttons:
self._create_buttons(button)
self._accept_button = button
def setRejectButton(self, button):
'''
Sets the button with given ID to reject button if more then one button with RejectRole was added to this dialog.
Adds the buttton to the box if is not already in.
:param int button: button id
'''
if not button & self._buttons:
self._create_buttons(button)
self._reject_button = button
def on_toggled_details(self, checked):
if checked:
self.verticalLayout.addWidget(self.textEdit)
else:
self.verticalLayout.removeWidget(self.textEdit)
self.textEdit.setVisible(checked)
if not self.isMaximized():
self.setMinimumSize(self.verticalLayout.totalMinimumSize())
self.resize(self._current_size.width(),
self.verticalLayout.totalSizeHint().height())
@staticmethod
def about(parent, title, text, detailed_text='', buttons=Close):
box = MessageBox(MessageBox.Information,
title,
text,
detailed_text=detailed_text,
buttons=buttons,
parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def information(parent, title, text, detailed_text='', buttons=Close):
box = MessageBox(MessageBox.Information,
title,
text,
detailed_text=detailed_text,
buttons=buttons,
parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def question(parent,
title,
text,
detailed_text='',
buttons=Yes | No | Cancel):
box = MessageBox(MessageBox.Question,
title,
text,
detailed_text=detailed_text,
buttons=buttons,
parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def warning(parent, title, text, detailed_text='', buttons=Ok):
box = MessageBox(MessageBox.Warning,
title,
text,
detailed_text=detailed_text,
buttons=buttons,
parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def critical(parent, title, text, detailed_text='', buttons=Ok):
box = MessageBox(MessageBox.Critical,
title,
text,
detailed_text=detailed_text,
buttons=buttons,
parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
def resizeEvent(self, event):
if not self.isMaximized():
self._current_size = event.size()
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%% close handling %%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
def exec_(self):
if self.text in IGNORED_ERRORS:
self.accept()
return self.result()
return QDialog.exec_(self)
def accept(self):
if self.result() == 0:
if self._accept_button is not None:
self.setResult(self._accept_button)
else:
self.setResult(1)
self.accepted.emit()
if self.isModal():
self.hide()
def reject(self):
if self.result() == 0:
if self._reject_button is not None:
self.setResult(self._reject_button)
self.rejected.emit()
self.hide()
def hideEvent(self, event):
# event.ignore()
self.close()
def closeEvent(self, event):
self.setAttribute(Qt.WA_DeleteOnClose, True)
QDialog.closeEvent(self, event)
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%% create buttons %%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
def _create_buttons(self, buttons):
if MessageBox.Ok & buttons:
self._accept_button = MessageBox.Ok
bt = QPushButton(self.tr("&ok"))
bt.clicked.connect(self._on_ok_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Open & buttons:
self._accept_button = MessageBox.Open
bt = QPushButton(self.tr("&Open"))
bt.clicked.connect(self._on_open_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Save & buttons:
self._accept_button = MessageBox.Save
bt = QPushButton(self.tr("&Save"))
bt.clicked.connect(self._on_save_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Cancel & buttons:
self._reject_button = MessageBox.Cancel
bt = QPushButton(self.tr("&Cancel"))
bt.clicked.connect(self._on_cancel_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.RejectRole)
if MessageBox.Close & buttons:
self._reject_button = MessageBox.Close
bt = QPushButton(self.tr("&Close"))
bt.clicked.connect(self._on_close_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.RejectRole)
if MessageBox.Discard & buttons:
bt = QPushButton(self.tr("&Discard"))
bt.clicked.connect(self._on_discard_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.DestructiveRole)
if MessageBox.Apply & buttons:
bt = QPushButton(self.tr("&Apply"))
bt.clicked.connect(self._on_apply_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.ApplyRole)
if MessageBox.Reset & buttons:
bt = QPushButton(self.tr("&Reset"))
bt.clicked.connect(self._on_reset_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.ResetRole)
if MessageBox.RestoreDefaults & buttons:
bt = QPushButton(self.tr("&RestoreDefaults"))
bt.clicked.connect(self._on_restore_defaults_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.ResetRole)
if MessageBox.Help & buttons:
bt = QPushButton(self.tr("&Help"))
bt.clicked.connect(self._on_help_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.HelpRole)
if MessageBox.SaveAll & buttons:
self._accept_button = MessageBox.SaveAll
bt = QPushButton(self.tr("&SaveAll"))
bt.clicked.connect(self._on_saveall_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Yes & buttons:
bt = QPushButton(self.tr("&Yes"))
bt.clicked.connect(self._on_yes_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.YesRole)
if MessageBox.YesToAll & buttons:
bt = QPushButton(self.tr("&YesToAll"))
bt.clicked.connect(self._on_yestoall_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.YesRole)
if MessageBox.No & buttons:
bt = QPushButton(self.tr("&No"))
bt.clicked.connect(self._on_no_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.NoRole)
if MessageBox.NoToAll & buttons:
bt = QPushButton(self.tr("&NoToAll"))
bt.clicked.connect(self._on_notoall_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.NoRole)
if MessageBox.Abort & buttons:
self._reject_button = MessageBox.Abort
bt = QPushButton(self.tr("&Abort"))
bt.clicked.connect(self._on_abort_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.RejectRole)
if MessageBox.Retry & buttons:
self._accept_button = MessageBox.Retry
bt = QPushButton(self.tr("&Retry"))
bt.clicked.connect(self._on_retry_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Ignore & buttons:
self._accept_button = MessageBox.Ignore
bt = QPushButton(self.tr("&Ignore"))
bt.clicked.connect(self._on_ignore_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Avoid & buttons:
if self._use_checkbox:
checkbox = QCheckBox("&Don't show again", self)
checkbox.stateChanged.connect(self._check_ignore)
self.buttonBox.addButton(checkbox, QDialogButtonBox.HelpRole)
else:
bt = QPushButton(self.tr("&Don't show again"))
bt.setMaximumHeight(24)
bt.clicked.connect(self._add_to_ignore)
self.buttonBox.addButton(bt, QDialogButtonBox.HelpRole)
def _on_ok_clicked(self):
self.done(MessageBox.Ok)
def _on_open_clicked(self):
self.done(MessageBox.Open)
def _on_save_clicked(self):
self.done(MessageBox.Save)
def _on_cancel_clicked(self):
self.done(MessageBox.Cancel)
def _on_close_clicked(self):
self.done(MessageBox.Close)
def _on_discard_clicked(self):
self.done(MessageBox.Discard)
def _on_apply_clicked(self):
self.done(MessageBox.Apply)
def _on_reset_clicked(self):
self.done(MessageBox.Reset)
def _on_restore_defaults_clicked(self):
self.done(MessageBox.RestoreDefaults)
def _on_help_clicked(self):
self.done(MessageBox.Help)
def _on_saveall_clicked(self):
self.done(MessageBox.SaveAll)
def _on_yes_clicked(self):
self.done(MessageBox.Yes)
def _on_yestoall_clicked(self):
self.done(MessageBox.YesToAll)
def _on_no_clicked(self):
self.done(MessageBox.No)
def _on_notoall_clicked(self):
self.done(MessageBox.NoToAll)
def _on_abort_clicked(self):
self.done(MessageBox.Abort)
def _on_retry_clicked(self):
self.done(MessageBox.Retry)
def _on_ignore_clicked(self):
self.done(MessageBox.Ignore)
def _add_to_ignore(self):
IGNORED_ERRORS.append(self.text)
self.accept()
def _check_ignore(self, state):
if state:
IGNORED_ERRORS.append(self.text)
else:
try:
IGNORED_ERRORS.remove(self.text)
except Exception:
pass
class CapabilityControlWidget(QFrame):
'''
The control widget contains buttons for control a capability. Currently this
are C{On} and C{Off} buttons. Additionally, the state of the capability is
color coded.
'''
start_nodes_signal = Signal(str, str, list)
'''@ivar: the signal is emitted to start on host(described by masteruri) the nodes described in the list, Parameter(masteruri, config, nodes).'''
stop_nodes_signal = Signal(str, list)
'''@ivar: the signal is emitted to stop on masteruri the nodes described in the list.'''
def __init__(self, masteruri, cfg, ns, nodes, parent=None):
QFrame.__init__(self, parent)
self._masteruri = masteruri
self._nodes = {cfg: {ns: nodes}}
frame_layout = QVBoxLayout(self)
frame_layout.setContentsMargins(0, 0, 0, 0)
# create frame for warning label
self.warning_frame = warning_frame = QFrame(self)
warning_layout = QHBoxLayout(warning_frame)
warning_layout.setContentsMargins(0, 0, 0, 0)
warning_layout.addItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
self.warning_label = QLabel()
icon = nm.settings().icon('crystal_clear_warning.png')
self.warning_label.setPixmap(icon.pixmap(QSize(40, 40)))
self.warning_label.setToolTip(
'Multiple configuration for same node found!\nA first one will be selected for the start a node!'
)
warning_layout.addWidget(self.warning_label)
warning_layout.addItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
frame_layout.addItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
frame_layout.addWidget(warning_frame)
# create frame for start/stop buttons
buttons_frame = QFrame()
buttons_layout = QHBoxLayout(buttons_frame)
buttons_layout.setContentsMargins(0, 0, 0, 0)
buttons_layout.addItem(QSpacerItem(20, 20))
self.on_button = QPushButton()
self.on_button.setFlat(False)
self.on_button.setText("On")
self.on_button.clicked.connect(self.on_on_clicked)
buttons_layout.addWidget(self.on_button)
self.off_button = QPushButton()
self.off_button.setFlat(True)
self.off_button.setText("Off")
self.off_button.clicked.connect(self.on_off_clicked)
buttons_layout.addWidget(self.off_button)
buttons_layout.addItem(QSpacerItem(20, 20))
frame_layout.addWidget(buttons_frame)
frame_layout.addItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
self.warning_frame.setVisible(False)
def hasConfigs(self):
'''
:return: True, if a configurations for this widget are available.
:rtype: bool
'''
return len(self._nodes) > 0
def nodes(self, cfg=''):
'''
:return: the list with nodes required by this capability. The nodes are
defined by ROS full name.
:rtype: [str]
'''
try:
if cfg:
return [n for l in self._nodes[cfg].values() for n in l]
else:
return [
n for c in self._nodes.values() for l in c.values()
for n in l
]
except Exception:
return []
def setNodeState(self, running_nodes, stopped_nodes, error_nodes):
'''
Sets the state of this capability.
:param running_nodes: a list with running nodes.
:type running_nodes: [str]
:param stopped_nodes: a list with not running nodes.
:type stopped_nodes: [str]
:param error_nodes: a list with nodes having a problem.
:type error_nodes: [str]
'''
self.setAutoFillBackground(True)
self.setBackgroundRole(QPalette.Base)
palette = QPalette()
if error_nodes:
brush = QBrush(QColor(255, 100, 0))
elif running_nodes and stopped_nodes:
brush = QBrush(QColor(140, 185, 255)) # 30, 50, 255
elif running_nodes:
self.on_button.setFlat(True)
self.off_button.setFlat(False)
brush = QBrush(QColor(59, 223, 18)) # 59, 223, 18
else:
brush = QBrush(QColor(255, 255, 255))
self.on_button.setFlat(False)
self.off_button.setFlat(True)
palette.setBrush(QPalette.Active, QPalette.Base, brush)
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Inactive, QPalette.Base, brush)
self.setPalette(palette)
def removeCfg(self, cfg):
try:
del self._nodes[cfg]
except Exception:
pass
def updateNodes(self, cfg, ns, nodes):
self._nodes[cfg] = {ns: nodes}
test_nodes = self.nodes()
self.warning_frame.setVisible(len(test_nodes) != len(set(test_nodes)))
def on_on_clicked(self):
started = set() # do not start nodes multiple times
for c in self._nodes.keys():
node2start = set(self.nodes(c)) - started
self.start_nodes_signal.emit(self._masteruri, c, list(node2start))
started.update(node2start)
self.on_button.setFlat(True)
self.off_button.setFlat(False)
def on_off_clicked(self):
self.stop_nodes_signal.emit(self._masteruri, self.nodes())
self.on_button.setFlat(False)
self.off_button.setFlat(True)
class SelectDialog(QDialog):
'''
This dialog creates an input mask for a string list and return selected entries.
'''
def __init__(self, items=list(), buttons=QDialogButtonBox.Cancel | QDialogButtonBox.Ok, exclusive=False,
preselect_all=False, title='', description='', icon='', parent=None, select_if_single=True,
checkitem1='', checkitem2=''):
'''
Creates an input dialog.
@param items: a list with strings
@type items: C{list()}
'''
QDialog.__init__(self, parent=parent)
self.setObjectName(' - '.join(['SelectDialog', str(items)]))
self.verticalLayout = QVBoxLayout(self)
self.verticalLayout.setObjectName("verticalLayout")
self.verticalLayout.setContentsMargins(1, 1, 1, 1)
# add filter row
self.filter_frame = QFrame(self)
filterLayout = QHBoxLayout(self.filter_frame)
filterLayout.setContentsMargins(1, 1, 1, 1)
label = QLabel("Filter:", self.filter_frame)
self.filter_field = EnchancedLineEdit(self.filter_frame)
filterLayout.addWidget(label)
filterLayout.addWidget(self.filter_field)
self.filter_field.textChanged.connect(self._on_filter_changed)
self.verticalLayout.addWidget(self.filter_frame)
if description:
self.description_frame = QFrame(self)
descriptionLayout = QHBoxLayout(self.description_frame)
# descriptionLayout.setContentsMargins(1, 1, 1, 1)
if icon:
self.icon_label = QLabel(self.description_frame)
self.icon_label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.icon_label.setPixmap(QPixmap(icon).scaled(30, 30, Qt.KeepAspectRatio))
descriptionLayout.addWidget(self.icon_label)
self.description_label = QLabel(self.description_frame)
self.description_label.setWordWrap(True)
self.description_label.setText(description)
descriptionLayout.addWidget(self.description_label)
self.verticalLayout.addWidget(self.description_frame)
# create area for the parameter
self.content = MainBox(self)
if items:
self.scroll_area = QScrollArea(self)
self.scroll_area.setFocusPolicy(Qt.NoFocus)
self.scroll_area.setObjectName("scroll_area")
self.scroll_area.setWidgetResizable(True)
self.scroll_area.setWidget(self.content)
self.verticalLayout.addWidget(self.scroll_area)
self.checkitem1 = checkitem1
self.checkitem1_result = False
self.checkitem2 = checkitem2
self.checkitem2_result = False
# add select all option
if not exclusive and items:
self._ignore_next_toggle = False
self.select_all_checkbox = QCheckBox('all entries')
self.select_all_checkbox.setTristate(True)
self.select_all_checkbox.stateChanged.connect(self._on_select_all_checkbox_stateChanged)
self.verticalLayout.addWidget(self.select_all_checkbox)
self.content.toggled.connect(self._on_main_toggle)
if self.checkitem1:
self.checkitem1_checkbox = QCheckBox(self.checkitem1)
self.checkitem1_checkbox.stateChanged.connect(self._on_select_checkitem1_checkbox_stateChanged)
self.verticalLayout.addWidget(self.checkitem1_checkbox)
if self.checkitem2:
self.checkitem2_checkbox = QCheckBox(self.checkitem2)
self.checkitem2_checkbox.stateChanged.connect(self._on_select_checkitem2_checkbox_stateChanged)
self.verticalLayout.addWidget(self.checkitem2_checkbox)
if not items:
spacerItem = QSpacerItem(1, 1, QSizePolicy.Expanding, QSizePolicy.Expanding)
self.verticalLayout.addItem(spacerItem)
# create buttons
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.setObjectName("buttonBox")
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(buttons)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.verticalLayout.addWidget(self.buttonBox)
# set the input fields
if items:
self.content.createFieldsFromValues(items, exclusive)
if (select_if_single and len(items) == 1) or preselect_all:
self.select_all_checkbox.setCheckState(Qt.Checked)
if not items or len(items) < 7:
self.filter_frame.setVisible(False)
# print '=============== create', self.objectName()
#
# def __del__(self):
# print "************ destroy", self.objectName()
def _on_main_toggle(self, state):
self._ignore_next_toggle = state != self.select_all_checkbox.checkState()
self.select_all_checkbox.setCheckState(state)
def _on_select_all_checkbox_stateChanged(self, state):
if not self._ignore_next_toggle:
self.content.setState(state)
self._ignore_next_toggle = False
def _on_select_checkitem1_checkbox_stateChanged(self, state):
if state == Qt.Checked:
self.checkitem1_result = True
elif state == Qt.Unchecked:
self.checkitem1_result = False
def _on_select_checkitem2_checkbox_stateChanged(self, state):
if state == Qt.Checked:
self.checkitem2_result = True
elif state == Qt.Unchecked:
self.checkitem2_result = False
def _on_filter_changed(self):
self.content.filter(self.filter_field.text())
def getSelected(self):
return self.content.getSelected()
@staticmethod
def getValue(title, description='', items=list(), exclusive=False, preselect_all=False, icon='', parent=None, select_if_single=True, checkitem1='', checkitem2=''):
selectDia = SelectDialog(items, exclusive=exclusive, preselect_all=preselect_all, description=description, icon=icon, parent=parent, select_if_single=select_if_single, checkitem1=checkitem1, checkitem2=checkitem2)
selectDia.setWindowTitle(title)
selectDia.resize(480, 256)
if selectDia.exec_():
if selectDia.checkitem2:
return selectDia.getSelected(), True, selectDia.checkitem1_result, selectDia.checkitem2_result
if selectDia.checkitem1:
return selectDia.getSelected(), True, selectDia.checkitem1_result
return selectDia.getSelected(), True
if selectDia.checkitem2:
return list(), False, False, False
if selectDia.checkitem1:
return list(), False, False
return list(), False
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%% close handling %%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
def accept(self):
self.setResult(QDialog.Accepted)
self.hide()
def reject(self):
self.setResult(QDialog.Rejected)
self.hide()
def hideEvent(self, event):
self.close()
def closeEvent(self, event):
'''
Test the open files for changes and save this if needed.
'''
self.setAttribute(Qt.WA_DeleteOnClose, True)
QDialog.closeEvent(self, event)
class CapabilityControlWidget(QFrame):
'''
The control widget contains buttons for control a capability. Currently this
are C{On} and C{Off} buttons. Additionally, the state of the capability is
color coded.
'''
start_nodes_signal = Signal(str, str, list)
'''@ivar: the signal is emitted to start on host(described by masteruri) the nodes described in the list, Parameter(masteruri, config, nodes).'''
stop_nodes_signal = Signal(str, list)
'''@ivar: the signal is emitted to stop on masteruri the nodes described in the list.'''
def __init__(self, masteruri, cfg, ns, nodes, parent=None):
QFrame.__init__(self, parent)
self._masteruri = masteruri
self._nodes = {cfg: {ns: nodes}}
frame_layout = QVBoxLayout(self)
frame_layout.setContentsMargins(0, 0, 0, 0)
# create frame for warning label
self.warning_frame = warning_frame = QFrame(self)
warning_layout = QHBoxLayout(warning_frame)
warning_layout.setContentsMargins(0, 0, 0, 0)
warning_layout.addItem(QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
self.warning_label = QLabel()
icon = QIcon(':/icons/crystal_clear_warning.png')
self.warning_label.setPixmap(icon.pixmap(QSize(40, 40)))
self.warning_label.setToolTip('Multiple configuration for same node found!\nA first one will be selected for the start a node!')
warning_layout.addWidget(self.warning_label)
warning_layout.addItem(QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
frame_layout.addItem(QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
frame_layout.addWidget(warning_frame)
# create frame for start/stop buttons
buttons_frame = QFrame()
buttons_layout = QHBoxLayout(buttons_frame)
buttons_layout.setContentsMargins(0, 0, 0, 0)
buttons_layout.addItem(QSpacerItem(20, 20))
self.on_button = QPushButton()
self.on_button.setFlat(False)
self.on_button.setText("On")
self.on_button.clicked.connect(self.on_on_clicked)
buttons_layout.addWidget(self.on_button)
self.off_button = QPushButton()
self.off_button.setFlat(True)
self.off_button.setText("Off")
self.off_button.clicked.connect(self.on_off_clicked)
buttons_layout.addWidget(self.off_button)
buttons_layout.addItem(QSpacerItem(20, 20))
frame_layout.addWidget(buttons_frame)
frame_layout.addItem(QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Expanding))
self.warning_frame.setVisible(False)
def hasConfigs(self):
'''
@return: True, if a configurations for this widget are available.
@rtype: bool
'''
return len(self._nodes) > 0
def nodes(self, cfg=''):
'''
@return: the list with nodes required by this capability. The nodes are
defined by ROS full name.
@rtype: C{[str]}
'''
try:
if cfg:
return [n for l in self._nodes[cfg].itervalues() for n in l]
else:
return [n for c in self._nodes.itervalues() for l in c.itervalues() for n in l]
except:
return []
def setNodeState(self, running_nodes, stopped_nodes, error_nodes):
'''
Sets the state of this capability.
@param running_nodes: a list with running nodes.
@type running_nodes: C{[str]}
@param stopped_nodes: a list with not running nodes.
@type stopped_nodes: C{[str]}
@param error_nodes: a list with nodes having a problem.
@type error_nodes: C{[str]}
'''
self.setAutoFillBackground(True)
self.setBackgroundRole(QPalette.Base)
palette = QPalette()
if error_nodes:
brush = QBrush(QColor(255, 100, 0))
elif running_nodes and stopped_nodes:
brush = QBrush(QColor(140, 185, 255)) # 30, 50, 255
elif running_nodes:
self.on_button.setFlat(True)
self.off_button.setFlat(False)
brush = QBrush(QColor(59, 223, 18)) # 59, 223, 18
else:
brush = QBrush(QColor(255, 255, 255))
self.on_button.setFlat(False)
self.off_button.setFlat(True)
palette.setBrush(QPalette.Active, QPalette.Base, brush)
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Inactive, QPalette.Base, brush)
self.setPalette(palette)
def removeCfg(self, cfg):
try:
del self._nodes[cfg]
except:
pass
def updateNodes(self, cfg, ns, nodes):
self._nodes[cfg] = {ns: nodes}
test_nodes = self.nodes()
self.warning_frame.setVisible(len(test_nodes) != len(set(test_nodes)))
def on_on_clicked(self):
started = set() # do not start nodes multiple times
for c in self._nodes.iterkeys():
node2start = set(self.nodes(c)) - started
self.start_nodes_signal.emit(self._masteruri, c, list(node2start))
started.update(node2start)
self.on_button.setFlat(True)
self.off_button.setFlat(False)
def on_off_clicked(self):
self.stop_nodes_signal.emit(self._masteruri, self.nodes())
self.on_button.setFlat(False)
self.off_button.setFlat(True)
class MessageBox(QDialog):
NoIcon = 0
Information = 1
Warning = 2
Critical = 3
Question = 4
NoButton = 0
Ok = 1 # An "OK" button defined with the AcceptRole .
Open = 2 # A "Open" button defined with the AcceptRole .
Save = 4 # A "Save" button defined with the AcceptRole .
Cancel = 8 # A "Cancel" button defined with the RejectRole .
Close = 16 # A "Close" button defined with the RejectRole .
Discard = 32 # A "Discard" or "Don't Save" button, depending on the platform, defined with the DestructiveRole .
Apply = 64 # An "Apply" button defined with the ApplyRole .
Reset = 128 # A "Reset" button defined with the ResetRole .
RestoreDefaults = 256 # A "Restore Defaults" button defined with the ResetRole .
Help = 512 # A "Help" button defined with the HelpRole .
SaveAll = 1024 # A "Save All" button defined with the AcceptRole .
Yes = 2048 # A "Yes" button defined with the YesRole .
YesToAll = 4096 # A "Yes to All" button defined with the YesRole .
No = 8192 # A "No" button defined with the NoRole .
NoToAll = 16384 # A "No to All" button defined with the NoRole .
Abort = 32768 # An "Abort" button defined with the RejectRole .
Retry = 65536 # A "Retry" button defined with the AcceptRole .
Ignore = 131072 # An "Ignore" button defined with the AcceptRole .
Avoid = 262144 # An "'Don't show again'" button defined with the HelpRole, returns a default AcceptButton .
def __init__(self, icon, title, text, detailed_text="", buttons=Cancel | Ok, parent=None):
QDialog.__init__(self, parent=parent)
self.setWindowFlags(self.windowFlags() & ~Qt.WindowTitleHint)
self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint & ~Qt.WindowMinimizeButtonHint)
self.setObjectName('MessageBox')
self._use_checkbox = True
self.text = text
self.verticalLayout = QVBoxLayout(self)
self.verticalLayout.setObjectName("verticalLayout")
self.verticalLayout.setContentsMargins(1, 1, 1, 1)
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setObjectName("horizontalLayout")
self.horizontalLayout.setContentsMargins(1, 1, 1, 1)
# create icon
pixmap = None
if icon == self.NoIcon:
pass
elif icon == self.Question:
pixmap = QApplication.style().standardPixmap(QStyle.SP_MessageBoxQuestion)
elif icon == self.Information:
pixmap = QApplication.style().standardPixmap(QStyle.SP_MessageBoxInformation)
elif icon == self.Warning:
pixmap = QPixmap(":icons/crystal_clear_warning_56.png")
elif icon == self.Critical:
pixmap = QApplication.style().standardPixmap(QStyle.SP_MessageBoxCritical)
spacerItem = QSpacerItem(10, 60, QSizePolicy.Minimum, QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem)
self.icon_label = QLabel()
if pixmap is not None:
self.icon_label.setPixmap(pixmap)
self.icon_label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.horizontalLayout.addWidget(self.icon_label)
spacerItem = QSpacerItem(10, 60, QSizePolicy.Minimum, QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem)
# add message
self.message_label = QLabel(text)
self.message_label.setWordWrap(True)
self.message_label.setScaledContents(True)
self.message_label.setOpenExternalLinks(True)
self.horizontalLayout.addWidget(self.message_label)
self.verticalLayout.addLayout(self.horizontalLayout)
# create buttons
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.setObjectName("buttonBox")
self.buttonBox.setOrientation(Qt.Horizontal)
self._accept_button = None
self._reject_button = None
self._buttons = buttons
self._create_buttons(buttons)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.verticalLayout.addWidget(self.buttonBox)
if detailed_text:
self.btn_show_details = QPushButton(self.tr('Details...'))
self.btn_show_details.setCheckable(True)
self.btn_show_details.setChecked(True)
self.btn_show_details.toggled.connect(self.on_toggled_details)
self.buttonBox.addButton(self.btn_show_details, QDialogButtonBox.ActionRole)
# create area for detailed text
self.textEdit = textEdit = QTextEdit(self)
textEdit.setObjectName("textEdit")
textEdit.setReadOnly(True)
textEdit.setText(detailed_text)
# textEdit.setVisible(False)
self.verticalLayout.addWidget(self.textEdit)
self.resize(480, self.verticalLayout.totalSizeHint().height())
buttons_in_box = self.buttonBox.buttons()
if buttons_in_box:
self.buttonBox.buttons()[0].setFocus()
def setAcceptButton(self, button):
'''
Sets the button with given ID to accept button if more then one button with AcceptRole was added to this dialog.
Adds the buttton to the box if is not already in.
:param button: int
'''
if not button & self._buttons:
self._create_buttons(button)
self._accept_button = button
def setRejectButton(self, button):
'''
Sets the button with given ID to reject button if more then one button with RejectRole was added to this dialog.
Adds the buttton to the box if is not already in.
:param button: int
'''
if not button & self._buttons:
self._create_buttons(button)
self._reject_button = button
def on_toggled_details(self, checked):
if checked:
self.verticalLayout.addWidget(self.textEdit)
else:
self.verticalLayout.removeWidget(self.textEdit)
self.textEdit.setVisible(checked)
if not self.isMaximized():
self.setMinimumSize(self.verticalLayout.totalMinimumSize())
self.resize(self._current_size.width(), self.verticalLayout.totalSizeHint().height())
@staticmethod
def about(parent, title, text, detailed_text='', buttons=Close):
box = MessageBox(MessageBox.Information, title, text, detailed_text=detailed_text, buttons=buttons, parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def information(parent, title, text, detailed_text='', buttons=Close):
box = MessageBox(MessageBox.Information, title, text, detailed_text=detailed_text, buttons=buttons, parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def question(parent, title, text, detailed_text='', buttons=Yes | No | Cancel):
box = MessageBox(MessageBox.Question, title, text, detailed_text=detailed_text, buttons=buttons, parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def warning(parent, title, text, detailed_text='', buttons=Ok):
box = MessageBox(MessageBox.Warning, title, text, detailed_text=detailed_text, buttons=buttons, parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
@staticmethod
def critical(parent, title, text, detailed_text='', buttons=Ok):
box = MessageBox(MessageBox.Critical, title, text, detailed_text=detailed_text, buttons=buttons, parent=parent)
if MessageBox.Yes & buttons:
box.setAcceptButton(MessageBox.Yes)
if MessageBox.Cancel & buttons:
box.setRejectButton(MessageBox.Cancel)
elif MessageBox.No & buttons:
box.setRejectButton(MessageBox.No)
return box.exec_()
def resizeEvent(self, event):
if not self.isMaximized():
self._current_size = event.size()
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%% close handling %%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
def exec_(self):
if self.text in IGNORED_ERRORS:
self.accept()
return self.result()
return QDialog.exec_(self)
def accept(self):
if self.result() == 0:
if self._accept_button is not None:
self.setResult(self._accept_button)
else:
self.setResult(1)
self.accepted.emit()
if self.isModal():
self.hide()
def reject(self):
if self.result() == 0:
if self._reject_button is not None:
self.setResult(self._reject_button)
self.rejected.emit()
self.hide()
def hideEvent(self, event):
# event.ignore()
self.close()
def closeEvent(self, event):
self.setAttribute(Qt.WA_DeleteOnClose, True)
QDialog.closeEvent(self, event)
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%% create buttons %%%%%%%%%%%%%%%%%%%%%
# %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
def _create_buttons(self, buttons):
if MessageBox.Ok & buttons:
self._accept_button = MessageBox.Ok
bt = QPushButton(self.tr("&ok"))
bt.clicked.connect(self._on_ok_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Open & buttons:
self._accept_button = MessageBox.Open
bt = QPushButton(self.tr("&Open"))
bt.clicked.connect(self._on_open_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Save & buttons:
self._accept_button = MessageBox.Save
bt = QPushButton(self.tr("&Save"))
bt.clicked.connect(self._on_save_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Cancel & buttons:
self._reject_button = MessageBox.Cancel
bt = QPushButton(self.tr("&Cancel"))
bt.clicked.connect(self._on_cancel_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.RejectRole)
if MessageBox.Close & buttons:
self._reject_button = MessageBox.Close
bt = QPushButton(self.tr("&Close"))
bt.clicked.connect(self._on_close_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.RejectRole)
if MessageBox.Discard & buttons:
bt = QPushButton(self.tr("&Discard"))
bt.clicked.connect(self._on_discard_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.DestructiveRole)
if MessageBox.Apply & buttons:
bt = QPushButton(self.tr("&Apply"))
bt.clicked.connect(self._on_apply_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.ApplyRole)
if MessageBox.Reset & buttons:
bt = QPushButton(self.tr("&Reset"))
bt.clicked.connect(self._on_reset_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.ResetRole)
if MessageBox.RestoreDefaults & buttons:
bt = QPushButton(self.tr("&RestoreDefaults"))
bt.clicked.connect(self._on_restore_defaults_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.ResetRole)
if MessageBox.Help & buttons:
bt = QPushButton(self.tr("&Help"))
bt.clicked.connect(self._on_help_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.HelpRole)
if MessageBox.SaveAll & buttons:
self._accept_button = MessageBox.SaveAll
bt = QPushButton(self.tr("&SaveAll"))
bt.clicked.connect(self._on_saveall_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Yes & buttons:
bt = QPushButton(self.tr("&Yes"))
bt.clicked.connect(self._on_yes_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.YesRole)
if MessageBox.YesToAll & buttons:
bt = QPushButton(self.tr("&YesToAll"))
bt.clicked.connect(self._on_yestoall_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.YesRole)
if MessageBox.No & buttons:
bt = QPushButton(self.tr("&No"))
bt.clicked.connect(self._on_no_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.NoRole)
if MessageBox.NoToAll & buttons:
bt = QPushButton(self.tr("&NoToAll"))
bt.clicked.connect(self._on_notoall_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.NoRole)
if MessageBox.Abort & buttons:
self._reject_button = MessageBox.Abort
bt = QPushButton(self.tr("&Abort"))
bt.clicked.connect(self._on_abort_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.RejectRole)
if MessageBox.Retry & buttons:
self._accept_button = MessageBox.Retry
bt = QPushButton(self.tr("&Retry"))
bt.clicked.connect(self._on_retry_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Ignore & buttons:
self._accept_button = MessageBox.Ignore
bt = QPushButton(self.tr("&Ignore"))
bt.clicked.connect(self._on_ignore_clicked)
self.buttonBox.addButton(bt, QDialogButtonBox.AcceptRole)
if MessageBox.Avoid & buttons:
if self._use_checkbox:
checkbox = QCheckBox("&Don't show again", self)
checkbox.stateChanged.connect(self._check_ignore)
self.buttonBox.addButton(checkbox, QDialogButtonBox.HelpRole)
else:
bt = QPushButton(self.tr("&Don't show again"))
bt.setMaximumHeight(24)
bt.clicked.connect(self._add_to_ignore)
self.buttonBox.addButton(bt, QDialogButtonBox.HelpRole)
def _on_ok_clicked(self):
self.done(MessageBox.Ok)
def _on_open_clicked(self):
self.done(MessageBox.Open)
def _on_save_clicked(self):
self.done(MessageBox.Save)
def _on_cancel_clicked(self):
self.done(MessageBox.Cancel)
def _on_close_clicked(self):
self.done(MessageBox.Close)
def _on_discard_clicked(self):
self.done(MessageBox.Discard)
def _on_apply_clicked(self):
self.done(MessageBox.Apply)
def _on_reset_clicked(self):
self.done(MessageBox.Reset)
def _on_restore_defaults_clicked(self):
self.done(MessageBox.RestoreDefaults)
def _on_help_clicked(self):
self.done(MessageBox.Help)
def _on_saveall_clicked(self):
self.done(MessageBox.SaveAll)
def _on_yes_clicked(self):
self.done(MessageBox.Yes)
def _on_yestoall_clicked(self):
self.done(MessageBox.YesToAll)
def _on_no_clicked(self):
self.done(MessageBox.No)
def _on_notoall_clicked(self):
self.done(MessageBox.NoToAll)
def _on_abort_clicked(self):
self.done(MessageBox.Abort)
def _on_retry_clicked(self):
self.done(MessageBox.Retry)
def _on_ignore_clicked(self):
self.done(MessageBox.Ignore)
def _add_to_ignore(self):
IGNORED_ERRORS.append(self.text)
self.accept()
def _check_ignore(self, state):
if state:
IGNORED_ERRORS.append(self.text)
else:
try:
IGNORED_ERRORS.remove(self.text)
except Exception:
pass
