upgini 1.1.305__tar.gz → 1.1.306__tar.gz

{upgini-1.1.305 → upgini-1.1.306}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.305
+Version: 1.1.306
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

upgini-1.1.306/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.306"

{upgini-1.1.305 → upgini-1.1.306}/src/upgini/data_source/data_source_publisher.py

@@ -418,3 +418,19 @@ class DataSourcePublisher:
         response = self._rest_client.union_search_tasks(request, "trace_id")
         print(response)
         return response
+
+    def reannounce_all_ads(self):
+        trace_id = str(uuid.uuid4())
+        with MDC(trace_id=trace_id):
+            try:
+                task_id = self._rest_client.reannounce_all_ads(trace_id)
+                with Spinner():
+                    status_response = self._rest_client.poll_ads_management_task_status(task_id, trace_id)
+                    while status_response["status"] not in self.FINAL_STATUSES:
+                        time.sleep(5)
+                        status_response = self._rest_client.poll_ads_management_task_status(task_id, trace_id)
+
+                if status_response["status"] != "COMPLETED":
+                    raise Exception("Failed to reannounce all ADS: " + status_response["errorMessage"])
+            except Exception:
+                self.logger.exception("Failed to reannounce all ADS-es")

{upgini-1.1.305 → upgini-1.1.306}/src/upgini/http.py

@@ -285,6 +285,7 @@ class _RestClient:
     DEACTIVATE_ADS_URI = "private/api/v2/ads/deactivate"
     TOGGLE_ADS_URI_FMT = "private/api/v2/ads/{0}/toggle"
     DELETE_ADS_URI_FMT = "private/api/v2/ads/{0}"
+    REANNOUNCE_ALL_ADS_URI_FMT = "private/api/v2/ads/reannounce-all-ads"
     POLL_ADS_MANAGEMENT_STATUS_URI_FMT = "private/api/v2/ads/management-task/{0}"
     GET_ADS_DESCRIPTION_URI_FMT = "private/api/v2/ads/{0}"
     GET_ALL_ADS_DESCRIPTIONS_URI = "private/api/v2/ads/descriptions"
@@ -726,6 +727,11 @@ class _RestClient:
         api_path = self.DEACTIVATE_ADS_URI
         self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id, request, result_format=None))
 
+    def reannounce_all_ads(self, trace_id: str) -> str:
+        api_path = self.REANNOUNCE_ALL_ADS_URI_FMT
+        response = self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id))
+        return response["adsManagementTaskId"]
+
     def toggle_ads(self, ads_definition_id: str, trace_id: str):
         api_path = self.TOGGLE_ADS_URI_FMT.format(ads_definition_id)
         return self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id))

upgini-1.1.305/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.305"