added OTW and workouttype to favoritechart model

rowers/models.py

@@ -136,6 +136,12 @@ class FavoriteChart(models.Model):
         ('finish','Finish Angle'),
         ('peakforceangle','Peak Force Angle'),
         )
+
+    workouttypechoices = (
+        ('ote','Erg/SkiErg'),
+        ('otw','On The Water'),
+        ('both','both')
+        )
     
     plottypes = (
         ('line','Line Chart'),
@@ -143,15 +149,20 @@ class FavoriteChart(models.Model):
         )
     
     yparam1 = models.CharField(max_length=50,choices=y1params,verbose_name='Y1')
-    yparam2 = models.CharField(max_length=50,choices=y2params,verbose_name='Y2')
+    yparam2 = models.CharField(max_length=50,choices=y2params,verbose_name='Y2',default='None',blank=True)
     xparam = models.CharField(max_length=50,choices=xparams,verbose_name='X')
-    plottype = models.CharField(max_length=50,choices=plottypes,default='line')
+    plottype = models.CharField(max_length=50,choices=plottypes,
+                                default='line',
+                                verbose_name='Chart Type')
+    workouttype = models.CharField(max_length=50,choices=workouttypechoices,
+                                   default='both',
+                                   verbose_name='Workout Type')
     user = models.ForeignKey(Rower)
 
 class FavoriteForm(ModelForm):
     class Meta:
         model = FavoriteChart
-        fields = ['xparam','yparam1','yparam2','plottype']
+        fields = ['xparam','yparam1','yparam2','plottype','workouttype']
 
 class BaseFavoriteFormSet(BaseFormSet):
     def clean(self):
@@ -170,6 +181,15 @@ class BaseFavoriteFormSet(BaseFormSet):
                     'Must have x parameter.',
                     code='missing_xparam'
                     )
+
+            if not yparam1:
+                raise forms.ValidationError(
+                    'Must have Y1 parameter.',
+                    code='missing_yparam1'
+                    )
+
+            if not yparam2:
+                yparam2 = 'None'
         
 class Workout(models.Model):
     workouttypes = (

rowers/templates/favoritecharts.html

@@ -15,6 +15,9 @@
   </table>
   </div>
   {% endfor %}
+  <div class="grid_12 alpha">
+    <p>&nbsp;</p>
+  </div>
   <div class="grid_12 alpha">
     <div class="grid_2">
       <p><input type="submit" value="Update Favorites" class="button green small"/></p>
@@ -22,6 +25,8 @@
   </div>
 </form>
 
+
+
 <!-- Include formset plugin - including jQuery dependency -->
 <script src="//ajax.googleapis.com/ajax/libs/jquery/2.1.3/jquery.min.js"></script>
 <script src="/static/js/jquery.formset.js"></script>

rowers/templates/flexchart3.html

@@ -163,6 +163,7 @@
 
 </div>
 
+{% if user.rower.rowerplan == 'pro' %}
 <div id="favorites" class="grid_12 alpha">
   <div class="grid_2 suffix_4 alpha">
     {% if maxfav >= 0 %}
@@ -193,7 +194,7 @@
     {% endif %}
   </div>                        
 </div>
-
+{% endif %}
 
 {% endblock %}
 {% endlocaltime %}

rowers/templates/flexchart3otw.html

@@ -198,6 +198,38 @@
 
 </div>
 
+{% if user.rower.rowerplan == 'pro' %}
+<div id="favorites" class="grid_12 alpha">
+  <div class="grid_2 suffix_4 alpha">
+    {% if maxfav >= 0 %}
+    <a class="button gray small" href="/rowers/me/favoritecharts">Manage Favorites</a>
+    {% else %}
+    &nbsp;
+    {% endif %}
+  </div>
+  <div class="grid_1">
+    {% if favoritenr > 0 %}
+    <a class="button blue small" href="/rowers/workout/{{ id }}/flexchart?favoritechart={{ favoritenr|add:-1 }}">&lt</a>
+    {% else %}
+    <p>&nbsp;</p>
+    {% endif %}
+  </div>
+  <div class="grid_2">
+    <form enctype="multipart/form-data" action="{{ formloc }}" method="post">
+      {% csrf_token %}
+      <input class="grid_2 alpha button blue small" type="hidden" name="savefavorite" value="True">
+      <input class="grid_2 alpha button blue small" value="Make Favorite" type="Submit">
+    </form>
+  </div>
+  <div class="grid_1">
+  {% if favoritenr < maxfav %}
+  <a class="button blue small" href="/rowers/workout/{{ id }}/flexchart?favoritechart={{ favoritenr|add:1 }}">&gt</a>
+    {% else %}
+  <p>&nbsp;</p>
+    {% endif %}
+  </div>                        
+</div>
+{% endif %}
 
 {% endblock %}
 {% endlocaltime %}

rowers/views.py

@@ -2612,7 +2612,13 @@ def workout_flexchart3_view(request,*args,**kwargs):
 		mayedit=1
 
 
-    favorites = FavoriteChart.objects.filter(user=r).order_by("id")
+    workouttype = 'ote'
+    if row.workouttype == 'water':
+        workouttype = 'otw'
+
+
+    favorites = FavoriteChart.objects.filter(user=r,
+                                             workouttype__in=[workouttype,'both']).order_by("id")
     maxfav = len(favorites)-1
                 
     if 'xparam' in kwargs:
@@ -2633,9 +2639,13 @@ def workout_flexchart3_view(request,*args,**kwargs):
             
     if 'yparam2' in kwargs:
         yparam2 = kwargs['yparam2']
+        if yparam2 == '':
+            yparam2 = 'None'
     else:
         if favorites:
             yparam2 = favorites[favoritenr].yparam2
+            if yparam2 == '':
+                yparam2 = 'None'
         else:
             yparam2 = 'hr'
 
@@ -4492,18 +4502,19 @@ def workout_summary_edit_view(request,id,message="",successmessage=""
 		   })
 
 
-@login_required()
+@user_passes_test(promember,login_url="/login")
 def rower_favoritecharts_view(request):
     message = ''
     successmessage = ''
     r = Rower.objects.get(user=request.user)
     favorites = FavoriteChart.objects.filter(user=r).order_by('id')
     favorites_data = [{'yparam1':f.yparam1,
-                        'yparam2':f.yparam2,
+                       'yparam2':f.yparam2,
-                        'xparam':f.xparam,
+                       'xparam':f.xparam,
-                        'plottype':f.plottype}
+                       'plottype':f.plottype,
+                       'workouttype':f.workouttype}
                        for f in favorites]
-    FavoriteChartFormSet = formset_factory(FavoriteForm,formset=BaseFavoriteFormSet)
+    FavoriteChartFormSet = formset_factory(FavoriteForm,formset=BaseFavoriteFormSet,extra=0)
 
     if request.method == 'POST':
         favorites_formset = FavoriteChartFormSet(request.POST)
@@ -4515,11 +4526,13 @@ def rower_favoritecharts_view(request):
                 yparam2 = favorites_form.cleaned_data.get('yparam2')
                 xparam = favorites_form.cleaned_data.get('xparam')
                 plottype = favorites_form.cleaned_data.get('plottype')
+                workouttype = favorites_form.cleaned_data.get('workouttype')
                 new_instances.append(FavoriteChart(user=r,
                                                    yparam1=yparam1,
                                                    yparam2=yparam2,
                                                    xparam=xparam,
-                                                   plottype=plottype))
+                                                   plottype=plottype,
+                                                   workouttype=workouttype))
             try:
                 with transaction.atomic():
                     FavoriteChart.objects.filter(user=r).delete()