Source code for taurus.qt.qtgui.button.taurusbutton
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#############################################################################
##
# This file is part of Taurus
##
# http://taurus-scada.org
##
# Copyright 2011 CELLS / ALBA Synchrotron, Bellaterra, Spain
##
# Taurus is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
##
# Taurus is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
##
# You should have received a copy of the GNU Lesser General Public License
# along with Taurus. If not, see <http://www.gnu.org/licenses/>.
##
#############################################################################
"""This module provides a taurus QPushButton based widgets"""
__all__ = ["TaurusLauncherButton", "TaurusCommandButton", "TaurusLockButton"]
__docformat__ = 'restructuredtext'
import PyTango
from taurus.external.qt import Qt
from taurus.core.taurusbasetypes import LockStatus, TaurusLockInfo
from taurus.core.taurusdevice import TaurusDevice
from taurus.qt.qtgui.base import TaurusBaseWidget
from taurus.core.util import eventfilters
from taurus.qt.qtgui.dialog import ProtectTaurusMessageBox
class _ButtonDialog(Qt.QDialog):
_widget = None
deleteWidgetOnClose = False
def __init__(self, parent=None):
Qt.QDialog.__init__(self, parent, Qt.Qt.WindowTitleHint)
l = Qt.QVBoxLayout()
self.setLayout(l)
self.previousWidgetConfig = None
def setWidget(self, widget):
oldWidget = self.widget()
if oldWidget is not None:
try:
self._widget.setModel(None)
except:
pass
oldWidget.hide()
oldWidget.setParent(None)
oldWidget.deleteLater()
if widget is not None:
self.layout().addWidget(widget)
self._widget = widget
def widget(self):
return self._widget
def closeEvent(self, event):
if self.deleteWidgetOnClose:
try:
self.previousWidgetConfig = self.widget().createConfig()
except:
self.previousWidgetConfig = None
self.setWidget(None)
Qt.QDialog.closeEvent(self, event)
[docs]class TaurusLauncherButton(Qt.QPushButton, TaurusBaseWidget):
'''This class provides a button that launches a modeless dialog containing
a specified Taurus widget which gets the same model as the button.
The button does not use the model directly. Instead it passes it to the
associated widget.
Code examples::
# a button that launches a TaurusAttrForm when clicked
button = TaurusLauncherButton(widget = TaurusAttrForm())
button.setModel('a/b/c') #a device name, which will be set at the TaurusAttrForm when clicking
# a button that launches a taurusLabel (whose model is an attribute: 'a/b/c/attrname')
button = TaurusLauncherButton(widget = TaurusLabel())
button.setModel('a/b/c/attrname') # attr name, which will be set at the TaurusLabel when clicking
#same as the previous one, but using the parent model and putting a custom text and icon:
button = TaurusLauncherButton(widget = TaurusLabel(), text='click me', icon='logos:taurus.png')
button.setUseParentModel(True) #let's assume that the button's parent has a model of type "/a/b/c"
button.setModel('/attrname')
'''
_widgetClassName = ''
_args = []
_kwargs = {}
_deleteWidgetOnClose = True
_icon = None
_text = None
def __init__(self, parent=None, designMode=False, widget=None, icon=None, text=None):
'''Constructor
:param parent: (Qt.QWidget or None) parent of this widget
:param designMode: (bool) flag for Qt designer
:param widget: (Qt.QWidget) a QWidget that will be shown when clicking
in the button
:param icon: (Qt.QIcon) icon for the button
:param text: (str) the button text (if None passed, the widget's class name
is used) '''
name = self.__class__.__name__
self.call__init__wo_kw(Qt.QPushButton, parent)
self.call__init__(TaurusBaseWidget, name, designMode=designMode)
self._dialog = _ButtonDialog(self)
if icon is None and self._icon is not None:
icon = Qt.QIcon(self._icon)
if icon is not None:
self.setIcon(Qt.QIcon(icon))
if text is not None:
self._text = text
if isinstance(widget, Qt.QWidget):
# we cannot be sure on recreating the same widget again
self._deleteWidgetOnClose = False
self.setWidget(widget)
elif widget is not None:
self._widgetClassName = widget
self.clicked.connect(self.onClicked)
self.setDefault(False)
self.setAutoDefault(False)
# no need to listen to change events!
self.insertEventFilter(eventfilters.IGNORE_CHANGE_AND_PERIODIC)
self._updateText()
[docs] def getModelClass(self):
'''see :meth:`TaurusBaseComponent.getModelClass`. Note that in the case of
:class:`TaurusLauncherButton`, the class is completely dependent on the
widget's class'''
try:
return self.widget().getModelClass()
except:
# return None #@TODO: Uncommenting this avoids the exception when
# TaurusBaseWidget.getModelClass chokes with relative classes. But
# the thing should be solved at TaurusBaseWidget.getModelClass
# level
return TaurusBaseWidget.getModelClass(self)
[docs] def setText(self, text):
'''Sets the text of the button. see :meth:`Qt.QPushButton.setText`'''
self._text = text
Qt.QPushButton.setText(self, text)
[docs] def setWidgetClassName(self, className, args=None, kwargs=None):
self._widgetClassName = str(className)
if args is not None:
self._args = args
if kwargs is not None:
self._kwargs = kwargs
self._updateText()
[docs] def resetWidgetClassName(self, className, args=None, kwargs=None):
self.setWidgetClassName(self.__class__._widgetClassName)
[docs] def createWidget(self):
from taurus.qt.qtgui.util import TaurusWidgetFactory
klass = TaurusWidgetFactory().getWidgetClass(self._widgetClassName)
widget = klass(*self._args, **self._kwargs)
self.setWidget(widget)
if self._dialog.previousWidgetConfig is not None:
try:
widget.applyConfig(self._dialog.previousWidgetConfig)
except Exception, e:
self.warning(
'Cannot apply previous configuration to widget. Reason: %s', repr(e))
[docs] def setWidget(self, widget):
'''sets the widget that will be shown when clicking the button
:param widget: (Qt.QWidget)
'''
self._dialog.setWidget(widget)
self._updateText()
[docs] def displayValue(self, v):
'''see :meth:`TaurusBaseComponent.displayValue`'''
if self._text is not None:
return self._text # make sure the text is not changed once set
TaurusBaseWidget.displayValue(self, v)
[docs] def getDisplayValue(self):
'''see :meth:`TaurusBaseComponent.getDisplayValue`'''
if self._text is not None:
return self._text
if self.widget() is not None:
return self.widget().__class__.__name__
return self._widgetClassName or '---'
def _updateText(self):
Qt.QPushButton.setText(self, self.getDisplayValue())
[docs] def onClicked(self):
'''
Slot called when the button is clicked.
Note that the dialog will only be created once. Subsequent clicks on
the button will only raise the existing dialog'''
if self.widget() is None:
self.createWidget()
self.widget().setModel(self.getModelName())
self._dialog.deleteWidgetOnClose = self._deleteWidgetOnClose
# dialog.resize(400,300)
# It's always necessary to set the title since the model could have
# changed
self._dialog.setWindowTitle(str(self.getModelName()))
self._dialog.show()
self._dialog.raise_()
@classmethod
[docs] def getQtDesignerPluginInfo(cls):
return {
'group': 'Taurus Buttons',
'icon': 'designer:pushbutton.png',
'module': 'taurus.qt.qtgui.button',
'container': False}
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
# Qt Properties #
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
Model = Qt.pyqtProperty("QString", TaurusBaseWidget.getModel,
TaurusBaseWidget.setModel, TaurusBaseWidget.resetModel)
UseParentModel = Qt.pyqtProperty("bool", TaurusBaseWidget.getUseParentModel,
TaurusBaseWidget.setUseParentModel, TaurusBaseWidget.resetUseParentModel)
widgetClassName = Qt.pyqtProperty(
"QString", getWidgetClassName, setWidgetClassName, resetWidgetClassName)
[docs]class TaurusCommandButton(Qt.QPushButton, TaurusBaseWidget):
'''This class provides a button that executes a tango command on its device.
Code examples::
# a button that executes the "status" command for the 'a/b/c' device server
button = TaurusCommandButton(command = 'Status', icon='logos:taurus.png')
button.setModel('a/b/c')
# a button that executes the "exec" command for the 'a/b/c' device server with one parameter
button = TaurusCommandButton(command = 'Status', parameters=['2+2'],icon='logos:taurus.png')
button.setModel('a/b/c')
.. seealso:: :class:`TaurusCommandsForm` provides a good example of use of
TaurusCommandButton (including managing the return value) '''
commandExecuted = Qt.pyqtSignal(object)
def __init__(self, parent=None, designMode=False, command=None,
parameters=None, icon=None, text=None,
timeout=None):
'''Constructor
:param parent: (Qt.QWidget or None) parent of this widget
:param designMode: (bool) flag for Qt designer
:param command: (str) the name of the command to be executed
:param parameters: (sequence<str>) the list of parameteres. Default value is None meaning no parameters
:param icon: (Qt.QIcon) icon for the button
:param text: (str) the button text (if None passed, `command` is used)
:param timeout: (float) the command timeout (in seconds)
'''
name = self.__class__.__name__
if command is None:
command = ""
if parameters is None:
parameters = []
self._command = command
self._parameters = parameters
self._timeout = timeout
self._customText = text
self.call__init__wo_kw(Qt.QPushButton, parent)
self.call__init__(TaurusBaseWidget, name, designMode=designMode)
if icon is not None:
self.setIcon(Qt.QIcon(icon))
self.setCustomText(text)
self.setDefault(False)
self.setAutoDefault(False)
self.clicked.connect(self.onClicked)
[docs] def getDisplayValue(self):
'''see :meth:`TaurusBaseComponent.displayValue`'''
if self._customText is not None:
return self._customText
if len(self._command) == 0:
return '---'
modelobj = self.getModelObj()
if modelobj is None or not hasattr(modelobj, self._command):
return '---'
return self._command
@ProtectTaurusMessageBox(title="Unexpected error when executing command")
def _onClicked(self):
'''Slot called when the button is clicked. It executes the command with
parameters. It may issue a warning if the command is flagged as
dangerous. On successful execution, it returns the command result and it
emits a "commandExecuted" signal with the result as well.
:return: The result of the command. The type depends on the command. It
may be None.
.. seealso:: :meth:`setCommand`, :meth:`setParameters`, :meth:`TaurusBaseComponent.isDangerous`
'''
self.debug("launch command %s" % str(self._command))
if len(self._command) == 0:
return
modelobj = self.getModelObj()
if modelobj is None or not hasattr(modelobj, self._command):
self.warning('Device %s does not implement command %s' %
(modelobj, self._command))
return
if self.isDangerous() and not self.getForceDangerousOperations():
result = Qt.QMessageBox.question(self, "Potentially dangerous action",
"%s\nProceed?" % self.getDangerMessage(),
Qt.QMessageBox.Ok | Qt.QMessageBox.Cancel,
Qt.QMessageBox.Ok)
if result != Qt.QMessageBox.Ok:
return
# After all the checks, we can finally do the action
orig_timeout = modelobj.get_timeout_millis()
try:
if self._timeout is not None:
modelobj.set_timeout_millis(int(self._timeout * 1000))
result = modelobj.command_inout(self._command, self._castParameters(
self._parameters, self._command, modelobj))
except Exception, e:
self.error('Unexpected error when executing command %s of %s: %s' % (
self._command, modelobj.getNormalName(), str(e)))
raise
finally:
modelobj.set_timeout_millis(orig_timeout)
self.commandExecuted.emit(result)
return result
def _castParameters(self, parameters=None, command=None, dev=None):
'''Internal method used to cast the command paramters to the appropriate
type required for the given command
:param parameters: (sequence) a sequence of parameters. If None is
passed, the currently set parameters are used.
:param command: (str) the command name. If None is passed, the currently
set command is used.
:param dev: (taurus.core.taurusdevice.TaurusDevice) the device on which the command is
executed. If None is passed, the current model is used.
:return: (sequence or scalar) a sequence of parameters (or a scalar if only one parameter)
'''
if parameters is None:
parameters = self._parameters
if command is None:
command = self._command
if dev is None:
dev = self.getModelObj()
try:
param_type = dev.command_query(command).in_type
except Exception, e:
self.warning(
'Cannot get parameters info for command %s:%s' % (command, str(e)))
return parameters
if param_type == PyTango.CmdArgType.DevVoid:
return None
if PyTango.is_int_type(param_type, True):
cast_type = int
elif PyTango.is_float_type(param_type, True):
cast_type = float
elif param_type == PyTango.CmdArgType.DevVarStringArray or param_type == PyTango.CmdArgType.DevString:
cast_type = str
elif param_type == PyTango.CmdArgType.DevVarBooleanArray or param_type == PyTango.CmdArgType.DevBoolean:
cast_type = bool
else:
self.info(
'Unsupported parameters type (%s). Casting to "str"' % str(param_type))
cast_type = str
if PyTango.is_scalar_type(param_type):
if parameters:
return cast_type(parameters[0])
else:
return parameters
else:
return map(cast_type, parameters)
[docs] def setCommand(self, commandName):
'''sets the command to be executed when the button is clicked
:param commandName: (str or None) the command name
'''
if commandName is None:
self._command = ""
else:
self._command = str(commandName)
self._setText(self.getDisplayValue())
[docs] def getCommand(self):
'''returns the command name to be executed when the button is clicked
:return: (str or None) the command name
'''
return self._command
[docs] def setParameters(self, parameters):
'''
Sets the parameters to be used on command execution.
:param parameters: (sequence) a sequence of parameters. If the
elements of the sequence are not of the right type
required for the parameter, an automatic conversion
will be attempted on execution time. As a special
case, if parameters is a string not starting and
ending in quote characters, it will be splitted on
whitespace to obtain a sequence of parameters. If
it is a string starting and ending with quotes, the
quotes will be removed and the quoted text will not
be splitted.
'''
if isinstance(parameters, (basestring, Qt.QString)):
parameters = str(parameters).strip()
if parameters[0] in ('"', "'") and parameters[0] == parameters[-1]:
parameters = [parameters[1:-1]]
else:
parameters = parameters.split()
self._parameters = parameters
[docs] def getParameters(self):
'''returns the parameters to be used on command execution
:param parameters: (sequence)
'''
return self._parameters
[docs] def setCustomText(self, customText=None):
'''Sets a custom text for the button (by default it is the command name)
:param customText: (str or None) the custom text. If None passed, it
will use the command name
'''
self._customText = customText
self._setText(self.getDisplayValue())
[docs] def getCustomText(self):
'''Returns the custom text of the buttom, or None if no custom text is
used
'''
return self._customText
@Qt.pyqtSlot(float)
@Qt.pyqtSlot(int)
[docs] def setTimeout(self, timeout):
'''Sets the number of seconds to wait for the result of the command.
.. seealso:: :meth:`PyTango.DeviceProxy.command_inout`
:param timeout: (float) the command timeout in seconds
(timeout <0 or timeout=None disables the timeout)
'''
if timeout < 0:
timeout = None
self._timeout = timeout
[docs] def getTimeout(self):
'''
Returns the number of seconds to wait for the result of the command
(or -1 if timeout is disabled)
'''
ret = self._timeout
if ret is None or ret < 0:
ret = -1
return ret
@classmethod
[docs] def getQtDesignerPluginInfo(cls):
return {
'group': 'Taurus Buttons',
'icon': 'designer:pushbutton.png',
'module': 'taurus.qt.qtgui.button',
'container': False}
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
# Qt Properties #
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
Model = Qt.pyqtProperty("QString", TaurusBaseWidget.getModel,
TaurusBaseWidget.setModel, TaurusBaseWidget.resetModel)
UseParentModel = Qt.pyqtProperty("bool", TaurusBaseWidget.getUseParentModel,
TaurusBaseWidget.setUseParentModel, TaurusBaseWidget.resetUseParentModel)
Command = Qt.pyqtProperty("QString", getCommand, setCommand, resetCommand)
Parameters = Qt.pyqtProperty(
"QStringList", getParameters, setParameters, resetParameters)
DangerMessage = Qt.pyqtProperty("QString", TaurusBaseWidget.getDangerMessage,
TaurusBaseWidget.setDangerMessage, TaurusBaseWidget.resetDangerMessage)
CustomText = Qt.pyqtProperty(
"QString", getCustomText, setCustomText, resetCustomText)
Timeout = Qt.pyqtProperty("double", getTimeout, setTimeout, resetTimeout)
[docs]class TaurusLockButton(Qt.QPushButton, TaurusBaseWidget):
_LOCK_MAP = {LockStatus.Unlocked: "extra_icon:lock_unlocked.svg",
LockStatus.Locked: "extra_icon:lock_locked_unpreviledged.svg",
LockStatus.LockedMaster: "extra_icon:lock_locked.svg",
LockStatus.Unknown: "extra_icon:lock_unknown.svg"}
def __init__(self, parent=None, designMode=False):
self._lock_info = TaurusLockInfo()
name = self.__class__.__name__
self.call__init__wo_kw(Qt.QPushButton, parent)
self.call__init__(TaurusBaseWidget, name, designMode=designMode)
self.toggled.connect(self.on_toggle)
self.setCheckable(True)
self.setAutoTooltip(False)
self.insertEventFilter(eventfilters.IGNORE_ALL)
self.update_button()
@classmethod
[docs] def getQtDesignerPluginInfo(cls):
return {
'group': 'Taurus Buttons',
'icon': 'designer:pushbutton.png',
'module': 'taurus.qt.qtgui.button',
'container': False}
[docs] def get_lock_info(self, cache=False):
dev = self.getModelObj()
if dev is not None:
self._lock_info = dev.getLockInfo(cache=cache)
return self._lock_info
[docs] def update_button(self, lock_info=None):
if lock_info is None:
lock_info = self.get_lock_info()
status = lock_info.status
self.setIcon(Qt.QIcon(self._LOCK_MAP[status]))
self.setDown(status in (LockStatus.Locked, LockStatus.LockedMaster))
self.setToolTip(lock_info.status_msg)
self.update()
return lock_info
def _on_toggle(self, down):
dev = self.getModelObj()
if down:
dev.lock()
else:
dev.unlock()
self.update_button()
[docs] def on_toggle(self, down):
try:
self._on_toggle(down)
except:
import sys
from taurus.qt.qtgui.dialog import TaurusMessageBox
msgbox = TaurusMessageBox(*sys.exc_info())
msgbox.setWindowTitle("Error locking device")
if self.update_button().status == LockStatus.Locked:
msgbox.setText(self._lock_info.status_msg)
msgbox.exec_()
model = Qt.pyqtProperty("QString", TaurusBaseWidget.getModel, setModel,
TaurusBaseWidget.resetModel)
def lockButtonMain():
import sys
import taurus.qt.qtgui.application
Application = taurus.qt.qtgui.application.TaurusApplication
app = Application.instance()
owns_app = app is None
if owns_app:
import taurus.core.util.argparse
parser = taurus.core.util.argparse.get_taurus_parser()
parser.usage = "%prog [options] <full_attribute_name(s)>"
app = Application(sys.argv, cmd_line_parser=parser,
app_name="Taurus lock button demo", app_version="1.0",
org_domain="Taurus", org_name="Tango community")
args = app.get_command_line_args()
if len(args) == 0:
w = demo()
else:
models = map(str.lower, args)
w = Qt.QWidget()
layout = Qt.QGridLayout()
w.setLayout(layout)
for model in models:
lock_button = TaurusLockButton()
lock_button.model = model
layout.addWidget(lock_button)
w.show()
if owns_app:
sys.exit(app.exec_())
else:
return w
def commandButtonMain():
import sys
from taurus.qt.qtgui.application import TaurusApplication
app = TaurusApplication()
form = TaurusCommandButton(parent=None, designMode=False, command='DevBoolean', parameters=[
123], icon='logos:taurus.png', text='launch: DevBoolean 123')
form.setModel('sys/tg_test/1')
form.setDangerMessage(
'Booo scary command!!\n Maybe you should think twice!')
def f(*a):
print a
form.commandExecuted.connect(f)
form.show()
sys.exit(app.exec_())
def launcherButtonMain():
import sys
from taurus.qt.qtgui.application import TaurusApplication
app = TaurusApplication()
# Creating button giving the widget
# from taurus.qt.qtgui.plot import TaurusPlot
# w = TaurusPlot()
# form = TaurusLauncherButton(parent=None, designMode=False, widget=w,
# icon='logos:taurus.png'), text='show')
# Creating button giving the widget class name
# form = TaurusLauncherButton(parent=None, designMode=False,
# widget='TaurusPlot', icon='logos:taurus.png',
# text='show')
# Creating button using a derived class with the name widget class
# hardcoded
class MyButton(TaurusLauncherButton):
_widgetClassName = 'TaurusPlot'
_icon = 'logos:taurus.png'
_text = 'show'
form = MyButton()
form.setModel('sys/tg_test/1/wave')
form.show()
sys.exit(app.exec_())
def main():
lockButtonMain()
def demo():
'''Lock button'''
lock_button = TaurusLockButton()
lock_button.model = "sys/tg_test/1"
return lock_button
if __name__ == '__main__':
# lockButtonMain()
launcherButtonMain()
# commandButtonMain()