From e2187daeaf93975717071319723a6acd1bfa3b53 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Thu, 1 Jul 2021 20:08:43 +0200
Subject: [PATCH] added method for refreshing creators

---
 openpype/tools/new_publisher/widgets.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/openpype/tools/new_publisher/widgets.py b/openpype/tools/new_publisher/widgets.py
index c7721bc965..37554f5e05 100644
--- a/openpype/tools/new_publisher/widgets.py
+++ b/openpype/tools/new_publisher/widgets.py
@@ -185,6 +185,7 @@ class CreateDialog(QtWidgets.QDialog):
 
     def refresh(self):
         self._refresh_asset()
+        self._refresh_creators()
 
     def _refresh_asset(self):
         asset_name = self.dbcon.Session.get("AVALON_ASSET")
@@ -216,6 +217,30 @@ class CreateDialog(QtWidgets.QDialog):
             )
             self._subset_names = set(subset_docs.distinct("name"))
 
+    def _refresh_creators(self):
+        # Refresh creators and add their families to list
+        existing_items = {}
+        old_families = set()
+        for row in range(self.family_model.rowCount()):
+            item = self.family_model.item(row, 0)
+            family = item.data(QtCore.Qt.DisplayRole)
+            existing_items[family] = item
+            old_families.add(family)
+
+        # Add new families
+        new_families = set()
+        for family, creator in self.controller.creators.items():
+            # TODO add details about creator
+            new_families.add(family)
+            if family not in existing_items:
+                item = QtGui.QStandardItem(family)
+                self.family_model.appendRow(item)
+
+        # Remove families that are no more available
+        for family in (old_families - new_families):
+            item = existing_items[family]
+            self.family_model.takeRow(item.row())
+
     def _on_control_reset(self):
         self.refresh()