Source code for shotgun_fields.status_list_widget

# Copyright (c) 2016 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.

import sgtk
from sgtk.platform.qt import QtGui, QtCore
from tank_vendor import six
from .label_base_widget import LabelBaseWidget
from .shotgun_field_meta import ShotgunFieldMeta

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


@six.add_metaclass(ShotgunFieldMeta)
class StatusListWidget(LabelBaseWidget):
    """
    Display a ``status_list`` field value as returned by the Shotgun API.
    """

    _DISPLAY_TYPE = "status_list"

    def _string_value(self, value):
        """
        Convert the Shotgun value for this field into a string

        :param str value: valid Shotgun status code
        """
        str_val = shotgun_globals.get_status_display_name(value)
        color_str = shotgun_globals.get_status_color(value)

        if color_str:
            # append colored box to indicate status color
            str_val = "<span style='color: rgb(%s)'>&#9608;</span>&nbsp;%s" % (
                color_str,
                str_val,
            )

        return str_val


@six.add_metaclass(ShotgunFieldMeta)
class StatusListEditorWidget(QtGui.QComboBox):
    """
    Allows editing of a ``status_list`` field value as returned by the Shotgun API.
    """

    _EDITOR_TYPE = "status_list"
    _IMMEDIATE_APPLY = True

    def get_value(self):
        """
        :return: The internal value being displayed by the widget.
        """
        return self.itemData(self.currentIndex())

    def setup_widget(self):
        """
        Prepare the widget for display

        Called by the metaclass during initialization.
        """
        self.addItem("")
        self.setMinimumWidth(125)

        valid_values = shotgun_globals.get_valid_values(
            self._entity_type, self._field_name
        )
        for value in valid_values:
            self.addItem(shotgun_globals.get_status_display_name(value), value)

        self.activated.connect(lambda i: self.value_changed.emit())

        self.setSizeAdjustPolicy(self.AdjustToContents)

    def _begin_edit(self):
        """
        Prepare the widget for editing by showing the popup
        """
        self.adjustSize()
        self.showPopup()

    def _display_default(self):
        """
        Display the default value of the widget.
        """
        self.setCurrentIndex(0)

    def _display_value(self, value):
        """
        Set the value displayed by the widget.

        :param value: The value returned by the Shotgun API to be displayed
        """
        if value is None:
            self.setCurrentIndex(0)
            return

        display_value = shotgun_globals.get_status_display_name(value)
        index = self.findText(display_value)
        if index != -1:
            self.setCurrentIndex(index)
        else:
            self.setCurrentIndex(0)