97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
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))
|