Add pyblish family whitelist for photoshop

Ellipsanime · Feb 14, 2022 · 687d4b7 · 687d4b7
1 parent a6c2802
commit 687d4b7
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/openpype/hosts/photoshop/plugins/publish/collect_instances.py b/openpype/hosts/photoshop/plugins/publish/collect_instances.py
@@ -1,3 +1,5 @@
+import os
+
 import pyblish.api
 
 from openpype.hosts.photoshop import api as photoshop
@@ -25,6 +27,11 @@ def process(self, context):
  layers = stub.get_layers()
  layers_meta = stub.get_layers_metadata()
  instance_names = []
+
+ families_whitelist = os.getenv("PYBLISH_FAMILY_WHITELIST")
+ if families_whitelist:
+ families_whitelist = families_whitelist.split(',')
+
  for layer in layers:
  layer_data = stub.read(layer, layers_meta)
 
@@ -36,6 +43,12 @@ def process(self, context):
  if "container" in layer_data["id"]:
  continue
 
+ if families_whitelist:
+ if layer_data["family"] not in families_whitelist:
+ self.log.info("Skipped instance with not whitelisted "
+ "family: {}".format(layer_data["family"]))
+ continue
+
  # child_layers = [*layer.Layers]
  # self.log.debug("child_layers {}".format(child_layers))
  # if not child_layers:

diff --git a/openpype/hosts/photoshop/plugins/publish/collect_review.py b/openpype/hosts/photoshop/plugins/publish/collect_review.py
@@ -12,6 +12,16 @@ class CollectReview(pyblish.api.ContextPlugin):
 
  def process(self, context):
  family = "review"
+
+ families_whitelist = os.getenv("PYBLISH_FAMILY_WHITELIST")
+ if families_whitelist:
+ families_whitelist = families_whitelist.split(',')
+ if families_whitelist:
+ if family not in families_whitelist:
+ self.log.info("Skipped instance with not whitelisted "
+ "family: {}".format(family))
+ return
+
  task = os.getenv("AVALON_TASK", None)
  subset = family + task.capitalize()
 

diff --git a/openpype/hosts/photoshop/plugins/publish/collect_workfile.py b/openpype/hosts/photoshop/plugins/publish/collect_workfile.py
@@ -18,6 +18,15 @@ def process(self, context):
  staging_dir = os.path.dirname(file_path)
  base_name = os.path.basename(file_path)
 
+ families_whitelist = os.getenv("PYBLISH_FAMILY_WHITELIST")
+ if families_whitelist:
+ families_whitelist = families_whitelist.split(',')
+ if families_whitelist:
+ if 'workfile' not in families_whitelist:
+ self.log.info("Skipped instance with not whitelisted "
+ "family: {}".format('workfile'))
+ return
+
  # Create instance
  instance = context.create_instance(subset)
  instance.data.update({