Add "monetization_types" to OfferPackage

Return all available monetization types for the given provider.

Author: Electronic-Mango

src/simplejustwatchapi/graphql.py

@@ -307,6 +307,7 @@
   technicalName
   shortName
   slug
+  monetizationTypes
   icon(profile: S100, format: $formatOfferIcon)
   __typename
 }

src/simplejustwatchapi/query.py

@@ -711,5 +711,14 @@ def _parse_package(json: Any) -> OfferPackage:
     name = json.get("clearName")
     technical_name = json.get("technicalName")
     short_name = json.get("shortName")
+    monetization_types = json.get("monetizationTypes", [])
     icon = _IMAGES_URL + icon if (icon := json.get("icon")) else ""
-    return OfferPackage(platform_id, package_id, name, technical_name, short_name, icon)
+    return OfferPackage(
+        platform_id,
+        package_id,
+        name,
+        technical_name,
+        short_name,
+        monetization_types,
+        icon,
+    )

src/simplejustwatchapi/tuples.py

@@ -30,6 +30,8 @@ class OfferPackage(NamedTuple):
         technical_name (str): Technical name of the platform,
             usually all lowercase with no whitespaces.
         short_name (str): 3-letter provider name.
+        monetization_types (list[str]): List of monetization types available for this
+            provider (e.g., `ADS`, `FLATRATE` (streaming), `RENT`).
         icon (str): Platform icon URL.
 
     """
@@ -39,6 +41,7 @@ class OfferPackage(NamedTuple):
     name: str
     technical_name: str
     short_name: str
+    monetization_types: list[str]
     icon: str
 
 

test/simplejustwatchapi/test_graphql.py

@@ -304,6 +304,7 @@
   technicalName
   shortName
   slug
+  monetizationTypes
   icon(profile: S100, format: $formatOfferIcon)
   __typename
 }

test/simplejustwatchapi/test_parser.py

@@ -29,6 +29,7 @@
     "clearName": "Service 1",
     "technicalName": "service1",
     "shortName": "sr1",
+    "monetizationTypes": ["MON_TYPE_1", "MON_TYPE_2"],
     "icon": "/icon/url/service1",
 }
 PARSED_PACKAGE_1 = OfferPackage(
@@ -37,6 +38,7 @@
     "Service 1",
     "service1",
     "sr1",
+    ["MON_TYPE_1", "MON_TYPE_2"],
     IMAGES_URL + "/icon/url/service1",
 )
 
@@ -46,6 +48,7 @@
     "clearName": "Service 2",
     "technicalName": "service2",
     "shortName": "sr2",
+    "monetizationTypes": ["MON_TYPE_3"],
     "icon": "/icon/url/service2",
 }
 PARSED_PACKAGE_2 = OfferPackage(
@@ -54,6 +57,7 @@
     "Service 2",
     "service2",
     "sr2",
+    ["MON_TYPE_3"],
     IMAGES_URL + "/icon/url/service2",
 )
 
@@ -71,6 +75,7 @@
     "Service 3",
     "service3",
     "sr3",
+    [],
     IMAGES_URL + "/icon/url/service3",
 )