adding cp chart to workout menu

rowers/views/analysisviews.py

@@ -34,11 +34,19 @@ defaultoptions = {
                   message="This functionality requires a Pro plan or higher. If you are already a Pro user, please log in to access this functionality",
                   redirect_field_name=None)
 @permission_required('rower.is_coach',fn=get_user_by_userid,raise_exception=True)
-def analysis_new(request,userid=0,function='boxplot',teamid=0):
+def analysis_new(request,userid=0,function='boxplot',teamid=0,id=0):
     r = getrequestrower(request, userid=userid)
     user = r.user
     userid = user.id
 
+    if id:
+        firstworkout = get_workout(id)
+        if not is_workout_team(request.user,firstworkout):
+            raise PermissionDenied("You are not allowed to use this workout")
+        firstworkoutquery = Workout.objects.filter(id=encoder.decode_hex(id))
+    else:
+        firtworkout = None
+
     try:
         theteam = Team.objects.get(id=teamid)
     except Team.DoesNotExist:
@@ -188,7 +196,8 @@ def analysis_new(request,userid=0,function='boxplot',teamid=0):
                                           startdatetime__lte=enddate,
                                           workouttype__in=modalities,
         )
-
+    if firstworkout:
+        workouts = firstworkoutquery | workouts
     workouts = workouts.order_by(
             "-date", "-starttime"
         ).exclude(boattype__in=negtypes)
@@ -212,6 +221,8 @@ def analysis_new(request,userid=0,function='boxplot',teamid=0):
 
     if request.method != 'POST':
         form = WorkoutMultipleCompareForm()
+        if id:
+            form.fields["workouts"].initial = [firstworkout]
         chartform = AnalysisChoiceForm(initial={'function':function})
         selectedworkouts = Workout.objects.none()
     else: