Display sankey from eesyplan method instead of openplan

Author: paulapreuss

app/dashboard/views.py

@@ -1,9 +1,16 @@
+import tempfile
+from pathlib import Path
+
 import numpy as np
+import pandas as pd
 from django.core.exceptions import PermissionDenied
 from django.template.loader import get_template
 from django.db.models import Count, Value, F, Q, Case, When
 from django.db.models.functions import Concat, Replace
 from django.http.response import Http404, HttpResponse
+from oemof.datapackage import datapackage
+from oemof.eesyplan.datapackage.energy_system import create_energy_system_from_dp
+
 from dashboard.helpers import *
 from dashboard.models import (
     AssetsResults,
@@ -64,7 +71,9 @@
 import datetime
 import logging
 import traceback
-from projects.helpers import parameters_helper
+from projects.helpers import parameters_helper, add_timeseries_to_database_datapackage
+
+import oemof.eesyplan.postprocessing.graphs as eesyplan_graphs
 
 logger = logging.getLogger(__name__)
 
@@ -1223,21 +1232,30 @@ def scenario_visualize_sankey(request, scen_id, ts=None):
         is False
     ):
         raise PermissionDenied
-    if ts is not None:
-        ts = int(ts)
-    results_json = report_item_render_to_json(
-        report_item_id="sankey",
-        data=REPORT_GRAPHS[GRAPH_SANKEY](
-            simulation=scenario.simulation,
-            energy_vector=scenario.energy_vectors,
-            timestep=ts,
-        ),
-        title="Sankey",
-        report_item_type=GRAPH_SANKEY,
-    )
 
+    # TODO this sankey implementation does not allow for single timesteps
+    # if ts is not None:
+    #     ts = int(ts)
+    rebuilt_dp = add_timeseries_to_database_datapackage(scenario)
+
+    # TODO these flows do not generate the correct sankey result yet
+    with tempfile.TemporaryDirectory(prefix="dp_") as td:
+        temp_path = Path(td)
+        dp_path = datapackage.rebuild_dp_from_json(rebuilt_dp, temp_path)
+        es = create_energy_system_from_dp(dp_path)
+
+        qs = FancyResults.objects.filter(simulation=scenario.simulation)
+
+        if qs.exists():
+            flows = {
+                (asset, bus): json.loads(flow_data)
+                for asset, bus, flow_data in qs.values_list("asset", "bus", "flow_data")
+            }
+            flows_df = pd.DataFrame(flows)
+            flows_df.index = scenario.get_timestamps()[: len(flows_df)]
+        fig, links_df = eesyplan_graphs.sankey(flows=flows_df, es=es)
     return JsonResponse(
-        results_json, status=200, content_type="application/json", safe=False
+        fig.to_dict(), status=200, content_type="application/json", safe=False
     )
 
 

app/static/js/update_results_page.js

@@ -227,8 +227,8 @@ function scenario_visualize_sankey(scen_id, ts=null){
  $.ajax({
             url: urlVisualizeSankey + urlParams,
             type: "GET",
-            success: async (parameters) => {
-                await graph_type_mapping[parameters.type](parameters.id, parameters);
+            success: async (eesyplan_fig) => {
+                await Plotly.newPlot("sankey", eesyplan_fig.data, eesyplan_fig.layout);
             },
         });
 }