Merge pull request #1078 from pypeclub/PS_psb_in_workfiles

PS - added support for .psb in workfiles
ynput · Mar 3, 2021 · 6803943 · 6803943
2 parents 2053052 + f84d68c
commit 6803943
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/pype/plugins/photoshop/publish/collect_workfile.py b/pype/plugins/photoshop/publish/collect_workfile.py
@@ -31,9 +31,10 @@ def process(self, context):
  })
 
  # creating representation
+ _, ext = os.path.splitext(file_path)
  instance.data["representations"].append({
- "name": "psd",
- "ext": "psd",
+ "name": ext[1:],
+ "ext": ext[1:],
  "files": base_name,
  "stagingDir": staging_dir,
  })
diff --git a/pype/plugins/photoshop/publish/increment_workfile.py b/pype/plugins/photoshop/publish/increment_workfile.py
@@ -1,3 +1,4 @@
+import os
 import pyblish.api
 from pype.action import get_errored_plugins_from_data
 from pype.lib import version_up
@@ -25,6 +26,7 @@ def process(self, instance):
  )
 
  scene_path = version_up(instance.context.data["currentFile"])
- photoshop.stub().saveAs(scene_path, 'psd', True)
+ _, ext = os.path.splitext(scene_path)
+ photoshop.stub().saveAs(scene_path, ext[1:], True)
 
  self.log.info("Incremented workfile to: {}".format(scene_path))