mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 16:34:53 +01:00
158 lines
4.7 KiB
Python
158 lines
4.7 KiB
Python
import logging
|
|
from . import QtCore, QtGui
|
|
from . import TreeModel, Node
|
|
from . import style, awesome
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def _iter_model_rows(model,
|
|
column,
|
|
include_root=False):
|
|
"""Iterate over all row indices in a model"""
|
|
indices = [QtCore.QModelIndex()] # start iteration at root
|
|
|
|
for index in indices:
|
|
|
|
# Add children to the iterations
|
|
child_rows = model.rowCount(index)
|
|
for child_row in range(child_rows):
|
|
child_index = model.index(child_row, column, index)
|
|
indices.append(child_index)
|
|
|
|
if not include_root and not index.isValid():
|
|
continue
|
|
|
|
yield index
|
|
|
|
|
|
class AssetModel(TreeModel):
|
|
"""A model listing assets in the silo in the active project.
|
|
|
|
The assets are displayed in a treeview, they are visually parented by
|
|
a `visualParent` field in the database containing an `_id` to a parent
|
|
asset.
|
|
|
|
"""
|
|
|
|
COLUMNS = ["label"]
|
|
Name = 0
|
|
Deprecated = 2
|
|
ObjectId = 3
|
|
|
|
DocumentRole = QtCore.Qt.UserRole + 2
|
|
ObjectIdRole = QtCore.Qt.UserRole + 3
|
|
|
|
def __init__(self, parent):
|
|
super(AssetModel, self).__init__(parent=parent)
|
|
self.parent_widget = parent
|
|
self.refresh()
|
|
|
|
@property
|
|
def db(self):
|
|
return self.parent_widget.db
|
|
|
|
def _add_hierarchy(self, parent=None):
|
|
|
|
# Find the assets under the parent
|
|
find_data = {
|
|
"type": "asset"
|
|
}
|
|
if parent is None:
|
|
find_data['$or'] = [
|
|
{'data.visualParent': {'$exists': False}},
|
|
{'data.visualParent': None}
|
|
]
|
|
else:
|
|
find_data["data.visualParent"] = parent['_id']
|
|
|
|
assets = self.db.find(find_data).sort('name', 1)
|
|
for asset in assets:
|
|
# get label from data, otherwise use name
|
|
data = asset.get("data", {})
|
|
label = data.get("label", asset['name'])
|
|
tags = data.get("tags", [])
|
|
|
|
# store for the asset for optimization
|
|
deprecated = "deprecated" in tags
|
|
|
|
node = Node({
|
|
"_id": asset['_id'],
|
|
"name": asset["name"],
|
|
"label": label,
|
|
"type": asset['type'],
|
|
"tags": ", ".join(tags),
|
|
"deprecated": deprecated,
|
|
"_document": asset
|
|
})
|
|
self.add_child(node, parent=parent)
|
|
|
|
# Add asset's children recursively
|
|
self._add_hierarchy(node)
|
|
|
|
def refresh(self):
|
|
"""Refresh the data for the model."""
|
|
|
|
self.clear()
|
|
if (
|
|
self.db.active_project() is None or
|
|
self.db.active_project() == ''
|
|
):
|
|
return
|
|
self.beginResetModel()
|
|
self._add_hierarchy(parent=None)
|
|
self.endResetModel()
|
|
|
|
def flags(self, index):
|
|
return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable
|
|
|
|
def data(self, index, role):
|
|
|
|
if not index.isValid():
|
|
return
|
|
|
|
node = index.internalPointer()
|
|
if role == QtCore.Qt.DecorationRole: # icon
|
|
|
|
column = index.column()
|
|
if column == self.Name:
|
|
|
|
# Allow a custom icon and custom icon color to be defined
|
|
data = node["_document"]["data"]
|
|
icon = data.get("icon", None)
|
|
color = data.get("color", style.colors.default)
|
|
|
|
if icon is None:
|
|
# Use default icons if no custom one is specified.
|
|
# If it has children show a full folder, otherwise
|
|
# show an open folder
|
|
has_children = self.rowCount(index) > 0
|
|
icon = "folder" if has_children else "folder-o"
|
|
|
|
# Make the color darker when the asset is deprecated
|
|
if node.get("deprecated", False):
|
|
color = QtGui.QColor(color).darker(250)
|
|
|
|
try:
|
|
key = "fa.{0}".format(icon) # font-awesome key
|
|
icon = awesome.icon(key, color=color)
|
|
return icon
|
|
except Exception as exception:
|
|
# Log an error message instead of erroring out completely
|
|
# when the icon couldn't be created (e.g. invalid name)
|
|
log.error(exception)
|
|
|
|
return
|
|
|
|
if role == QtCore.Qt.ForegroundRole: # font color
|
|
if "deprecated" in node.get("tags", []):
|
|
return QtGui.QColor(style.colors.light).darker(250)
|
|
|
|
if role == self.ObjectIdRole:
|
|
return node.get("_id", None)
|
|
|
|
if role == self.DocumentRole:
|
|
return node.get("_document", None)
|
|
|
|
return super(AssetModel, self).data(index, role)
|