From c67e672acaaa0ed5a459ef0e52a1cd81b6ee049d Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakub@orbi.tools>
Date: Thu, 6 Jan 2022 15:08:06 +0100
Subject: [PATCH] flame: improving testing plugin

adding maintained seqment selection functionality
---
 openpype/hosts/flame/api/lib.py               |  2 +-
 .../plugins/publish/collect_test_selection.py | 27 +++++++------------
 2 files changed, 11 insertions(+), 18 deletions(-)

diff --git a/openpype/hosts/flame/api/lib.py b/openpype/hosts/flame/api/lib.py
index 2d30390d21..e5642dd6f9 100644
--- a/openpype/hosts/flame/api/lib.py
+++ b/openpype/hosts/flame/api/lib.py
@@ -14,7 +14,7 @@ class ctx:
     # OpenPype marker workflow variables
     marker_name = "OpenPypeData"
     marker_duration = 0
-    marker_color = "red"
+    marker_color = "cyan"
     publish_default = False
     color_map = {
         "red": (1.0, 0.0, 0.0),
diff --git a/openpype/hosts/flame/plugins/publish/collect_test_selection.py b/openpype/hosts/flame/plugins/publish/collect_test_selection.py
index cd7355d1f5..875ef34a07 100644
--- a/openpype/hosts/flame/plugins/publish/collect_test_selection.py
+++ b/openpype/hosts/flame/plugins/publish/collect_test_selection.py
@@ -43,23 +43,16 @@ class CollectTestSelection(pyblish.api.ContextPlugin):
         self.log.info("Otio exported to: {}".format(export_path))
 
     def test_imprint_data(self, sequence):
-        # test segment markers
-        for ver in sequence.versions:
-            for track in ver.tracks:
-                if len(track.segments) == 0 and track.hidden:
+        with lib.maintained_segment_selection(sequence) as selected_segments:
+            for segment in selected_segments:
+                if str(segment.name)[1:-1] == "":
                     continue
 
-                for segment in track.segments:
-                    if str(segment.name)[1:-1] == "":
-                        continue
-                    if not segment.selected:
-                        continue
+                self.log.debug("Segment with OpenPypeData: {}".format(
+                    segment.name))
 
-                    self.log.debug("Segment with OpenPypeData: {}".format(
-                        segment.name))
-
-                    pipeline.imprint(segment, {
-                        'asset': segment.name.get_value(),
-                        'family': 'render',
-                        'subset': 'subsetMain'
-                    })
+                pipeline.imprint(segment, {
+                    'asset': segment.name.get_value(),
+                    'family': 'render',
+                    'subset': 'subsetMain'
+                })