add tmdbId

Author: stu247

src/simplejustwatchapi/query.py

@@ -92,6 +92,7 @@
     }
     externalIds {
       imdbId
+      tmdbId
       __typename
     }
     posterUrl(profile: $profile, format: $formatPoster)
@@ -259,6 +260,9 @@ class MediaEntry(NamedTuple):
     imdb_id: str | None
     """ID of this entry in IMDB."""
 
+    tmdb_id: str | None
+    """ID of this entry in TMDB."""
+
     poster: str | None
     """URL to poster for this ID."""
 
@@ -469,6 +473,7 @@ def _parse_entry(json: any) -> MediaEntry:
     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
+    tmdb_id = external_ids.get("tmdbId") if external_ids else None
     poster_url_field = content.get("posterUrl")
     poster = _IMAGES_URL + poster_url_field if poster_url_field else None
     backdrops = [_IMAGES_URL + bd.get("backdropUrl") for bd in content.get("backdrops", []) if bd]
@@ -485,6 +490,7 @@ def _parse_entry(json: any) -> MediaEntry:
         short_description,
         genres,
         imdb_id,
+        tmdb_id,
         poster,
         backdrops,
         offers,

test/simplejustwatchapi/test_parser.py

@@ -24,7 +24,7 @@
         "runtime": 123,
         "shortDescription": "Movie 1 description",
         "genres": [{"shortName": "gen1"}, {"shortName": "gen2"}],
-        "externalIds": {"imdbId": "imdbId1"},
+        "externalIds": {"imdbId": "imdbId1", "tmdbId": "tmdbId1"},
         "posterUrl": "/poster/url/1.jpg",
         "backdrops": [
             {"backdropUrl": "/back/drop/url/1.jpg"},
@@ -96,6 +96,7 @@
     "Movie 1 description",
     ["gen1", "gen2"],
     "imdbId1",
+    "tmdbId1",
     IMAGES_URL + "/poster/url/1.jpg",
     [IMAGES_URL + "/back/drop/url/1.jpg", IMAGES_URL + "/back/drop/url/2.jpg"],
     [
@@ -164,7 +165,7 @@
         "runtime": 456,
         "shortDescription": "TV show 2 description",
         "genres": [{"shortName": "gen2"}, {"shortName": "gen3"}],
-        "externalIds": {"imdbId": "imdbId2"},
+        "externalIds": {"imdbId": "imdbId1", "tmdbId": "tmdbId1"},
         "posterUrl": "/poster/url/2.jpg",
         "backdrops": [
             {"backdropUrl": "/back/drop/url/3.jpg"},
@@ -210,6 +211,7 @@
     "TV show 2 description",
     ["gen2", "gen3"],
     "imdbId2",
+    "tmdbId2",
     IMAGES_URL + "/poster/url/2.jpg",
     [IMAGES_URL + "/back/drop/url/3.jpg"],
     [
@@ -265,6 +267,7 @@
     None,
     [],
     None,
+    None,
     IMAGES_URL + "/poster/url/3.jpg",
     [],
     [],

test/simplejustwatchapi/test_request.py

@@ -92,6 +92,7 @@
     }
     externalIds {
       imdbId
+      tmdbId
       __typename
     }
     posterUrl(profile: $profile, format: $formatPoster)