From 626cb387934956d2c1eea08a535dd83223a876cd Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Fri, 7 Oct 2022 12:48:34 +0200
Subject: [PATCH] added ability to get existing subsets for passet asset name
 via controller

---
 openpype/tools/publisher/control.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/openpype/tools/publisher/control.py b/openpype/tools/publisher/control.py
index 8abe62e4b1..89619f70f7 100644
--- a/openpype/tools/publisher/control.py
+++ b/openpype/tools/publisher/control.py
@@ -11,6 +11,7 @@ import pyblish.api
 from openpype.client import (
     get_assets,
     get_asset_by_id,
+    get_subsets,
 )
 from openpype.lib.events import EventSystem
 from openpype.pipeline import (
@@ -837,6 +838,10 @@ class AbstractPublisherController(object):
     def get_task_names_by_asset_names(self, asset_names):
         pass
 
+    @abstractmethod
+    def get_existing_subset_names(self, asset_name):
+        pass
+
     @abstractmethod
     def reset(self):
         """Reset whole controller.
@@ -1223,6 +1228,21 @@ class PublisherController(AbstractPublisherController):
             )
         return result
 
+    def get_existing_subset_names(self, asset_name):
+        project_name = self.project_name
+        asset_doc = self._asset_docs_cache.get_asset_by_name(asset_name)
+        if not asset_doc:
+            return None
+
+        asset_id = asset_doc["_id"]
+        subset_docs = get_subsets(
+            project_name, asset_ids=[asset_id], fields=["name"]
+        )
+        return {
+            subset_doc["name"]
+            for subset_doc in subset_docs
+        }
+
     def reset(self):
         """Reset everything related to creation and publishing."""
         # Stop publishing