From 32923208687a221fb11f2f85cb43680292bc3981 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Thu, 10 Mar 2022 15:56:39 +0100
Subject: [PATCH] implemented get asset icon function

---
 openpype/tools/utils/__init__.py      |  1 +
 openpype/tools/utils/assets_widget.py | 27 +++++------------
 openpype/tools/utils/lib.py           | 42 +++++++++++++++++++++++++++
 3 files changed, 50 insertions(+), 20 deletions(-)

diff --git a/openpype/tools/utils/__init__.py b/openpype/tools/utils/__init__.py
index c15e9f8139..6ab9e75b52 100644
--- a/openpype/tools/utils/__init__.py
+++ b/openpype/tools/utils/__init__.py
@@ -16,6 +16,7 @@ from .lib import (
     set_style_property,
     DynamicQThread,
     qt_app_context,
+    get_asset_icon,
 )
 
 from .models import (
diff --git a/openpype/tools/utils/assets_widget.py b/openpype/tools/utils/assets_widget.py
index d410b0f1c3..4c77b81c0e 100644
--- a/openpype/tools/utils/assets_widget.py
+++ b/openpype/tools/utils/assets_widget.py
@@ -16,7 +16,10 @@ from .views import (
 )
 from .widgets import PlaceholderLineEdit
 from .models import RecursiveSortFilterProxyModel
-from .lib import DynamicQThread
+from .lib import (
+    DynamicQThread,
+    get_asset_icon
+)
 
 if Qt.__binding__ == "PySide":
     from PySide.QtGui import QStyleOptionViewItemV4
@@ -508,25 +511,9 @@ class AssetModel(QtGui.QStandardItemModel):
                 item.setData(asset_label, QtCore.Qt.DisplayRole)
                 item.setData(asset_label, ASSET_LABEL_ROLE)
 
-            icon_color = asset_data.get("color") or style.colors.default
-            icon_name = asset_data.get("icon")
-            if not icon_name:
-                # Use default icons if no custom one is specified.
-                # If it has children show a full folder, otherwise
-                # show an open folder
-                if item.rowCount() > 0:
-                    icon_name = "folder"
-                else:
-                    icon_name = "folder-o"
-
-            try:
-                # font-awesome key
-                full_icon_name = "fa.{0}".format(icon_name)
-                icon = qtawesome.icon(full_icon_name, color=icon_color)
-                item.setData(icon, QtCore.Qt.DecorationRole)
-
-            except Exception:
-                pass
+            has_children = item.rowCount() > 0
+            icon = get_asset_icon(asset_data, has_children)
+            item.setData(icon, QtCore.Qt.DecorationRole)
 
     def _threaded_fetch(self):
         asset_docs = self._fetch_asset_docs()
diff --git a/openpype/tools/utils/lib.py b/openpype/tools/utils/lib.py
index 1cbc632804..d57b44728d 100644
--- a/openpype/tools/utils/lib.py
+++ b/openpype/tools/utils/lib.py
@@ -98,6 +98,48 @@ application = qt_app_context
 
 class SharedObjects:
     jobs = {}
+    icons = {}
+
+
+def get_qta_icon_by_name_and_color(icon_name, icon_color):
+    if not icon_name or not icon_color:
+        return None
+
+    full_icon_name = "{0}-{1}".format(icon_name, icon_color)
+    if full_icon_name in SharedObjects.icons:
+        return SharedObjects.icons[full_icon_name]
+
+    variants = [icon_name]
+    qta_instance = qtawesome._instance()
+    for key in qta_instance.charmap.keys():
+        variants.append("{0}.{1}".format(key, icon_name))
+
+    icon = None
+    for variant in variants:
+        try:
+            icon = qtawesome.icon(variant, color=icon_color)
+            break
+        except Exception:
+            pass
+
+    SharedObjects.icons[full_icon_name] = icon
+    return icon
+
+
+def get_asset_icon(asset_doc, has_children=False):
+    asset_data = asset_doc.get("data") or {}
+    icon_color = asset_data.get("color") or style.colors.default
+    icon_name = asset_data.get("icon")
+    if not icon_name:
+        # Use default icons if no custom one is specified.
+        # If it has children show a full folder, otherwise
+        # show an open folder
+        if has_children:
+            icon_name = "folder"
+        else:
+            icon_name = "folder-o"
+
+    return get_qta_icon_by_name_and_color(icon_name, icon_color)
 
 
 def schedule(func, time, channel="default"):