logos-sdk 0.0.25.dev8__tar.gz → 0.0.25.dev9__tar.gz

{logos_sdk-0.0.25.dev8 → logos_sdk-0.0.25.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: logos-sdk
-Version: 0.0.25.dev8
+Version: 0.0.25.dev9
 Summary: SDK for Logos platform
 Home-page: https://bitbucket.org/databy/logos-sdk-pip/src/master/
 Author: Databy.io

{logos_sdk-0.0.25.dev8 → logos_sdk-0.0.25.dev9}/logos_sdk/big_query/BigQuery.py

@@ -36,6 +36,19 @@ class BigQuery:
         except DefaultCredentialsError:
             self.logger = None
 
+    def parse_fields(self, fields):
+        result = []
+        for row in fields:
+            result.append(
+                bigquery.schema.SchemaField(
+                    row["name"],
+                    row["col_type"],
+                    mode=row["mode"],
+                    fields=self.parse_fields(row["fields"]) if "fields" in row else [],
+                )
+            )
+        return result
+
     def get_dataset(self, dataset_id: str):
         return self._service.get_dataset(dataset_id)
 
@@ -57,17 +70,17 @@ class BigQuery:
         return self._service.get_table(sql_format)
 
     def insert_into_table(
-            self, dataset_id: str, table_id: str, records: List[Dict]
+        self, dataset_id: str, table_id: str, records: List[Dict]
     ) -> None:
         bq_table = self.get_table(dataset_id, table_id)
         self._insert_into_table(bq_table, records)
 
     def insert_create_table(
-            self,
-            dataset_id: str,
-            table_id: str,
-            records: List[Dict],
-            schema_columns: List[Dict],
+        self,
+        dataset_id: str,
+        table_id: str,
+        records: List[Dict],
+        schema_columns: List[Dict],
     ) -> None:
         bq_table = self.check_table_exists(dataset_id, table_id)
         if bq_table is None:
@@ -87,10 +100,15 @@ class BigQuery:
             return None
 
     def create_table(
-            self, dataset_id: str, table_id: str, schema_columns: List[Dict]
+        self, dataset_id: str, table_id: str, schema_columns: List[Dict]
     ) -> Union[bool, Table]:
         table_schema = [
-            bigquery.schema.SchemaField(row["name"], row["col_type"], mode=row["mode"])
+            bigquery.schema.SchemaField(
+                row["name"],
+                row["col_type"],
+                mode=row["mode"],
+                fields=self.parse_fields(row["fields"]) if "fields" in row else [],
+            )
             for row in schema_columns
         ]
         try:
@@ -132,13 +150,13 @@ class BigQuery:
 
     @retry_on_not_found
     def _insert_into_table(
-            self, bq_table: Table, records: List[Dict], attempts: int
+        self, bq_table: Table, records: List[Dict], attempts: int
     ) -> None:
         if len(records) > self.BQ_ROWS_LIMIT:
             for index in range(0, len(records), self.BQ_ROWS_LIMIT):
                 errors = self._service.insert_rows(
                     bq_table,
-                    records[index: (index + self.BQ_ROWS_LIMIT)],
+                    records[index : (index + self.BQ_ROWS_LIMIT)],
                     retry=Retry(
                         total=2, connect=4, backoff_factor=2, allowed_methods=None
                     ),

{logos_sdk-0.0.25.dev8 → logos_sdk-0.0.25.dev9}/logos_sdk/services/MerchantCenter.py

@@ -92,26 +92,39 @@ class MerchantCenterService:
             service_response = response.json()
             yield service_response["data"]["results"]
 
-    def list_products_statuses(self, merchant_account_id: str, secret_id: str):
+    def list_products_statuses(self, merchant_account_id: str, secret_id: str, page_size: int = 250):
         """
         Lists the statuses of the products in your Merchant Center account
         :param merchant_account_id: The ID of the
         account that contains the products. This account cannot be a multi-client account
         :param secret_id: The ID of the secret in secret manager
+        :param page_size: size of the page
         :return: List[Dict]
         """
-        body = {"merchant_account_id": merchant_account_id, "secret_id": secret_id}
+        body = {"merchant_account_id": merchant_account_id, "secret_id": secret_id, "page_size": page_size}
         header = get_headers(self._LIST_PRODUCT_STATUSES)
         response = request(
             "post", url=self._LIST_PRODUCT_STATUSES, json=body, headers=header
         )
 
-        if response.status_code == HTTPStatus.OK:
-            service_response = response.json()
-            return service_response["data"]
-        else:
+        if response.status_code != HTTPStatus.OK:
             raise MerchantServiceException(response.content)
 
+        service_response = response.json()
+        yield service_response["data"]["results"]
+
+        while service_response["data"]["nextPageToken"] is not None:
+            body["page_token"] = service_response["data"]["nextPageToken"]
+            response = request(
+                "post", url=self._LIST_PRODUCT_STATUSES, json=body, headers=header
+            )
+
+            if response.status_code != HTTPStatus.OK:
+                raise MerchantServiceException(response.content)
+
+            service_response = response.json()
+            yield service_response["data"]["results"]
+
     def reports_search(
             self,
             merchant_account_id: str,

{logos_sdk-0.0.25.dev8 → logos_sdk-0.0.25.dev9}/logos_sdk/services/Sklik.py

@@ -69,6 +69,7 @@ class SklikService:
         self._CREATE_REPORT = self._URL + "/create-report"
         self._READ_REPORT = self._URL + "/read-report"
         self._API = self._URL + "/call-api"
+        self._GET_CLIENT = self._URL + "/get-client"
         self._CHECK_DATA_READY = self._URL + "/check-data-ready"
         self._LOGOUT = self._URL + "/logout"
         self._API_LIMITS = self._URL + "/api-limits"
@@ -283,6 +284,33 @@ class SklikService:
             raise SklikServiceException(response.content)
 
     @get_session_if_malformed
+    def get_client(
+            self,
+            secret_id: str,
+            account_email: str,
+    ) -> Union[List, Dict]:
+        """
+        Function to call SklikService get-client route
+        :param secret_id: The ID of the secret in secret manager
+        :param account_email: Account email to refers to Sklik accountId
+        :return:
+        """
+
+        if self.session is None:
+            self.get_session(secret_id, account_email)
+
+        response = execute_request(
+            "post",
+            url=self._GET_CLIENT,
+            json={"sklik_session": self.session["sklik_session"]},
+        )
+
+        if response.status_code == HTTPStatus.OK:
+            service_response = response.json()
+            return service_response["data"]
+        else:
+            raise SklikServiceException(response.content)
+    @get_session_if_malformed
     def check_data_ready(
         self,
         secret_id: str,
@@ -343,4 +371,4 @@ class SklikService:
             service_response = response.json()
             return service_response["data"]
         else:
-            raise SklikServiceException(response.content)
+            raise SklikServiceException(response.content)

{logos_sdk-0.0.25.dev8 → logos_sdk-0.0.25.dev9}/logos_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: logos-sdk
-Version: 0.0.25.dev8
+Version: 0.0.25.dev9
 Summary: SDK for Logos platform
 Home-page: https://bitbucket.org/databy/logos-sdk-pip/src/master/
 Author: Databy.io