Add source code

Author: Electronic-Mango

src/simplejustwatchpythonapi/justwatch.py

@@ -0,0 +1,28 @@
+from httpx import post
+
+from simplejustwatchpythonapi.parser import parse_search_response, MediaEntry
+from simplejustwatchpythonapi.requests import prepare_search_request
+
+_GRAPHQL_API_URL = "https://apis.justwatch.com/graphql"
+
+
+def search(
+    title: str, country: str = "US", language: str = "en", count: int = 4, best_only: bool = True
+) -> list[MediaEntry]:
+    """Search JustWatch for given title.
+    Returns a list of entries up to count.
+
+    Args:
+        title: title to search
+        country: country to search for offers, "US" by default
+        language: language of responses, "en" by default
+        count: how many responses should be returned
+        best_only: return only best offers if True, return all offers if False
+
+    Returns:
+        List of MediaEntry NamedTuples parsed from JustWatch response
+    """
+    request = prepare_search_request(title, country, language, count, best_only)
+    response = post(_GRAPHQL_API_URL, json=request)
+    response.raise_for_status()
+    return parse_search_response(response.json())

src/simplejustwatchpythonapi/parser.py

@@ -0,0 +1,99 @@
+from typing import NamedTuple
+
+_IMAGES_URL = "https://images.justwatch.com"
+
+
+class Offer(NamedTuple):
+    monetization_type: str
+    presentation_type: str
+    url: str
+    price_string: str | None
+    price_value: float | None
+    price_currency: str
+    name: str
+    technical_name: str
+    icon: str
+
+
+class MediaEntry(NamedTuple):
+    entry_id: str
+    object_id: int
+    object_type: str
+    title: str
+    url: str
+    release_year: int
+    release_date: str
+    genres: list[str]
+    imdb_id: list[str]
+    poster: str
+    backdrops: list[str]
+    offers: list[Offer]
+
+
+def parse_search_response(json: dict) -> list[MediaEntry]:
+    """Parse response from search query from JustWatch GraphQL API.
+
+    Args:
+        json: JSON returned by JustWatch GraphQL API
+
+    Returns:
+        Parsed received JSON as a MediaEntry NamedTuple
+    """
+    nodes = json["data"]["popularTitles"]["edges"]
+    entries = [_parse_entry(node["node"]) for node in nodes]
+    return entries
+
+
+def _parse_entry(json: any) -> MediaEntry:
+    entry_id = json.get("id")
+    object_id = json.get("objectId")
+    object_type = json.get("objectType")
+    content = json["content"]
+    title = content.get("title")
+    url = _IMAGES_URL + content.get("fullPath")
+    year = content.get("originalReleaseYear")
+    date = content.get("originalReleaseDate")
+    genres = [node.get("shortName") for node in content.get("genres", []) if node]
+    external_ids = content.get("externalIds")
+    imdb_id = external_ids.get("imdbId") if external_ids else None
+    poster = content.get("posterUrl")
+    backdrops = [_IMAGES_URL + bd.get("backdropUrl") for bd in content.get("backdrops", []) if bd]
+    offers = [_parse_offer(offer) for offer in json.get("offers", []) if offer]
+    return MediaEntry(
+        entry_id,
+        object_id,
+        object_type,
+        title,
+        url,
+        year,
+        date,
+        genres,
+        imdb_id,
+        poster,
+        backdrops,
+        offers,
+    )
+
+
+def _parse_offer(json: any) -> Offer:
+    monetization_type = json.get("monetizationType")
+    presentation_type = json.get("presentationType")
+    url = json.get("standardWebURL")
+    price_string = json.get("retailPrice")
+    price_value = json.get("retailPriceValue")
+    price_currency = json.get("currency")
+    package = json["package"]
+    name = package.get("clearName")
+    technical_name = package.get("technicalName")
+    icon = _IMAGES_URL + package.get("icon")
+    return Offer(
+        monetization_type,
+        presentation_type,
+        url,
+        price_string,
+        price_value,
+        price_currency,
+        name,
+        technical_name,
+        icon,
+    )

src/simplejustwatchpythonapi/requests.py

@@ -0,0 +1,103 @@
+GRAPHQL_SEARCH_QUERY = """
+query GetSearchTitles(
+  $searchTitlesFilter: TitleFilter!,
+  $country: Country!,
+  $language: Language!,
+  $first: Int!,
+  $format: ImageFormat,
+  $profile: PosterProfile,
+  $backdropProfile: BackdropProfile,
+  $filter: OfferFilter!,
+) {
+  popularTitles(
+    country: $country
+    filter: $searchTitlesFilter
+    first: $first
+    sortBy: POPULAR
+    sortRandomSeed: 0
+  ) {
+    edges {
+      ...SearchTitleGraphql
+      __typename
+    }
+    __typename
+  }
+}
+
+fragment SearchTitleGraphql on PopularTitlesEdge {
+  node {
+    id
+    objectId
+    objectType
+    content(country: $country, language: $language) {
+      title
+      fullPath
+      originalReleaseYear
+      originalReleaseDate
+      genres {
+        shortName
+        __typename
+      }
+      externalIds {
+        imdbId
+        __typename
+      }
+      posterUrl(profile: $profile, format: $format)
+      backdrops(profile: $backdropProfile, format: $format) {
+        backdropUrl
+        __typename
+      }
+      __typename
+    }
+    offers(country: $country, platform: WEB, filter: $filter) {
+      monetizationType
+      presentationType
+      standardWebURL
+      retailPrice(language: $language)
+      retailPriceValue
+      currency
+      package {
+        id
+        packageId
+        clearName
+        technicalName
+        icon(profile: S100)
+        __typename
+      }
+      id
+      __typename
+    }
+    __typename
+  }
+  __typename
+}
+"""
+
+
+def prepare_search_request(title: str, county: str, lang: str, count: int, best_only: bool) -> dict:
+    """Prepare search request for JustWatch GraphQL API.
+
+    Args:
+        title: title to search
+        county: country to search for offers
+        lang: language of responses
+        count: how many responses should be returned
+        best_only: return only best offers if True, return all offers if False
+
+    Returns:
+        JSON/dict with GraphQL POST body
+    """
+    return {
+        "operationName": "GetSearchTitles",
+        "variables": {
+            "first": count,
+            "searchTitlesFilter": {"searchQuery": title},
+            "language": lang,
+            "country": county,
+            "format": "JPG",
+            "profile": "S718",
+            "backdropProfile": "S1920",
+            "filter": {"bestOnly": best_only},
+        },
+        "query": GRAPHQL_SEARCH_QUERY,
+    }