logos-sdk 0.0.25.dev2__tar.gz → 0.0.25.dev3__tar.gz

{logos_sdk-0.0.25.dev2 → logos_sdk-0.0.25.dev3}/PKG-INFO

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

{logos_sdk-0.0.25.dev2 → logos_sdk-0.0.25.dev3}/logos_sdk/__init__.py

@@ -1,4 +1,4 @@
-PRODUCTION = "PRODUCTION"
-DEVELOPMENT = "DEVELOPMENT"
-TESTING = "TESTING"
+PRODUCTION = "PRODUCTION"
+DEVELOPMENT = "DEVELOPMENT"
+TESTING = "TESTING"
 CLOUD_DEVELOPMENT = "CLOUD_DEVELOPMENT"

{logos_sdk-0.0.25.dev2 → logos_sdk-0.0.25.dev3}/logos_sdk/big_query/__init__.py

@@ -1,22 +1,22 @@
-from functools import wraps
-from google.api_core.exceptions import NotFound
-import time
-
-MAX_NUMBER_OF_ATTEMPTS = 2
-
-
-def retry_on_not_found(wrapped_function):
-    """This decorator retry call when table is not found. Insert into newly created table often fails with error
-    because API probably needs few seconds to see new created table"""
-
-    @wraps(wrapped_function)
-    def inner(*args, **kwargs):
-        for i in range(1, MAX_NUMBER_OF_ATTEMPTS + 1):
-            try:
-                kwargs["attempts"] = i
-                return wrapped_function(*args, **kwargs)
-                # this is because all request share same service
-            except NotFound as err:
-                time.sleep(2)
-
-    return inner
+from functools import wraps
+from google.api_core.exceptions import NotFound
+import time
+
+MAX_NUMBER_OF_ATTEMPTS = 2
+
+
+def retry_on_not_found(wrapped_function):
+    """This decorator retry call when table is not found. Insert into newly created table often fails with error
+    because API probably needs few seconds to see new created table"""
+
+    @wraps(wrapped_function)
+    def inner(*args, **kwargs):
+        for i in range(1, MAX_NUMBER_OF_ATTEMPTS + 1):
+            try:
+                kwargs["attempts"] = i
+                return wrapped_function(*args, **kwargs)
+                # this is because all request share same service
+            except NotFound as err:
+                time.sleep(2)
+
+    return inner

{logos_sdk-0.0.25.dev2 → logos_sdk-0.0.25.dev3}/logos_sdk/services/CampaignManager.py

@@ -194,7 +194,7 @@ class CampaignManagerService:
             if self.check_report_ready(report_id, file_id, secret_id):
                 return True
             else:
-                time.sleep((2 ** attempt) + randint(1, 20))
+                time.sleep((2**attempt) + randint(1, 20))
 
         return False
 
@@ -344,3 +344,56 @@ class CampaignManagerService:
 
             service_response = response.json()
             yield service_response["data"]["items"]
+
+    def create_and_get_report_results(
+        self,
+        account_id: str,
+        name: str,
+        start_date: str,
+        end_date: str,
+        dimensions: list,
+        metrics_names: list,
+        dimension_filters: list,
+        secret_id: str,
+        backoff_attempts: int,
+    ) -> Dict:
+        """
+        Method to create report, run it and fetch results in one go
+        """
+        report = {}
+        try:
+            report = self.create_report(
+                account_id=account_id,
+                name=name,
+                start_date=start_date,
+                end_date=end_date,
+                dimensions=dimensions,
+                metrics_names=metrics_names,
+                dimension_filters=dimension_filters,
+                secret_id=secret_id,
+            )
+            run_report = self.run_report(
+                account_id=account_id,
+                report_id=report["id"],
+                secret_id=secret_id,
+            )
+            if self.check_report_ready_with_exponential_backoff(
+                report_id=report["id"],
+                file_id=run_report["id"],
+                secret_id=secret_id,
+                backoff_attempts=backoff_attempts,
+            ):
+                return self.get_report_results(
+                    report_id=report["id"],
+                    file_id=run_report["id"],
+                    secret_id=secret_id,
+                )
+        except CampaignManagerServiceException as err:
+            raise err
+        finally:
+            if report:
+                self.delete_report(
+                    account_id=account_id,
+                    report_id=report["id"],
+                    secret_id=secret_id,
+                )