stigg-api-client-v2 0.579.8__py3-none-any.whl → 5.9.0__py3-none-any.whl

stigg/_edge_utils.py

@@ -0,0 +1,58 @@
+from typing import Tuple, Dict
+
+from stigg.generated import FetchEntitlementsQuery, GetPaywallInput, FetchEntitlementQuery, GetActiveSubscriptionsInput
+
+
+def build_get_entitlement_data(edge_url: str, query: FetchEntitlementQuery) -> Tuple[str, Dict]:
+  url = f"{edge_url}/v1/c/{query.customer_id}/entitlements.json"
+  params = {}
+  if query.resource_id is not None:
+    params["resourceId"] = query.resource_id
+
+  if query.feature_id is not None:
+    params["featureId"] = query.feature_id
+
+  if query.options is not None and query.options.requested_usage is not None:
+    params['requestedUsage'] = query.options.requested_usage
+
+  return url, params
+
+
+def build_get_entitlements_data(edge_url: str, query: FetchEntitlementsQuery) -> Tuple[str, Dict]:
+    url = f"{edge_url}/v1/c/{query.customer_id}/entitlements.json"
+    params = {}
+    if query.resource_id is not None:
+        params["resourceId"] = query.resource_id
+
+    return url, params
+
+
+def build_get_paywall_data(edge_url: str, _input: GetPaywallInput) -> Tuple[str, Dict]:
+    if _input.product_id is not None:
+        url = f"{edge_url}/v1/p/{_input.product_id}/paywall.json"
+    else:
+        url = f"{edge_url}/v1/paywall.json"
+
+    params = {}
+    if _input.billing_country_code is not None:
+        params["billingCountryCode"] = _input.billing_country_code
+
+    if _input.fetch_all_countries_prices is not None:
+        params["fetchAllCountriesPrices"] = _input.fetch_all_countries_prices
+
+    if _input.include_hidden_plans is not None:
+        params["includeHiddenPlans"] = _input.include_hidden_plans
+
+    return url, params
+
+def build_get_subscriptions_data(edge_url: str, _input: GetActiveSubscriptionsInput) -> Tuple[str, Dict]:
+    url = f"{edge_url}/v1/c/{_input.customer_id}/subscriptions.json"
+
+    params = {}
+    if _input.resource_id is not None:
+        params["resourceId"] = _input.resource_id
+
+    if _input.resource_ids is not None:
+        params["resourceIds"] = _input.resource_ids
+
+    return url, params

stigg/client.py

@@ -1,3 +1,5 @@
+from typing import Union, Optional, Dict, Any
+
 import httpx
 from httpx import Timeout
 from tenacity import retry, retry_if_exception_type, stop_after_attempt
@@ -5,8 +7,9 @@ from tenacity import retry, retry_if_exception_type, stop_after_attempt
 from stigg.generated.client import Client
 from stigg.generated.async_client import AsyncClient
 from stigg.generated import FetchEntitlementsQuery, GetEntitlements, GraphQLClientHttpError, GetPaywallInput, \
-    GetPaywall, GraphQlClientInvalidResponseError
-from stigg.edge_utils import build_get_entitlements_data, build_get_paywall_data
+  GetPaywall, GraphQLClientInvalidResponseError, FetchEntitlementQuery, GetEntitlement, UsageMeasurementCreateInput, \
+  ReportUsage, ReportUsageInput, GetActiveSubscriptionsInput, GetActiveSubscriptionsList
+from stigg._edge_utils import build_get_entitlements_data, build_get_paywall_data, build_get_entitlement_data, build_get_subscriptions_data
 
 PRODUCTION_API_URL = "https://api.stigg.io/graphql"
 PRODUCTION_EDGE_API_URL = "https://edge.api.stigg.io"
@@ -18,12 +21,13 @@ HTTP_TRANSPORT_RETRY_COUNT = 5
 
 RETRY_KWARGS = {
     "retry": (
-        retry_if_exception_type(GraphQLClientHttpError) | retry_if_exception_type(GraphQlClientInvalidResponseError)
+        retry_if_exception_type(GraphQLClientHttpError) | retry_if_exception_type(GraphQLClientInvalidResponseError)
     ),
     "stop": stop_after_attempt(STIGG_REQUESTS_RETRY_COUNT),
     "reraise": True,
 }
 
+X_GRAPHQL_OPERATION_NAME = 'x-graphql-operation-name'
 
 class StiggClient(Client):
     def __init__(self, enable_edge: bool, edge_api_url: str, *args, **kwargs):
@@ -32,29 +36,64 @@ class StiggClient(Client):
 
         super().__init__(*args, **kwargs)
 
+    def execute(
+        self,
+        query: str,
+        operation_name: Optional[str] = None,
+        variables: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ) -> httpx.Response:
+      headers = kwargs.get("headers", {})
+      headers[X_GRAPHQL_OPERATION_NAME] = operation_name
+      kwargs['headers'] = headers
+
+      return super().execute(query, operation_name, variables, **kwargs)
+
     @retry(**RETRY_KWARGS)
     def get_data(self,  *args, **kwargs):
         return super().get_data(*args, **kwargs)
 
-    def get_entitlements(self, query: FetchEntitlementsQuery) -> GetEntitlements:
+    def get_entitlement(self, input: FetchEntitlementQuery, **kwargs: Any) -> GetEntitlement:
         if self.enable_edge is False:
-            return super().get_entitlements(query)
+            return super().get_entitlement(input, **kwargs)
 
-        url, params = build_get_entitlements_data(self.edge_api_url, query)
+        url, params = build_get_entitlement_data(self.edge_api_url, input)
+
+        response = self.http_client.get(url=url, params=params, **kwargs)
+        data = self.get_data(response)
+        return GetEntitlement.model_validate(data)
+
+    def get_entitlements(self, query: FetchEntitlementsQuery, **kwargs: Any) -> GetEntitlements:
+        if self.enable_edge is False:
+            return super().get_entitlements(query, **kwargs)
 
-        response = self.http_client.get(url=url, params=params)
+        url, params = build_get_entitlements_data(self.edge_api_url, query)
+        response = self.http_client.get(url=url, params=params, **kwargs)
         data = self.get_data(response)
-        return GetEntitlements.parse_obj(data)
+        return GetEntitlements.model_validate(data)
 
-    def get_paywall(self, input: GetPaywallInput) -> GetPaywall:
+    def get_active_subscriptions_list(self, input: GetActiveSubscriptionsInput, **kwargs: Any) -> GetActiveSubscriptionsList:
         if self.enable_edge is False:
-            return super().get_paywall(input)
+            return super().get_active_subscriptions_list(input, **kwargs)
+
+        url, params = build_get_subscriptions_data(self.edge_api_url, input)
+
+        response = self.http_client.get(url=url, params=params, **kwargs)
+        data = self.get_data(response)
+        return GetActiveSubscriptionsList.model_validate(data)
+
+    def get_paywall(self, input: GetPaywallInput, **kwargs: Any) -> GetPaywall:
+        if input.customer_id is not None or self.enable_edge is False:
+            return super().get_paywall(input, **kwargs)
 
         url, params = build_get_paywall_data(self.edge_api_url, input)
 
-        response = self.http_client.get(url=url, params=params)
+        response = self.http_client.get(url=url, params=params, **kwargs)
         data = self.get_data(response)
-        return GetPaywall.parse_obj(data)
+        return GetPaywall.model_validate(data)
+
+    def report_usage(self, input: Union[UsageMeasurementCreateInput, ReportUsageInput], **kwargs: Any) -> ReportUsage:
+      return super().report_usage(input, **kwargs)
 
 
 class AsyncStiggClient(AsyncClient):
@@ -64,33 +103,70 @@ class AsyncStiggClient(AsyncClient):
 
         super().__init__(*args, **kwargs)
 
+    def execute(
+        self,
+        query: str,
+        operation_name: Optional[str] = None,
+        variables: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ) -> httpx.Response:
+      headers = kwargs.get("headers", {})
+      headers[X_GRAPHQL_OPERATION_NAME] = operation_name
+      kwargs['headers'] = headers
+
+      return super().execute(query, operation_name, variables, **kwargs)
+
+
     @retry(**RETRY_KWARGS)
     def get_data(self,  *args, **kwargs):
         return super().get_data(*args, **kwargs)
 
-    async def get_entitlements(self, query: FetchEntitlementsQuery) -> GetEntitlements:
+    async def get_entitlement(self, input: FetchEntitlementQuery, **kwargs: Any) -> GetEntitlement:
         if self.enable_edge is False:
-            return await super().get_entitlements(query)
+            return await super().get_entitlement(input, **kwargs)
+
+        url, params = build_get_entitlement_data(self.edge_api_url, input)
+
+        response = await self.http_client.get(url=url, params=params, **kwargs)
+        data = self.get_data(response)
+        return GetEntitlement.model_validate(data)
+
+    async def get_entitlements(self, query: FetchEntitlementsQuery, **kwargs: Any) -> GetEntitlements:
+        if self.enable_edge is False:
+            return await super().get_entitlements(query, **kwargs)
 
         url, params = build_get_entitlements_data(self.edge_api_url, query)
 
-        response = await self.http_client.get(url=url, params=params)
+        response = await self.http_client.get(url=url, params=params, **kwargs)
         data = self.get_data(response)
-        return GetEntitlements.parse_obj(data)
+        return GetEntitlements.model_validate(data)
 
-    async def get_paywall(self, input: GetPaywallInput) -> GetPaywall:
+    async def get_active_subscriptions_list(self, input: GetActiveSubscriptionsInput, **kwargs: Any) -> GetActiveSubscriptionsList:
         if self.enable_edge is False:
-            return await super().get_paywall(input)
+            return await super().get_active_subscriptions_list(input, **kwargs)
+
+        url, params = build_get_subscriptions_data(self.edge_api_url, input)
+
+        response = await self.http_client.get(url=url, params=params, **kwargs)
+        data = self.get_data(response)
+        return GetActiveSubscriptionsList.model_validate(data)
+
+    async def get_paywall(self, input: GetPaywallInput, **kwargs: Any) -> GetPaywall:
+        if input.customer_id is not None or self.enable_edge is False:
+            return await super().get_paywall(input, **kwargs)
 
         url, params = build_get_paywall_data(self.edge_api_url, input)
 
-        response = await self.http_client.get(url=url, params=params)
+        response = await self.http_client.get(url=url, params=params, **kwargs)
         data = self.get_data(response)
-        return GetPaywall.parse_obj(data)
+        return GetPaywall.model_validate(data)
+
+    async def report_usage(self, input: Union[UsageMeasurementCreateInput, ReportUsageInput], **kwargs: Any) -> ReportUsage:
+        return await super().report_usage(input, **kwargs)
 
 
 def get_headers(api_key: str):
-    return {"X-API-KEY": api_key, "Content-Type": "application/json"}
+    return {"X-API-KEY": api_key, "X-API-VERSION": "1", "Content-Type": "application/json"}
 
 
 class Stigg: