### Navigation

- index
- modules |
- Camelot documentation »
- Module code »

# Source code for camelot.view.controls.modeltree

    #  ============================================================================
    #
    #  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
    #
    #  ============================================================================

    """custom tree and tree-items widgets"""

    import logging
    logger = logging.getLogger('camelot.view.controls.modeltree')

    from PyQt4 import QtGui
    from PyQt4.QtCore import Qt

    from camelot.core.utils import ugettext as _

    class ModelItem(QtGui.QTreeWidgetItem):
        """Custom tree item widget"""

        def __init__(self, parent, columns_names, section_item):
            logger.debug('creating new modelitem')
            super(ModelItem, self).__init__(parent, columns_names)

            self.textColumn = 0
            self.iconColumn = 1
            self.section_item = section_item

            for column in (self.textColumn, self.iconColumn):
                self.setToolTip(column, _('Right click to open in New Tab'))

        def _underline(self, enable=False):
            font = self.font(self.textColumn)
            font.setUnderline(enable)
            self.setFont(self.textColumn, font)

        def set_icon(self, icon):
            self.setIcon(self.iconColumn, icon)

    class ModelTree(QtGui.QTreeWidget):
        """Custom tree widget"""

        def __init__(self, header_labels=[''], parent=None):
            logger.debug('creating new modeltree')
            super(ModelTree, self).__init__(parent)
            # we don't select entire rows
            self.setSelectionBehavior(QtGui.QAbstractItemView.SelectItems)

            # we track mouse movement when no button is pressed
            self.setMouseTracking(True)
            self.header_labels = header_labels

            self.setColumnCount(2)
            self.setColumnWidth(0, 160)
            self.setColumnWidth(1, 18)
            self.setHorizontalScrollBarPolicy( Qt.ScrollBarAlwaysOff )
            self.setSelectionBehavior( self.SelectRows )

            self.clear_model_items()
            self.clear_section_items()
            self.fix_header_labels()

        def resizeEvent(self, event):
            self.setColumnWidth(0, self.width()  - 30 )

        def fix_header_labels(self):
            self.setHeaderHidden(True)

        def clear_section_items(self):
            self.section_items = []

        def clear_model_items(self):
            self.modelitems = []

        def mousePressEvent(self, event):
            """Custom context menu"""
            if event.button() == Qt.RightButton:
                self.customContextMenuRequested.emit( event.pos() )
                event.accept()
            else:
                QtGui.QTreeWidget.mousePressEvent(self, event)

        def leaveEvent(self, event):
            if not self.modelitems: return
            for item in self.modelitems: item._underline(False)

        def mouseMoveEvent(self, event):
            if not self.modelitems: return
            for item in self.modelitems: item._underline(False)

            item = self.itemAt(self.mapFromGlobal(self.cursor().pos()))
            if item:
                item._underline(True)

### Navigation

- index
- modules |
- Camelot documentation »
- Module code »

Comments
blog comments powered by Disqus