From d87e70b3c90490e160d314373bf45da8c255fd93 Mon Sep 17 00:00:00 2001
From: Petr Kalis <petr.kalis@gmail.com>
Date: Tue, 19 Apr 2022 15:21:05 +0200
Subject: [PATCH] OP-3103 - added plugin to parse batch file to PS

Plugin in webpublisher folder doesn't get triggered in PS processing.
---
 .../plugins/publish/collect_batch_data.py     | 73 +++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 openpype/hosts/photoshop/plugins/publish/collect_batch_data.py

diff --git a/openpype/hosts/photoshop/plugins/publish/collect_batch_data.py b/openpype/hosts/photoshop/plugins/publish/collect_batch_data.py
new file mode 100644
index 0000000000..5e6e916611
--- /dev/null
+++ b/openpype/hosts/photoshop/plugins/publish/collect_batch_data.py
@@ -0,0 +1,73 @@
+"""Parses batch context from json and continues in publish process.
+
+Provides:
+    context -> Loaded batch file.
+        - asset
+        - task  (task name)
+        - taskType
+        - project_name
+        - variant
+
+Code is practically copy of `openype/hosts/webpublish/collect_batch_data` as
+webpublisher should be eventually ejected as an addon, eg. mentioned plugin
+shouldn't be pushed into general publish plugins.
+"""
+
+import os
+
+import pyblish.api
+from avalon import io
+from openpype.lib.plugin_tools import (
+    parse_json,
+    get_batch_asset_task_info
+)
+
+
+class CollectBatchData(pyblish.api.ContextPlugin):
+    """Collect batch data from json stored in 'OPENPYPE_PUBLISH_DATA' env dir.
+
+    The directory must contain 'manifest.json' file where batch data should be
+    stored.
+    """
+    # must be really early, context values are only in json file
+    order = pyblish.api.CollectorOrder - 0.495
+    label = "Collect batch data"
+    hosts = ["photoshop"]
+    targets = ["remotepublish"]
+
+    def process(self, context):
+        self.log.info("CollectBatchData")
+        batch_dir = os.environ.get("OPENPYPE_PUBLISH_DATA")
+
+        assert batch_dir, (
+            "Missing `OPENPYPE_PUBLISH_DATA`")
+
+        assert os.path.exists(batch_dir), \
+            "Folder {} doesn't exist".format(batch_dir)
+
+        project_name = os.environ.get("AVALON_PROJECT")
+        if project_name is None:
+            raise AssertionError(
+                "Environment `AVALON_PROJECT` was not found."
+                "Could not set project `root` which may cause issues."
+            )
+
+        batch_data = parse_json(os.path.join(batch_dir, "manifest.json"))
+
+        context.data["batchDir"] = batch_dir
+        context.data["batchData"] = batch_data
+
+        asset_name, task_name, task_type = get_batch_asset_task_info(
+            batch_data["context"]
+        )
+
+        os.environ["AVALON_ASSET"] = asset_name
+        io.Session["AVALON_ASSET"] = asset_name
+        os.environ["AVALON_TASK"] = task_name
+        io.Session["AVALON_TASK"] = task_name
+
+        context.data["asset"] = asset_name
+        context.data["task"] = task_name
+        context.data["taskType"] = task_type
+        context.data["project_name"] = project_name
+        context.data["variant"] = batch_data["variant"]