stash

stash/config/scrapers/community/MileHighMedia_Straight/MileHighMedia_Straight.py

@@ -0,0 +1,96 @@
+import json
+import sys
+from typing import Any
+from py_common import log
+from py_common.util import dig, replace_all
+from AyloAPI.scrape import (
+    gallery_from_url,
+    scraper_args,
+    scene_from_url,
+    scene_search,
+    scene_from_fragment,
+    performer_from_url,
+    performer_from_fragment,
+    performer_search,
+    movie_from_url,
+)
+
+studio_map = {
+    "dlf": "Dilfed",
+    "DogHouseDigital": "Doghouse Digital",
+    "LesbianOlderYounger": "Lesbian Older Younger",
+    "SweetHeartVideo": "Sweetheart Video",
+    "SweetSinner": "Sweet Sinner",
+    "RealityJunkies": "Reality Junkies",
+}
+
+
+def milehigh(obj: Any, _) -> Any:
+    fixed = replace_all(obj, "name", replacement=lambda x: studio_map.get(x, x))
+
+    replacement = None
+    match dig(fixed, "studio", "name"):
+        case "Dilfed":
+            replacement = "dilfed.com"
+        case "Doghouse Digital":
+            replacement = "doghousedigital.com"
+        case "Family Sinners":
+            replacement = "familysinners.com"
+        case "Milfed" | "Lesbian Older Younger":
+            replacement = "milfed.com"
+        case "Reality Junkies":
+            replacement = "realityjunkies.com"
+        case "Sweet Sinner":
+            replacement = "sweetsinner.com"
+        case "Sweetheart Video":
+            replacement = "sweetheartvideo.com"
+        case _:
+            replacement = "milehighmedia.com"
+
+    # Replace the studio name in all URLs: even if there's no specific studio,
+    # milehigh.com is wrong and needs to be replaced with milehighmedia.com
+    fixed = replace_all(fixed, "url", lambda x: x.replace("milehigh.com", replacement))
+
+    return fixed
+
+
+if __name__ == "__main__":
+    domains = [
+        "milehighmedia",
+        "dilfed",
+        "doghousedigital",
+        "familysinners",
+        "milfed",
+        "realityjunkies",
+        "sweetsinner",
+        "sweetheartvideo",
+    ]
+    op, args = scraper_args()
+    result = None
+
+    match op, args:
+        case "gallery-by-url" | "gallery-by-fragment", {"url": url} if url:
+            result = gallery_from_url(url, postprocess=milehigh)
+        case "scene-by-url", {"url": url} if url:
+            result = scene_from_url(url, postprocess=milehigh)
+        case "scene-by-name", {"name": name} if name:
+            result = scene_search(name, search_domains=domains, postprocess=milehigh)
+        case "scene-by-fragment" | "scene-by-query-fragment", args:
+            result = scene_from_fragment(
+                args, search_domains=domains, postprocess=milehigh
+            )
+        case "performer-by-url", {"url": url}:
+            result = performer_from_url(url, postprocess=milehigh)
+        case "performer-by-fragment", args:
+            result = performer_from_fragment(args)
+        case "performer-by-name", {"name": name} if name:
+            result = performer_search(
+                name, search_domains=domains, postprocess=milehigh
+            )
+        case "movie-by-url", {"url": url} if url:
+            result = movie_from_url(url, postprocess=milehigh)
+        case _:
+            log.error(f"Operation: {op}, arguments: {json.dumps(args)}")
+            sys.exit(1)
+
+    print(json.dumps(result))