Initial version of workout choice field

rowers/views.py

@@ -26,7 +26,7 @@ from rowers.forms import (
     StatsOptionsForm,PredictedPieceForm,DateRangeForm,DeltaDaysForm,
     EmailForm, RegistrationForm, RegistrationFormTermsOfService,
     RegistrationFormUniqueEmail,CNsummaryForm,UpdateWindForm,
-    UpdateStreamForm
+    UpdateStreamForm,WorkoutMultipleCompareForm,
     )
 from rowers.models import Workout, User, Rower, WorkoutForm,FavoriteChart
 from rowers.models import (
@@ -1955,7 +1955,97 @@ def workout_setprivate_view(request,id,
 		  })
     return HttpResponseRedirect(url)
 
+# Team comparison
+@login_required()
+def team_comparison_select(request,
+                           startdatestring="",
+                           enddatestring="",
+                           message='',
+                           successmessage='',
+                           startdate=timezone.now()-datetime.timedelta(days=30),
+                           enddate=timezone.now()+datetime.timedelta(days=1),
+                           teamid=0):
 
+    try:
+	r = Rower.objects.get(user=request.user)
+    except Rower.DoesNotExist:
+        raise Http404("Rower doesn't exist")
+    
+    if request.method == 'POST':
+        dateform = DateRangeForm(request.POST)
+        if dateform.is_valid():
+            startdate = dateform.cleaned_data['startdate']
+            enddate = dateform.cleaned_data['enddate']
+    else:
+        dateform = DateRangeForm(initial={
+            'startdate':startdate,
+            'enddate':enddate,
+            })
+
+    startdate = datetime.datetime.combine(startdate,datetime.time())
+    enddate = datetime.datetime.combine(enddate,datetime.time(23,59,59))
+    enddate = enddate+datetime.timedelta(days=1)
+
+    if startdatestring:
+        startdate = iso8601.parse_date(startdatestring)
+    if enddatestring:
+        enddate = iso8601.parse_date(enddatestring)
+
+    if enddate < startdate:
+        s = enddate
+        enddate = startdate
+        startdate = s
+
+    try:
+        theteam = Team.objects.get(id=teamid)
+    except Team.DoesNotExist:
+        raise Http404("Team doesn't exist")
+        
+    if theteam.viewing == 'allmembers' or theteam.manager == request.user:
+        workouts = Workout.objects.filter(team=theteam,
+                                          startdatetime__gte=startdate,
+                                          startdatetime__lte=enddate).order_by("-date", "-starttime")
+    elif theteam.viewing == 'coachonly':
+        workouts = Workout.objects.filter(team=theteam,user=r,
+                                          startdatetime__gte=startdate,
+                                          startdatetime__lte=enddate).order_by("-date","-starttime")
+
+            
+    else:
+        theteam = None
+        workouts = Workout.objects.filter(user=r,
+                                      startdatetime__gte=startdate,
+                                      startdatetime__lte=enddate).order_by("-date", "-starttime")
+
+    query = request.GET.get('q')
+    if query:
+        query_list = query.split()
+        workouts = workouts.filter(
+            reduce(operator.and_,
+                   (Q(name__icontains=q) for q in query_list)) |
+            reduce(operator.and_,
+                   (Q(notes__icontains=q) for q in query_list))
+            )
+
+    workouts1 = workouts
+    
+
+
+    form = WorkoutMultipleCompareForm()
+    form.fields["workouts"].queryset = workouts1
+    
+    return render(request, 'team_compare_select.html',
+                  {'workouts': workouts,
+                   'message': message,
+                   'successmessage':successmessage,
+                   'dateform':dateform,
+                   'startdate':startdate,
+                   'enddate':enddate,
+                   'team':theteam,
+                   'form':form,
+                   })
+
+        
 # List Workouts    
 @login_required()
 def workouts_view(request,message='',successmessage='',