stash

stash/config/scrapers/community/WhisparrWDTV/WhisparrWDTV.py

@@ -0,0 +1,84 @@
+import sys
+import pathlib
+
+import mimetypes
+import base64
+
+import json
+import xml.etree.ElementTree as ET
+
+import py_common.graphql as graphql
+import py_common.log as log
+
+"""  
+This script parses WDTV xml metadata files. 
+The .xml file must be in the same directory as the video file and must be named exactly alike.
+
+Code borrowed from the kodi nfo scraper (in https://github.com/stashapp/CommunityScrapers/pull/689)
+It was found the .nfo files exported from Whisparr, did not contain all details required.
+Using the WDTV format instead had all information. 
+
+The intention is not to be a generic WDTV metadata parser, but one that specifically parses WDTV metadata from Whisparr. Based on version v2.0.0.168. This simplifies the integration of Whisparr and Stash.
+"""
+def query_xml(path, title):
+    res = {"title": title}
+    try:        
+        tree = ET.parse(path)
+    except Exception as e:
+        log.error(f'xml parsing failed:{e}')
+        print(json.dumps(res))
+        exit(1)
+    
+    if title == tree.find("episode_name").text:
+        log.info("Exact match found for " + title)
+    else:
+        log.info("No exact match found for " + title + ". Matching with " + tree.find("title").text + "!")
+    
+    # Extract matadata from xml
+    if tree.find("episode_name") != None:
+        res["title"] = tree.find("episode_name").text
+    
+    if tree.find("overview") != None:
+        res["details"] = tree.find("overview").text
+    
+    if tree.find("firstaired") != None:
+        res["date"] = tree.find("firstaired").text
+
+    # This is based on how my version of Whisparr (v2.0.0.168) output the WDTV .xml
+    # It seperated actors by " / " 
+    # then for some reason had duplicated the name seperated by " - "
+    if tree.find("actor") != None and tree.find("actor").text:
+        res["performers"] = []
+        for actor in tree.find("actor").text.split(" / "):
+            res["performers"].append({"name": actor.split(" - ")[0]})
+    
+    if tree.find("series_name") != None:
+        res["studio"] = {"name":tree.find("series_name").text}
+    
+    return res
+
+if sys.argv[1] == "query":
+    fragment = json.loads(sys.stdin.read())
+    s_id = fragment.get("id")
+    if not s_id:
+        log.error(f"No ID found")
+        sys.exit(1)
+    
+    # Assume that .xml is named exactly alike the video file and is at the same location
+    # Query graphQL for the file path
+    scene = graphql.getScene(s_id)
+    if scene:
+        scene_path = scene.get("path")
+        if scene_path:
+            p = pathlib.Path(scene_path)
+            
+            res = {"title": fragment["title"]}
+            
+            f = p.with_suffix(".xml")
+            if f.is_file():
+                res = query_xml(f, fragment["title"])
+            else:
+                log.info(f"No xml files found for the scene: {p}")
+            
+            print(json.dumps(res))
+            exit(0)