stash

stash/config/scrapers/community/GayWire/GayWire.py

@@ -0,0 +1,89 @@
+import json
+import sys
+from requests import head
+from typing import Any
+from py_common import log
+from py_common.util import replace_all, replace_at
+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 = {
+    "Its Gonna Hurt": "It's Gonna Hurt",
+    "Poundhisass": "Pound His Ass",
+}
+
+
+def redirect(url: str) -> str:
+    if not url or "gaywire.com/scene/" not in url:
+        return url
+    if (res := head(url)) and (redirect := res.headers.get("Location", url)):
+        return redirect if not redirect.endswith("404") else url
+    return url
+
+
+def gaywire(obj: Any, _) -> Any:
+    if obj is None:
+        return None
+
+    # API returns Gay Wire substudios as bangbros.com
+    fixed = replace_all(
+        obj,
+        "url",
+        lambda x: x.replace("www.bangbros.com", "gaywire.com"),
+    )
+
+    # Rename certain studios according to the map
+    fixed = replace_at(
+        fixed, "studio", "name", replacement=lambda x: studio_map.get(x, x)
+    )
+
+    fixed = replace_at(
+        fixed, "studio", "parent", "name", replacement=lambda x: "Gay Wire"
+    )
+
+    return fixed
+
+
+if __name__ == "__main__":
+    domains = ["gaywire", "guyselector"]
+    op, args = scraper_args()
+    result = None
+
+    match op, args:
+        case "gallery-by-url" | "gallery-by-fragment", {"url": url} if url:
+            url = redirect(url)
+            result = gallery_from_url(url, postprocess=gaywire)
+        case "scene-by-url", {"url": url} if url:
+            url = redirect(url)
+            result = scene_from_url(url, postprocess=gaywire)
+        case "scene-by-name", {"name": name} if name:
+            result = scene_search(name, search_domains=domains, postprocess=gaywire)
+        case "scene-by-fragment" | "scene-by-query-fragment", args:
+            args = replace_all(args, "url", redirect)
+            result = scene_from_fragment(
+                args, search_domains=domains, postprocess=gaywire
+            )
+        case "performer-by-url", {"url": url}:
+            url = redirect(url)
+            result = performer_from_url(url, postprocess=gaywire)
+        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=gaywire)
+        case "movie-by-url", {"url": url} if url:
+            url = redirect(url)
+            result = movie_from_url(url, postprocess=gaywire)
+        case _:
+            log.error(f"Operation: {op}, arguments: {json.dumps(args)}")
+            sys.exit(1)
+
+    print(json.dumps(result))

stash/config/scrapers/community/GayWire/GayWire.yml

@@ -0,0 +1,76 @@
+name: Gay Wire
+# requires: py_common, AyloAPI
+# scrapes: Bait Bus, Bareback Attack, Bareback Casting, ExBF, Guy Selector, Haze Him, It's Gonna Hurt, Out In Public, Pound His Ass, Rub Him, Sausage Party, Thug Hunter, UngloryHole, Urban Invasion
+galleryByURL:
+  - action: script
+    url:
+      - gaywire.com/video
+      - guyselector.com/scene
+    script:
+      - python
+      - GayWire.py
+      - gallery-by-url
+galleryByFragment:
+  action: script
+  script:
+    - python
+    - GayWire.py
+    - gallery-by-fragment
+sceneByURL:
+  - action: script
+    url:
+      - gaywire.com/video
+      - guyselector.com/scene
+    script:
+      - python
+      - GayWire.py
+      - scene-by-url
+sceneByFragment:
+  action: script
+  script:
+    - python
+    - GayWire.py
+    - scene-by-fragment
+sceneByName:
+  action: script
+  script:
+    - python
+    - GayWire.py
+    - scene-by-name
+sceneByQueryFragment:
+  action: script
+  script:
+    - python
+    - GayWire.py
+    - scene-by-query-fragment
+performerByName:
+  action: script
+  script:
+    - python
+    - GayWire.py
+    - performer-by-name
+performerByURL:
+  - action: script
+    url:
+      - gaywire.com/model/
+      - guyselector.com/model/
+    script:
+      - python
+      - GayWire.py
+      - performer-by-url
+performerByFragment:
+  action: script
+  script:
+    - python
+    - GayWire.py
+    - performer-by-fragment
+movieByURL:
+  - action: script
+    url:
+      - gaywire.com/movie
+      - guyselector.com/movie
+    script:
+      - python
+      - GayWire.py
+      - movie-by-url
+# Last Updated January 14, 2024

stash/config/scrapers/community/GayWire/manifest

@@ -0,0 +1,10 @@
+id: GayWire
+name: Gay Wire
+metadata: {}
+version: 5f64fd8
+date: "2024-02-20 10:52:59"
+requires: []
+source_repository: https://stashapp.github.io/CommunityScrapers/stable/index.yml
+files:
+- GayWire.py
+- GayWire.yml