Fix exception even for different use case

ynput · May 7, 2024 · f1afd16 · f1afd16
1 parent a3c4817
commit f1afd16
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/client/ayon_core/hosts/photoshop/plugins/create/create_image.py b/client/ayon_core/hosts/photoshop/plugins/create/create_image.py
@@ -35,11 +35,12 @@ def create(self, product_name_from_ui, data, pre_create_data):
         create_empty_group = False
 
         stub = api.stub()  # only after PS is up
-        try:
-            top_level_selected_items = stub.get_selected_layers()
-        except ValueError:
-            raise CreatorError("Cannot group locked Background layer!")
         if pre_create_data.get("use_selection"):
+            try:
+                top_level_selected_items = stub.get_selected_layers()
+            except ValueError:
+                raise CreatorError("Cannot group locked Background layer!")
+
             only_single_item_selected = len(top_level_selected_items) == 1
             if (
                     only_single_item_selected or
@@ -53,8 +54,11 @@ def create(self, product_name_from_ui, data, pre_create_data):
                 group = stub.group_selected_layers(product_name_from_ui)
                 groups_to_create.append(group)
         else:
-            stub.select_layers(stub.get_layers())
-            group = stub.group_selected_layers(product_name_from_ui)
+            try:
+                stub.select_layers(stub.get_layers())
+                group = stub.group_selected_layers(product_name_from_ui)
+            except ValueError:
+                raise CreatorError("Cannot group locked Background layer!")
 
             groups_to_create.append(group)