Source code for elided_label.elided_label

# Copyright (c) 2015 Shotgun Software Inc.
#
# CONFIDENTIAL AND PROPRIETARY
#
# This work is provided "AS IS" and subject to the Shotgun Pipeline Toolkit
# Source Code License included in this distribution package. See LICENSE.
# By accessing, using, copying or modifying this work you indicate your
# agreement to the Shotgun Pipeline Toolkit Source Code License. All rights
# not expressly granted therein are reserved by Shotgun Software Inc.

"""
A QLabel that elides text and adds ellipses if the text doesn't fit
correctly within the widget frame.  Handles rich-text.
"""

import sgtk
from sgtk.platform.qt import QtCore, QtGui

utils = sgtk.platform.import_framework("tk-framework-shotgunutils", "utils")


[docs]class ElidedLabel(QtGui.QLabel): """ Label that gracefully handles when the text doesn't fit within the given space. """ def __init__(self, parent=None): """ :param parent: The parent QWidget :type parent: :class:`~PySide.QtGui.QWidget` """ QtGui.QLabel.__init__(self, parent) self._elide_mode = QtCore.Qt.ElideRight self._actual_text = "" self._line_width = 0 self._ideal_width = None self.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Preferred) def sizeHint(self): base_size_hint = super(ElidedLabel, self).sizeHint() return QtCore.QSize(self._get_width_hint(), base_size_hint.height()) def _get_width_hint(self): if not self._ideal_width: doc = QtGui.QTextDocument() try: # add the extra space to buffer the width a bit doc.setHtml(self._actual_text + " ") doc.setDefaultFont(self.font()) width = doc.idealWidth() except Exception: width = self.width() finally: utils.safe_delete_later(doc) self._ideal_width = width return self._ideal_width def _get_elide_mode(self): """ Returns current elide mode :returns: The current elide mode, either QtCore.Qt.ElideLeft or QtCore.Qt.ElideRight """ return self._elide_mode def _set_elide_mode(self, value): """ Set the current elide mode. :param value: The elide mode to use - must be either QtCore.Qt.ElideLeft or QtCore.Qt.ElideRight """ if value != QtCore.Qt.ElideLeft and value != QtCore.Qt.ElideRight: raise ValueError( "elide_mode must be set to either QtCore.Qt.ElideLeft or QtCore.Qt.ElideRight" ) self._elide_mode = value self._update_elided_text() #: Property to get or set the elide mode. The value provided #: should be either QtCore.Qt.ElideLeft or QtCore.Qt.ElideRight elide_mode = property(_get_elide_mode, _set_elide_mode)
[docs] def text(self): """ Overridden base method to return the original unmodified text :returns: The original unmodified text """ return self._actual_text
[docs] def setText(self, text): """ Overridden base method to set the text on the label :param text: The text to set on the label """ # clear out the ideal width so that the widget can recalculate based on # the new text self._ideal_width = None self._actual_text = text self._update_elided_text() # if we're elided, make the tooltip show the full text if super(ElidedLabel, self).text() != self._actual_text: # wrap the actual text in a paragraph so that it wraps nicely self.setToolTip("<p>%s</p>" % (self._actual_text,)) else: self.setToolTip("")
def resizeEvent(self, event): """ Overridden base method called when the widget is resized. :param event: The resize event """ self._update_elided_text() def _update_elided_text(self): """ Update the elided text on the label """ text = self._elide_text(self._actual_text, self._elide_mode) QtGui.QLabel.setText(self, text) def _elide_text(self, text, elide_mode): """ Elide the specified text using the specified mode :param text: The text to elide :param elide_mode: The elide mode to use :returns: The elided text. """ # target width is the label width: target_width = self.width() # Use a QTextDocument to measure html/richtext width doc = QtGui.QTextDocument() try: doc.setHtml(text) doc.setDefaultFont(self.font()) # if line width is already less than the target width then great! line_width = doc.idealWidth() if line_width <= target_width: self._line_width = line_width return text # depending on the elide mode, insert ellipses in the correct place cursor = QtGui.QTextCursor(doc) ellipses = "" if elide_mode != QtCore.Qt.ElideNone: # add the ellipses in the correct place: ellipses = "..." if elide_mode == QtCore.Qt.ElideLeft: cursor.setPosition(0) elif elide_mode == QtCore.Qt.ElideRight: char_count = doc.characterCount() cursor.setPosition(char_count - 1) cursor.insertText(ellipses) ellipses_len = len(ellipses) # remove characters until the text fits within the target width: while line_width > target_width: start_line_width = line_width # if string is less than the ellipses length then just return # an empty string char_count = doc.characterCount() if char_count <= ellipses_len: self._line_width = 0 return "" # calculate the number of characters to remove - should always remove at least 1 # to be sure the text gets shorter! line_width = doc.idealWidth() p = target_width / line_width # play it safe and remove a couple less than the calculated amount chars_to_delete = max(1, char_count - int(float(char_count) * p) - 2) # remove the characters: if elide_mode == QtCore.Qt.ElideLeft: start = ellipses_len end = chars_to_delete + ellipses_len else: # default is to elide right start = max(0, char_count - chars_to_delete - ellipses_len - 1) end = max(0, char_count - ellipses_len - 1) cursor.setPosition(start) cursor.setPosition(end, QtGui.QTextCursor.KeepAnchor) cursor.removeSelectedText() # update line width: line_width = doc.idealWidth() if line_width == start_line_width: break self._line_width = line_width return doc.toHtml() finally: utils.safe_delete_later(doc) @property def line_width(self): """ (:obj:`int`) width of the line of text in pixels """ return self._line_width