stigg-api-client 0.508.0__py3-none-any.whl → 0.512.0__py3-none-any.whl

stigg/client.py

@@ -1,21 +1,90 @@
-from sgqlc.endpoint.requests import RequestsEndpoint
+import requests
+from typing import Dict, Callable
+
 from sgqlc.operation import Operation
+from sgqlc.endpoint.requests import RequestsEndpoint
+from requests.exceptions import HTTPError
+from tenacity import retry, retry_if_exception_type, stop_after_attempt
 
 from stigg.generated.operations import Operations
 
+PRODUCTION_API_URL = "https://api.stigg.io/graphql"
+PRODUCTION_EDGE_API_URL = "https://edge.api.stigg.io"
+
+STIGG_REQUESTS_RETRY_COUNT = 3
+
+RETRY_KWARGS = {
+    "retry": retry_if_exception_type(HTTPError),
+    "stop": stop_after_attempt(STIGG_REQUESTS_RETRY_COUNT),
+    "reraise": True,
+}
+
 
 class StiggClient:
-    def __init__(self, endpoint: RequestsEndpoint):
+    def __init__(self, endpoint: RequestsEndpoint, enable_edge: bool, edge_api_url: str):
         self._endpoint: RequestsEndpoint = endpoint
 
-    def request(self, operation: Operation, variables: dict, raw_response=False):
+        self._enable_edge = enable_edge
+        self._edge_url = edge_api_url
+
+        self._EDGE_QUERIES: Dict[Operation, Callable[[Dict], Dict]] = {
+            Operations.query.get_entitlements: self.get_entitlements,
+            Operations.query.get_paywall: self.get_paywall,
+        }
+
+    @retry(**RETRY_KWARGS)
+    def _get_entitlements(self, variables: Dict[str, Dict]) -> Dict:
+        params = variables.get("query")
+        if params is None:
+            raise ValueError('Variables do not include required key "query".')
+
+        customer_id = params.pop("customerId")
+
+        url = f"{self._edge_url}/v1/c/{customer_id}/entitlements.json"
+
+        data = requests.get(url, headers=self._endpoint.base_headers, params=params, timeout=self._endpoint.timeout)
+        data.raise_for_status()
+
+        return data.json()
+
+    def get_entitlements(self, variables: Dict[str, Dict]) -> Dict:
+        if self._enable_edge is False:
+            return self.request(Operations.query.get_entitlements, variables, raw_response=True)
+
+        return self._get_entitlements(variables)
+
+    @retry(**RETRY_KWARGS)
+    def _get_paywall(self,variables: Dict[str, Dict]) -> Dict:
+        params = variables.get("input")
+        if params is None:
+            raise ValueError('Variables do not include required key "input".')
+
+        if params.get("productId") is not None:
+            url = f"{self._edge_url}/v1/p/{params['productId']}/paywall.json"
+            params.pop("productId")
+        else:
+            url = f"{self._edge_url}/v1/paywall.json"
+
+        data = requests.get(url, headers=self._endpoint.base_headers, params=params, timeout=self._endpoint.timeout)
+        data.raise_for_status()
+
+        return data.json()
+
+    def get_paywall(self, variables: Dict[str, Dict]) -> Dict:
+        if self._enable_edge is False:
+            return self.request(Operations.query.get_paywall, variables, raw_response=True)
+
+        return self._get_paywall(variables)
+
+    @retry(**RETRY_KWARGS)
+    def request(self, operation: Operation, variables: Dict, raw_response: bool = False):
         data = self._endpoint(operation, variables)
         if raw_response:
             return data
 
-        # clean up data to so an exepction will be raised
+        # clean up data to so an exception will be raised
         errors = data.get('errors')
-        if errors:
+        if errors and data.get('data') is not None:
             data.pop('data')
 
         # interpret results into native Python objects
@@ -24,11 +93,13 @@ class StiggClient:
 
 class Stigg(Operations):
     @staticmethod
-    def create_client(api_key: str,
-                      api_url: str = 'https://api.stigg.io/graphql',
-                      request_timeout=30) -> StiggClient:
+    def create_client(
+        api_key: str,
+        api_url: str = PRODUCTION_API_URL,
+        enable_edge: bool = True,
+        edge_api_url: str = PRODUCTION_EDGE_API_URL,
+        request_timeout: int = 30,
+    ) -> StiggClient:
         headers = {'X-API-KEY': f'{api_key}'}
-        endpoint = RequestsEndpoint(url=api_url,
-                                    base_headers=headers,
-                                    timeout=request_timeout)
-        return StiggClient(endpoint)
+        endpoint = RequestsEndpoint(url=api_url, base_headers=headers, timeout=request_timeout)
+        return StiggClient(endpoint, enable_edge, edge_api_url)

stigg/generated/schema.py

@@ -2701,7 +2701,8 @@ class SubscriptionAddonSort(sgqlc.types.Input):
 
 class SubscriptionBillingInfo(sgqlc.types.Input):
     __schema__ = schema
-    __field_names__ = ('tax_rate_ids',)
+    __field_names__ = ('tax_percentage', 'tax_rate_ids')
+    tax_percentage = sgqlc.types.Field(Float, graphql_name='taxPercentage')
     tax_rate_ids = sgqlc.types.Field(sgqlc.types.list_of(sgqlc.types.non_null(String)), graphql_name='taxRateIds')
 
 

{stigg_api_client-0.508.0.dist-info → stigg_api_client-0.512.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stigg-api-client
-Version: 0.508.0
+Version: 0.512.0
 Summary: 
 Author: Stigg
 Author-email: support@stigg.io
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: graphql-core (>=3.1,<4.0)
 Requires-Dist: requests (>=2.28,<3.0)
 Requires-Dist: sgqlc (>=16.0,<17.0)
+Requires-Dist: tenacity (>=8.2.2,<9.0.0)
 Description-Content-Type: text/markdown
 
 # stigg-api-client

stigg_api_client-0.512.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+stigg/__init__.py,sha256=uQeM3YjvH1X56xOPknIEQezw0yjNNS-m9gi3B0XlSOM,44
+stigg/client.py,sha256=wSb0yyW13mWMr_4pooJArTtCa2fL5079ho9jGaZTdpE,3699
+stigg/generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+stigg/generated/operations.py,sha256=E2qRZg7gj22JSCAbAme2SXizMtYmQaS-XldpgO3_X7Y,51059
+stigg/generated/schema.py,sha256=MKLuaKjceCx_Q7S9ccf5sRZ6juN2qpSmYw5QbCBzNuM,467141
+stigg_api_client-0.512.0.dist-info/METADATA,sha256=A2RXZJUq9dx0_MxcirNVIB0zIcM6x7ODA-85KyTnZmI,3122
+stigg_api_client-0.512.0.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
+stigg_api_client-0.512.0.dist-info/RECORD,,

stigg_api_client-0.508.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-stigg/__init__.py,sha256=uQeM3YjvH1X56xOPknIEQezw0yjNNS-m9gi3B0XlSOM,44
-stigg/client.py,sha256=yJWbqm6yjLFGxtTHJxHX2mLqkCYMtOKxMDsN9vyFNcg,1141
-stigg/generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-stigg/generated/operations.py,sha256=E2qRZg7gj22JSCAbAme2SXizMtYmQaS-XldpgO3_X7Y,51059
-stigg/generated/schema.py,sha256=k9CCjx3pcjA9zOHr4Z9B5BYR2PMvEMjZhhlMYiPuIDw,467048
-stigg_api_client-0.508.0.dist-info/METADATA,sha256=fXDxyF01aGXDsG0pdHm5yhiOjU2El1tGDVgmDsioLik,3081
-stigg_api_client-0.508.0.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
-stigg_api_client-0.508.0.dist-info/RECORD,,