making stack chart setting

rowers/views/userviews.py

@@ -250,6 +250,27 @@ def rower_favoritecharts_view(request, userid=0):
     staticchartform = StaticChartRowerForm(instance=r)
     datasettingsform = DataRowerForm(instance=r)
 
+    xparam = 'time'
+    yparam1 = 'pace'
+    yparam2 = 'power'
+    yparam3 = 'hr'
+    yparam4 = 'spm'
+
+    xparam = r.chartstacktemplate_x
+    yparam1 = r.chartstacktemplate_y[0]
+    yparam2 = r.chartstacktemplate_y[1]
+    yparam3 = r.chartstacktemplate_y[2]
+    yparam4 = r.chartstacktemplate_y[3]
+    
+    initial = {
+        'xaxis': xparam,
+        'yaxis1': yparam1,
+        'yaxis2': yparam2,
+        'yaxis3': yparam3,
+        'yaxis4': yparam4,
+    }
+    stackedchartform = StravaChartForm(request, initial=initial)
+
     favorites = FavoriteChart.objects.filter(user=r).order_by('id')
     aantal = len(favorites)
     favorites_data = [{'yparam1': f.yparam1,
@@ -272,6 +293,17 @@ def rower_favoritecharts_view(request, userid=0):
     workoutnametemplate_data = [{'element': element} for element in r.workoutnametemplate]
     workoutnametemplate_formset = WorkoutNameTemplateFormSet(initial=workoutnametemplate_data, prefix='workoutname')
 
+    if request.method == 'POST' and 'yaxis4' in request.POST:
+        stackedchartform = StravaChartForm(request, request.POST)
+        if stackedchartform.is_valid():
+            r.chartstacktemplate_x = stackedchartform.cleaned_data.get('xaxis')
+            r.chartstacktemplate_y = [stackedchartform.cleaned_data.get('yaxis1'),
+                                      stackedchartform.cleaned_data.get('yaxis2'),
+                                      stackedchartform.cleaned_data.get('yaxis3'),
+                                      stackedchartform.cleaned_data.get('yaxis4')]
+            r.save()
+            messages.info(request, "We have updated your stacked chart settings")
+
     if request.method == 'POST' and 'workoutname-TOTAL_FORMS' in request.POST:
         if 'defaults_workoutname' in request.POST:
             r.workoutnametemplate = ['date','name','distance','ownerfirst','ownerlast','duration','boattype','workouttype']
@@ -371,6 +403,7 @@ def rower_favoritecharts_view(request, userid=0):
         'favorites_formset': favorites_formset,
         'teams': get_my_teams(request.user),
         'rower': r,
+        'stackedchartform': stackedchartform,
         'staticchartform': staticchartform,
         'datasettingsform': datasettingsform,
         'workoutnametemplate_formset': workoutnametemplate_formset,

rowers/views/workoutviews.py

@@ -4240,6 +4240,12 @@ def workout_flexchart_stacked_view(request, *args, **kwargs):
     yparam3 = 'hr'
     yparam4 = 'spm'
 
+    xparam = r.chartstacktemplate_x
+    yparam1 = r.chartstacktemplate_y[0]
+    yparam2 = r.chartstacktemplate_y[1]
+    yparam3 = r.chartstacktemplate_y[2]
+    yparam4 = r.chartstacktemplate_y[3]
+
     if request.method == 'POST':
         flexaxesform = StravaChartForm(request, request.POST)
         if flexaxesform.is_valid():
@@ -4250,6 +4256,11 @@ def workout_flexchart_stacked_view(request, *args, **kwargs):
             yparam3 = cd['yaxis3']
             yparam4 = cd['yaxis4']
 
+            if 'save' in request.POST:
+                r.chartstacktemplate_x = xparam
+                r.chartstacktemplate_y = [yparam1, yparam2, yparam3, yparam4]
+                r.save()
+
     (
         script, div
     ) = interactive_flexchart_stacked(