Sections

.gitignore

@@ -5,4 +5,5 @@ __pycache__
 dist
 cover
 .coverage
-build
+build
+.idea/

pyblish_lite/app.css

@@ -37,6 +37,11 @@ QListView {
  background: "transparent"
 }
 
+QTreeView {
+ border: 0px;
+ background: "transparent"
+}
+
 QPushButton {
  width: 27px;
  height: 27px;

pyblish_lite/delegate.py

@@ -119,6 +119,66 @@ def sizeHint(self, option, index):
  return QtCore.QSize(option.rect.width(), 20)
 
 
+class Section(QtWidgets.QStyledItemDelegate):
+ """Generic delegate for section header"""
+
+ def paint(self, painter, option, index):
+ """Paint text
+ _
+ My label
+
+ """
+
+ body_rect = QtCore.QRectF(option.rect)
+
+ metrics = painter.fontMetrics()
+
+ label_rect = QtCore.QRectF(option.rect.adjusted(0, 2, 0, -2))
+
+ assert label_rect.width() > 0
+
+ label = index.data(model.Label)
+ label = metrics.elidedText(label,
+ QtCore.Qt.ElideRight,
+ label_rect.width())
+
+ font_color = colors["idle"]
+ if not index.data(model.IsChecked):
+ font_color = colors["inactive"]
+
+ # Maintain reference to state, so we can restore it once we're done
+ painter.save()
+
+ # Draw label
+ painter.setFont(fonts["h4"])
+ painter.setPen(QtGui.QPen(font_color))
+ painter.drawText(label_rect, label)
+
+ if option.state & QtWidgets.QStyle.State_MouseOver:
+ painter.fillRect(body_rect, colors["hover"])
+
+ if option.state & QtWidgets.QStyle.State_Selected:
+ painter.fillRect(body_rect, colors["selected"])
+
+ # Ok, we're done, tidy up.
+ painter.restore()
+
+ def sizeHint(self, option, index):
+ return QtCore.QSize(option.rect.width(), 20)
+
+
+class ItemAndSection(Item):
+ """Generic delegate for model items in proxy tree view"""
+ def paint(self, painter, option, index):
+
+ model = index.model()
+ if model.is_header(index):
+ Section().paint(painter, option, index)
+ return
+
+ super(ItemAndSection, self).paint(painter, option, index)
+
+
 class Artist(QtWidgets.QStyledItemDelegate):
  """Delegate used on Artist page"""
 

pyblish_lite/model.py

@@ -46,6 +46,7 @@
 Label = QtCore.Qt.DisplayRole + 0
 Families = QtCore.Qt.DisplayRole + 1
 Icon = QtCore.Qt.DisplayRole + 13
+Order = QtCore.Qt.UserRole + 62
 
 # The item has not been used
 IsIdle = QtCore.Qt.UserRole + 2
@@ -129,6 +130,7 @@ def __init__(self, parent=None):
  Actions: "actions",
  IsOptional: "optional",
  Icon: "icon",
+ Order: "order",
 
  # GUI-only data
  Type: "_type",