Add runtime to GraphQL query

Author: Electronic-Mango

src/simplejustwatchapi/query.py

@@ -46,6 +46,7 @@
       fullPath
       originalReleaseYear
       originalReleaseDate
+      runtime
       genres {
         shortName
         __typename
@@ -114,6 +115,7 @@ class MediaEntry(NamedTuple):
     url: str
     release_year: int
     release_date: str
+    runtime_minutes: int
     genres: list[str]
     imdb_id: str | None
     poster: str | None
@@ -180,6 +182,7 @@ def _parse_entry(json: any) -> MediaEntry:
     url = _DETAILS_URL + content.get("fullPath")
     year = content.get("originalReleaseYear")
     date = content.get("originalReleaseDate")
+    runtime_minutes = content.get("runtime")
     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
@@ -195,6 +198,7 @@ def _parse_entry(json: any) -> MediaEntry:
         url,
         year,
         date,
+        runtime_minutes,
         genres,
         imdb_id,
         poster,

test/simplejustwatchapi/test_parser.py

@@ -12,6 +12,7 @@
         "fullPath": "/full/path/1/",
         "originalReleaseYear": 2000,
         "originalReleaseDate": "21-06-2000",
+        "runtime": 123,
         "genres": [{"shortName": "gen1"}, {"shortName": "gen2"}],
         "externalIds": {"imdbId": "imdbId1"},
         "posterUrl": "/poster/url/1.jpg",
@@ -57,6 +58,7 @@
     DETAILS_URL + "/full/path/1/",
     2000,
     "21-06-2000",
+    123,
     ["gen1", "gen2"],
     "imdbId1",
     IMAGES_URL + "/poster/url/1.jpg",
@@ -96,6 +98,7 @@
         "fullPath": "/full/path/2/",
         "originalReleaseYear": 2010,
         "originalReleaseDate": "11-01-2010",
+        "runtime": 456,
         "genres": [{"shortName": "gen2"}, {"shortName": "gen3"}],
         "externalIds": {"imdbId": "imdbId2"},
         "posterUrl": "/poster/url/2.jpg",
@@ -127,6 +130,7 @@
     DETAILS_URL + "/full/path/2/",
     2010,
     "11-01-2010",
+    456,
     ["gen2", "gen3"],
     "imdbId2",
     IMAGES_URL + "/poster/url/2.jpg",
@@ -154,6 +158,7 @@
         "fullPath": "/full/path/3/",
         "originalReleaseYear": 2020,
         "originalReleaseDate": "12-02-2020",
+        "runtime": 123,
         "posterUrl": "/poster/url/3.jpg",
     },
 }
@@ -165,6 +170,7 @@
     DETAILS_URL + "/full/path/3/",
     2020,
     "12-02-2020",
+    123,
     [],
     None,
     IMAGES_URL + "/poster/url/3.jpg",

test/simplejustwatchapi/test_request.py

@@ -39,6 +39,7 @@
       fullPath
       originalReleaseYear
       originalReleaseDate
+      runtime
       genres {
         shortName
         __typename