From 40fbc3a7e21fbc5f34bdcf275dfd3c189bc70392 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Thu, 27 Oct 2022 18:57:07 +0200
Subject: [PATCH] create widget is handling enabled dropping of thumbnails

---
 openpype/tools/publisher/constants.py            |  7 ++++---
 .../tools/publisher/widgets/create_widget.py     | 16 ++++++++++++++--
 2 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/openpype/tools/publisher/constants.py b/openpype/tools/publisher/constants.py
index 8bea69c812..74337ea1ab 100644
--- a/openpype/tools/publisher/constants.py
+++ b/openpype/tools/publisher/constants.py
@@ -20,9 +20,10 @@ INSTANCE_ID_ROLE = QtCore.Qt.UserRole + 1
 SORT_VALUE_ROLE = QtCore.Qt.UserRole + 2
 IS_GROUP_ROLE = QtCore.Qt.UserRole + 3
 CREATOR_IDENTIFIER_ROLE = QtCore.Qt.UserRole + 4
-FAMILY_ROLE = QtCore.Qt.UserRole + 5
-GROUP_ROLE = QtCore.Qt.UserRole + 6
-CONVERTER_IDENTIFIER_ROLE = QtCore.Qt.UserRole + 7
+CREATOR_THUMBNAIL_ENABLED_ROLE = QtCore.Qt.UserRole + 5
+FAMILY_ROLE = QtCore.Qt.UserRole + 6
+GROUP_ROLE = QtCore.Qt.UserRole + 7
+CONVERTER_IDENTIFIER_ROLE = QtCore.Qt.UserRole + 8
 
 
 __all__ = (
diff --git a/openpype/tools/publisher/widgets/create_widget.py b/openpype/tools/publisher/widgets/create_widget.py
index a610c405a4..f0db132d98 100644
--- a/openpype/tools/publisher/widgets/create_widget.py
+++ b/openpype/tools/publisher/widgets/create_widget.py
@@ -20,7 +20,8 @@ from .precreate_widget import PreCreateWidget
 from ..constants import (
     VARIANT_TOOLTIP,
     CREATOR_IDENTIFIER_ROLE,
-    FAMILY_ROLE
+    FAMILY_ROLE,
+    CREATOR_THUMBNAIL_ENABLED_ROLE,
 )
 
 SEPARATORS = ("---separator---", "---")
@@ -457,6 +458,10 @@ class CreateWidget(QtWidgets.QWidget):
 
             item.setData(creator_item.label, QtCore.Qt.DisplayRole)
             item.setData(identifier, CREATOR_IDENTIFIER_ROLE)
+            item.setData(
+                creator_item.create_allow_thumbnail,
+                CREATOR_THUMBNAIL_ENABLED_ROLE
+            )
             item.setData(creator_item.family, FAMILY_ROLE)
 
         # Remove families that are no more available
@@ -558,6 +563,10 @@ class CreateWidget(QtWidgets.QWidget):
             self._set_context_enabled(creator_item.create_allow_context_change)
             self._refresh_asset()
 
+        self._thumbnail_widget.set_drop_enabled(
+            creator_item.create_allow_thumbnail
+        )
+
         default_variants = creator_item.default_variants
         if not default_variants:
             default_variants = ["Main"]
@@ -742,7 +751,10 @@ class CreateWidget(QtWidgets.QWidget):
             task_name = self._get_task_name()
 
         pre_create_data = self._pre_create_widget.current_value()
-        pre_create_data[PRE_CREATE_THUMBNAIL_KEY] = self._last_thumbnail_path
+        if index.data(CREATOR_THUMBNAIL_ENABLED_ROLE):
+            pre_create_data[PRE_CREATE_THUMBNAIL_KEY] = (
+                self._last_thumbnail_path
+            )
 
         # Where to define these data?
         # - what data show be stored?