stash

stash/config/scrapers/community/AssumeThePositionStudios/AssumeThePositionStudios.py

@@ -0,0 +1,118 @@
+import json
+import sys
+import requests
+import re
+
+try:
+    import py_common.graphql as graphql
+    import py_common.log as log
+except ModuleNotFoundError:
+    print(
+        "You need to download the folder 'py_common' from the community repo! (CommunityScrapers/tree/master/scrapers/py_common)",
+        file=sys.stderr,
+    )
+    sys.exit()
+
+
+def debugPrint(t):
+    log.debug(t + "\n");
+
+def scrape_scene(url):
+    query = """
+query scrapeSceneURL($url: String!) {
+    scrapeSceneURL(url: $url) {
+        title
+        details
+        code
+        date
+        image
+        urls
+        studio {
+            name
+            url
+            image
+            parent {
+                name
+                url
+                image
+            }
+        }
+        tags {
+            name
+        }
+        performers {
+            aliases
+            birthdate
+            career_length
+            country
+            death_date
+            details
+            ethnicity
+            eye_color
+            fake_tits
+            gender
+            hair_color
+            height
+            instagram
+            images
+            measurements
+            name
+            piercings
+            tags {
+                name
+            }
+            tattoos
+            twitter
+            url
+            weight
+        }
+    }
+}"""
+
+    variables = {"url": url}
+    result = graphql.callGraphQL(query, variables)
+    log.debug(f"result {result}")
+    res = result["scrapeSceneURL"]
+    res["url"] = url
+    return res
+
+
+fragment = json.loads(sys.stdin.read())
+debugPrint(json.dumps(fragment))
+title = fragment.get("title")
+
+debugPrint("title is " + title)
+if m := re.match(r"(\d+)_[a-z]+_[a-z]+_\d+_([A-Z]+).*", title):
+    #Turn the content id from the filename into a scene id
+    contentId = int(m.group(1))
+    siteCode = m.group(2)
+    debugPrint("Site code is " + siteCode)
+    if siteCode.casefold() == "WBP".casefold():
+        apiUrl = "https://www.worstbehaviorproductions.com/api/site/21/updates/0"
+        sceneUrlFragment = "https://worstbehaviorproductions.com/trailer?updateId="
+    elif siteCode.casefold() == "ATP".casefold():
+        apiUrl = "https://assumethepositionstudios.com/api/site/13/updates/0"
+        sceneUrlFragment = "https://assumethepositionstudios.com/trailer?updateId="
+    else:
+        debugPrint("Unknown site code " + siteCode)
+        sys.exit();
+
+    debugPrint("Searching for " + str(contentId) + " on " + apiUrl)
+    page = requests.get(apiUrl)
+    updates = json.loads(page.content)
+    matches = [scene for scene in updates["data"] if scene["scene"]["id"] == contentId]
+    if len(matches) != 1:
+        debugPrint("Couldn't find match for " + str(contentId) + " found " + str(len(matches)) + " matches")
+        sys.exit()
+    debugPrint("Match is " + json.dumps(matches[0]))
+    sceneId = matches[0]["id"]
+
+    #Build a scene url from the sceneId
+    sceneUrl = sceneUrlFragment + str(sceneId)
+    debugPrint("Scene URL is " + sceneUrl)
+
+    result = scrape_scene(sceneUrl)
+    print(json.dumps(result))
+else:
+    debugPrint("title didn't match")
+    print(json.dumps({}))

stash/config/scrapers/community/AssumeThePositionStudios/AssumeThePositionStudios.yml

@@ -0,0 +1,56 @@
+name: AssumeThePosition
+# requires: py_common
+sceneByURL:
+  - action: scrapeJson
+    url:
+      - assumethepositionstudios.com/trailer
+    scraper: sceneScraper
+    queryURL: "https://assumethepositionstudios.com/api/update/{url}/trailer/"
+    queryURLReplace:
+      url:
+        - regex: '.+/trailer\?updateId=(\d+).*'
+          with: "${1}"
+  - action: scrapeJson
+    url:
+      - worstbehaviorproductions.com/trailer
+    scraper: sceneScraper
+    queryURL: "https://worstbehaviorproductions.com/api/update/{url}/trailer/"
+    queryURLReplace:
+      url:
+        - regex: '.+/trailer\?updateId=(\d+).*'
+          with: "${1}"
+sceneByFragment:
+  action: script
+  script:
+    - python
+    - AssumeThePositionStudios.py
+    - sceneByFragment
+jsonScrapers:
+  sceneScraper:
+    scene:
+      Title: data.scene.title
+      Details:
+        selector: data.scene.description
+        postProcess:
+          - replace:
+              - regex: (.*?)[A-Z,/'\s]+\.?$
+                with: $1
+      Date: data.live_date
+      Image:
+        selector: "[data.producer.url,data.image]"
+        concat: "/"
+      Performers:
+        Name: data.models.#.name
+      Studio:
+        Name: data.producer.name
+      Tags:
+        Name:
+          selector: data.scene.description
+          postProcess:
+            - replace:
+                - regex: (.*?)([A-Z,/'\s]+)\.?$
+                  with: $2
+                - regex: ',\s+'
+                  with: ","
+          split: ","
+# Last Updated January 2, 2023

stash/config/scrapers/community/AssumeThePositionStudios/manifest

@@ -0,0 +1,10 @@
+id: AssumeThePositionStudios
+name: AssumeThePosition
+metadata: {}
+version: 0f658ea
+date: "2024-01-12 15:32:58"
+requires: []
+source_repository: https://stashapp.github.io/CommunityScrapers/stable/index.yml
+files:
+- AssumeThePositionStudios.yml
+- AssumeThePositionStudios.py