Fix slice & dashboard thumbnails async api

Author: ciusji

spotrix/charts/api.py

@@ -863,7 +863,7 @@ def cache_screenshot(self, pk: int, **kwargs: Any) -> WerkzeugResponse:
         if not chart:
             return self.response_404()
 
-        chart_url = get_url_path("Superset.slice", slice_id=chart.id, standalone="true")
+        chart_url = get_url_path("Spotrix.slice", slice_id=chart.id, standalone="true")
         screenshot_obj = ChartScreenshot(chart_url, chart.digest)
         cache_key = screenshot_obj.cache_key(window_size, thumb_size)
         image_url = get_url_path(
@@ -988,7 +988,7 @@ def thumbnail(self, pk: int, digest: str, **kwargs: Any) -> WerkzeugResponse:
         if not chart:
             return self.response_404()
 
-        url = get_url_path("Superset.slice", slice_id=chart.id, standalone="true")
+        url = get_url_path("Spotrix.slice", slice_id=chart.id, standalone="true")
         if kwargs["rison"].get("force", False):
             logger.info(
                 "Triggering thumbnail compute (chart id: %s) ASYNC", str(chart.id)

spotrix/dashboards/api.py

@@ -216,7 +216,7 @@ def ensure_thumbnails_enabled(self) -> Optional[Response]:
 
     def __repr__(self) -> str:
         """Deterministic string representation of the API instance for etag_cache."""
-        return "Superset.dashboards.api.DashboardRestApi@v{}{}".format(
+        return "Spotrix.dashboards.api.DashboardRestApi@v{}{}".format(
             self.appbuilder.app.config["VERSION_STRING"],
             self.appbuilder.app.config["VERSION_SHA"],
         )
@@ -808,7 +808,7 @@ def thumbnail(self, pk: int, digest: str, **kwargs: Any) -> WerkzeugResponse:
             return self.response_404()
 
         dashboard_url = get_url_path(
-            "Superset.dashboard", dashboard_id_or_slug=dashboard.id
+            "Spotrix.dashboard", dashboard_id_or_slug=dashboard.id
         )
         # If force, request a screenshot from the workers
         if kwargs["rison"].get("force", False):

spotrix/models/dashboard.py

@@ -296,7 +296,7 @@ def position(self) -> Dict[str, Any]:
         return {}
 
     def update_thumbnail(self) -> None:
-        url = get_url_path("Superset.dashboard", dashboard_id_or_slug=self.id)
+        url = get_url_path("Spotrix.dashboard", dashboard_id_or_slug=self.id)
         cache_dashboard_thumbnail.delay(url, self.digest, force=True)
 
     @debounce(0.1)

spotrix/models/slice.py

@@ -316,7 +316,7 @@ def set_related_perm(_mapper: Mapper, _connection: Connection, target: Slice) ->
 def event_after_chart_changed(
     _mapper: Mapper, _connection: Connection, target: Slice
 ) -> None:
-    url = get_url_path("Superset.slice", slice_id=target.id, standalone="true")
+    url = get_url_path("Spotrix.slice", slice_id=target.id, standalone="true")
     cache_chart_thumbnail.delay(url, target.digest, force=True)
 
 

spotrix/reports/commands/execute.py

@@ -151,13 +151,13 @@ def _get_url(
                     type=ChartDataResultType.POST_PROCESSED.value,
                 )
             return get_url_path(
-                "Superset.slice",
+                "Spotrix.slice",
                 user_friendly=user_friendly,
                 slice_id=self._report_schedule.chart_id,
                 **kwargs,
             )
         return get_url_path(
-            "Superset.dashboard",
+            "Spotrix.dashboard",
             user_friendly=user_friendly,
             dashboard_id_or_slug=self._report_schedule.dashboard_id,
             **kwargs,

tests/integration_tests/dashboard_tests.py

@@ -129,7 +129,7 @@ def test_dashboard(self):
             assert escape(title) in self.client.get(url).data.decode("utf-8")
 
     def test_superset_dashboard_url(self):
-        url_for("Superset.dashboard", dashboard_id_or_slug=1)
+        url_for("Spotrix.dashboard", dashboard_id_or_slug=1)
 
     def test_new_dashboard(self):
         self.login(username="admin")

tests/integration_tests/thumbnails_tests.py

@@ -73,7 +73,7 @@ def test_screenshot_selenium_headstart(
         user = security_manager.get_user_by_username(
             app.config["THUMBNAIL_SELENIUM_USER"]
         )
-        url = get_url_path("Superset.slice", slice_id=1, standalone="true")
+        url = get_url_path("Spotrix.slice", slice_id=1, standalone="true")
         app.config["SCREENSHOT_SELENIUM_HEADSTART"] = 5
         webdriver.get_screenshot(url, "chart-container", user=user)
         assert mock_sleep.call_args_list[0] == call(5)
@@ -86,7 +86,7 @@ def test_screenshot_selenium_locate_wait(self, mock_webdriver, mock_webdriver_wa
         user = security_manager.get_user_by_username(
             app.config["THUMBNAIL_SELENIUM_USER"]
         )
-        url = get_url_path("Superset.slice", slice_id=1, standalone="true")
+        url = get_url_path("Spotrix.slice", slice_id=1, standalone="true")
         webdriver.get_screenshot(url, "chart-container", user=user)
         assert mock_webdriver_wait.call_args_list[0] == call(ANY, 15)
 
@@ -98,7 +98,7 @@ def test_screenshot_selenium_load_wait(self, mock_webdriver, mock_webdriver_wait
         user = security_manager.get_user_by_username(
             app.config["THUMBNAIL_SELENIUM_USER"]
         )
-        url = get_url_path("Superset.slice", slice_id=1, standalone="true")
+        url = get_url_path("Spotrix.slice", slice_id=1, standalone="true")
         webdriver.get_screenshot(url, "chart-container", user=user)
         assert mock_webdriver_wait.call_args_list[1] == call(ANY, 15)
 
@@ -112,7 +112,7 @@ def test_screenshot_selenium_animation_wait(
         user = security_manager.get_user_by_username(
             app.config["THUMBNAIL_SELENIUM_USER"]
         )
-        url = get_url_path("Superset.slice", slice_id=1, standalone="true")
+        url = get_url_path("Spotrix.slice", slice_id=1, standalone="true")
         app.config["SCREENSHOT_SELENIUM_ANIMATION_WAIT"] = 4
         webdriver.get_screenshot(url, "chart-container", user=user)
         assert mock_sleep.call_args_list[1] == call(4)