diff --git a/visualizer/service/api/v10.py b/visualizer/service/api/v10.py index 1bbd79f..684bed2 100644 --- a/visualizer/service/api/v10.py +++ b/visualizer/service/api/v10.py @@ -109,7 +109,8 @@ def stop_visualization(data, app_id): raise exceptions.BadRequestException() plugin = data['plugin'] - if plugin == 'kubejobs': + # TODO We need to think in a better design for this + if plugin == 'kubejobs' or plugin == 'stream_kubejobs': # Call the executor by app_id and stop the visualization. visualized_apps[app_id].stop_visualization() diff --git a/visualizer/utils/datasources/datasource_influx.py b/visualizer/utils/datasources/datasource_influx.py index 1776945..d479bc9 100644 --- a/visualizer/utils/datasources/datasource_influx.py +++ b/visualizer/utils/datasources/datasource_influx.py @@ -45,7 +45,9 @@ def __init__(self, monitor_plugin, database_data, app_id): self.datasource_url = database_data['url'] self.datasource_port = database_data['port'] self.database_name = database_data['name'] - if(monitor_plugin == 'kubejobs'): + # TODO We need to think in a better design for this + if(monitor_plugin == 'kubejobs' or + monitor_plugin == 'stream_kubejobs'): self.dashboard_path = K8S_TEMPLATE_PATH elif(monitor_plugin == 'external_api'):