From a87f778f1e95d16d09097d923ca3f1d519e86126 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakub@orbi.tools>
Date: Thu, 24 Mar 2022 12:56:18 +0100
Subject: [PATCH] flame: reuse batch groups

---
 openpype/hosts/flame/api/__init__.py          |  4 ++-
 openpype/hosts/flame/api/lib.py               |  9 +++++++
 .../plugins/publish/integrate_batch_group.py  | 26 ++++++++++++++-----
 3 files changed, 31 insertions(+), 8 deletions(-)

diff --git a/openpype/hosts/flame/api/__init__.py b/openpype/hosts/flame/api/__init__.py
index 97f83ccf07..561aaab3de 100644
--- a/openpype/hosts/flame/api/__init__.py
+++ b/openpype/hosts/flame/api/__init__.py
@@ -29,7 +29,8 @@ from .lib import (
     get_frame_from_filename,
     get_padding_from_filename,
     maintained_object_duplication,
-    get_clip_segment
+    get_clip_segment,
+    get_batch_group_from_desktop
 )
 from .utils import (
     setup,
@@ -105,6 +106,7 @@ __all__ = [
     "get_padding_from_filename",
     "maintained_object_duplication",
     "get_clip_segment",
+    "get_batch_group_from_desktop",
 
     # pipeline
     "install",
diff --git a/openpype/hosts/flame/api/lib.py b/openpype/hosts/flame/api/lib.py
index 74d9e7607a..9a6b86209d 100644
--- a/openpype/hosts/flame/api/lib.py
+++ b/openpype/hosts/flame/api/lib.py
@@ -708,3 +708,12 @@ def get_clip_segment(flame_clip):
         raise ValueError("Clip `{}` has too many segments!".format(name))
 
     return segments[0]
+
+
+def get_batch_group_from_desktop(name):
+    project = get_current_project()
+    project_desktop = project.current_workspace.desktop
+
+    for bgroup in project_desktop.batch_groups:
+        if bgroup.name.get_value() == name:
+            return bgroup
diff --git a/openpype/hosts/flame/plugins/publish/integrate_batch_group.py b/openpype/hosts/flame/plugins/publish/integrate_batch_group.py
index 0a21d6ca2d..3a8173791a 100644
--- a/openpype/hosts/flame/plugins/publish/integrate_batch_group.py
+++ b/openpype/hosts/flame/plugins/publish/integrate_batch_group.py
@@ -43,13 +43,25 @@ class IntegrateBatchGroup(pyblish.api.InstancePlugin):
             self.log.debug(
                 "__ batch_data: {}".format(pformat(batch_data)))
 
-            # create batch with utils
-            opfapi.create_batch(
-                batchgroup_name,
-                frame_start,
-                frame_duration,
-                **batch_data
-            )
+            # check if the batch group already exists
+            bgroup = opfapi.get_batch_group_from_desktop(batchgroup_name)
+
+            if not bgroup:
+                self.log.info(
+                    "Creating new batch group: {}".format(batchgroup_name))
+                # create batch with utils
+                opfapi.create_batch(
+                    batchgroup_name,
+                    frame_start,
+                    frame_duration,
+                    **batch_data
+                )
+            else:
+                self.log.info(
+                    "Updating batch group: {}".format(batchgroup_name))
+                # update already created batch group
+                bgroup.start_frame = frame_start
+                bgroup.duration = frame_duration
 
     def _get_write_prefs(self, instance, task_data):
         anatomy_data = instance.data["anatomyData"]