### Navigation

- index
- modules |
- Camelot documentation »
- Module code »
- camelot.view.controls.delegates »

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

    #  ============================================================================
    #
    #  Copyright (C) 2007-2011 Conceptive Engineering bvba. All rights reserved.
    #  www.conceptive.be / project-camelot@conceptive.be
    #
    #  This file is part of the Camelot Library.
    #
    #  This file may be used under the terms of the GNU General Public
    #  License version 2.0 as published by the Free Software Foundation
    #  and appearing in the file license.txt included in the packaging of
    #  this file.  Please review this information to ensure GNU
    #  General Public Licensing requirements will be met.
    #
    #  If you are unsure which license is appropriate for your use, please
    #  visit www.python-camelot.com or contact project-camelot@conceptive.be
    #
    #  This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
    #  WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
    #
    #  For use of this library in commercial applications, please contact
    #  project-camelot@conceptive.be
    #
    #  ============================================================================

    from PyQt4 import QtGui, QtCore
    from PyQt4.QtCore import Qt

    from customdelegate import CustomDelegate, DocumentationMetaclass
    from camelot.view.controls import editors
    from camelot.core.utils import ugettext as _, variant_to_pyobject
    from camelot.view.proxy import ValueLoading

    class BoolDelegate(CustomDelegate):
        """Custom delegate for boolean values"""

        __metaclass__ = DocumentationMetaclass

        editor = editors.BoolEditor

        def paint(self, painter, option, index):
            painter.save()
            self.drawBackground(painter, option, index)
            checked = index.model().data(index, Qt.EditRole).toBool()

            background_color = QtGui.QColor(index.model().data(index, Qt.BackgroundRole))

            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())
            else:
                painter.fillRect(option.rect, background_color)

            if checked:
                check_option.state = option.state | QtGui.QStyle.State_On
            else:
                check_option.state = option.state | QtGui.QStyle.State_Off

            QtGui.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_pyobject(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 = index.model().data(index, Qt.EditRole).toBool()
            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()

### Navigation

- index
- modules |
- Camelot documentation »
- Module code »
- camelot.view.controls.delegates »

Comments
blog comments powered by Disqus