[markerTagToScene] Add all tags of Scene Marker to the Scene

Author: IAmKontrast

plugins/markerTagToScene/markerTagToScene.py

@@ -4,6 +4,8 @@
 import requests
 import log
 
+ALL_TAGS = False
+
 START_TIME = time.time()
 FRAGMENT = json.loads(sys.stdin.read())
 
@@ -104,29 +106,34 @@ def graphql_setSceneTags(sceneID, tagIDs: list):
 def main():
 
     CONTEXT = FRAGMENT['args']['hookContext']['input']
-
-    primaryTagID = CONTEXT['primary_tag_id']
     sceneID = CONTEXT['scene_id']
-
-    if not primaryTagID or not sceneID:
+    if not sceneID:
         return
+    
+    prevSceneTags = graphql_getSceneTags(sceneID)
+    prevSceneTagIDs = []
+    for prevSceneTag in prevSceneTags:
+        prevSceneTagIDs.append(prevSceneTag['id'])
 
-    sceneTags = graphql_getSceneTags(sceneID)
-    tagIDs = []
+    nextTagIDs = []
 
-    for sceneTags in sceneTags:
-        tagID = sceneTags['id'];
-        if tagID == primaryTagID:
-            log.LogDebug("Primary tag already exists on scene")
-            return
+    # Primary tag
+    primaryTagID = CONTEXT['primary_tag_id']
+    if primaryTagID is not None:
+        nextTagIDs = set(prevSceneTagIDs + [primaryTagID])
 
-        tagIDs.append(tagID);
+    # All tags
+    if ALL_TAGS:
+        tagIDs = CONTEXT['tag_ids']
+        if tagIDs is not None:
+            nextTagIDs = set(set(nextTagIDs) | set(prevSceneTagIDs) | set(tagIDs))
 
-    # set the tag on the scene if not present
-    tagIDs.append(primaryTagID);
+    if len(prevSceneTagIDs) >= len(nextTagIDs):
+        log.LogDebug("No new tag added")
+        return
 
-    graphql_setSceneTags(sceneID, tagIDs);
-    log.LogDebug("Added primary tag " + primaryTagID + " to scene " + sceneID)
+    graphql_setSceneTags(sceneID, list(nextTagIDs))
+    log.LogDebug("Added new tags to scene " + sceneID)
 
 
 

plugins/markerTagToScene/markerTagToScene.yml

@@ -1,14 +1,14 @@
 name: Scene Marker Tags to Scene
-description: Adds primary tag of Scene Marker to the Scene on marker create/update.
+description: Adds primary tag and opt-in all tags of Scene Marker to the Scene on marker create/update.
 url: https://github.com/stashapp/CommunityScripts
-version: 2.0.0
+version: 2.1.0
 exec:
   - python
   - "{pluginDir}/markerTagToScene.py"
 interface: raw
 hooks:
   - name: Update scene with scene marker tag
-    description: Adds primary tag of Scene Marker to the Scene on marker create/update.
+    description: Adds primary tag and opt-in all tags of Scene Marker to the Scene on marker create/update.
     triggeredBy: 
       - SceneMarker.Create.Post
       - SceneMarker.Update.Post