Source code for camelot.view.controls.delegates.booldelegate

#  ============================================================================
#
#  Copyright (C) 2007-2016 Conceptive Engineering bvba.
#  www.conceptive.be / info@conceptive.be
#
#  Redistribution and use in source and binary forms, with or without
#  modification, are permitted provided that the following conditions are met:
#      * Redistributions of source code must retain the above copyright
#        notice, this list of conditions and the following disclaimer.
#      * Redistributions in binary form must reproduce the above copyright
#        notice, this list of conditions and the following disclaimer in the
#        documentation and/or other materials provided with the distribution.
#      * Neither the name of Conceptive Engineering nor the
#        names of its contributors may be used to endorse or promote products
#        derived from this software without specific prior written permission.
#  
#  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
#  ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
#  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
#  DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
#  DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
#  (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
#  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
#  ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
#  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
#  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
#  ============================================================================

import six

from ....core.qt import variant_to_py, Qt, QtCore, QtGui, QtWidgets
from .customdelegate import CustomDelegate, DocumentationMetaclass
from camelot.view.controls import editors
from camelot.core.utils import ugettext as _
from camelot.view.proxy import ValueLoading

@six.add_metaclass(DocumentationMetaclass)
[docs]class BoolDelegate(CustomDelegate): """Custom delegate for boolean values""" editor = editors.BoolEditor def paint(self, painter, option, index): painter.save() self.drawBackground( painter, option, index ) checked = variant_to_py(index.model().data(index, Qt.EditRole)) check_option = QtGui.QStyleOptionButton() rect = QtCore.QRect(option.rect.left(), option.rect.top(), option.rect.width(), option.rect.height()) check_option.rect = rect check_option.palette = option.palette if (option.state & QtGui.QStyle.State_Selected): painter.fillRect(option.rect, option.palette.highlight()) elif not self.editable: painter.fillRect(option.rect, option.palette.window()) if checked in (ValueLoading, None): check_option.state = option.state | QtGui.QStyle.State_Off elif checked: check_option.state = option.state | QtGui.QStyle.State_On else: check_option.state = option.state | QtGui.QStyle.State_Off QtWidgets.QApplication.style().drawControl(QtGui.QStyle.CE_CheckBox, check_option, painter) painter.restore()
class TextBoolDelegate(CustomDelegate): editor = editors.TextBoolEditor def __init__(self, parent=None, editable=True, yes='Yes', no='No', color_yes=None, color_no=None, **kwargs): CustomDelegate.__init__(self, parent, editable, **kwargs) self.yes = yes self.no = no self.color_no = color_no self.color_yes = color_yes def paint(self, painter, option, index): painter.save() self.drawBackground(painter, option, index) field_attributes = variant_to_py(index.data(Qt.UserRole)) editable, background_color = True, None if field_attributes != ValueLoading: editable = field_attributes.get( 'editable', True ) background_color = field_attributes.get( 'background_color', None ) rect = option.rect value = variant_to_py(index.model().data(index, Qt.EditRole)) font_color = QtGui.QColor() if value: text = self.yes if self.color_yes: color = self.color_yes else: text = self.no if self.color_no: color = self.color_no font_color.setRgb(color.red(), color.green(), color.blue()) if( option.state & QtGui.QStyle.State_Selected ): painter.fillRect(option.rect, option.palette.highlight()) else: if editable: painter.fillRect(option.rect, background_color or option.palette.base()) else: painter.fillRect(option.rect, background_color or option.palette.window()) painter.setPen(font_color.toRgb()) painter.drawText( rect.x() + 2, rect.y(), rect.width() - 4, rect.height(), Qt.AlignVCenter | Qt.AlignLeft, _(text) ) painter.restore()