team multicreate

rowers/views/planviews.py

@@ -454,6 +454,7 @@ def plannedsession_multicreate_view(request,
 
     
     startdate,enddate = get_dates_timeperiod(request)
+    teamid = get_team(request)
     try:
         trainingplan = TrainingPlan.objects.filter(
             startdate__lte =  startdate,
@@ -472,6 +473,16 @@ def plannedsession_multicreate_view(request,
     ).order_by("startdate","preferreddate","enddate").exclude(
         sessiontype='race')
         
+    if teamid:
+        qset = qset.filter(team__in=[teamid])
+        try:
+            team = Team.objects.get(id=teamid)
+            if team.manager != request.user:
+                team = None
+        except Team.DoesNotExist:
+            team = None
+    else:
+        team = None
 
 
     initial = {
@@ -500,6 +511,8 @@ def plannedsession_multicreate_view(request,
             for ps in instances:
                 ps.save()
                 add_rower_session(r,ps)
+                if team:
+                    add_team_session(team,ps)
                 messages.info(request,"Saved changes for Planned Session "+str(ps))
             for obj in ps_formset.deleted_objects:
                 messages.info(request,"Deleted Planned Session "+str(obj))
@@ -515,6 +528,9 @@ def plannedsession_multicreate_view(request,
         startdatestring = startdate.strftime('%Y-%m-%d')
         enddatestring = enddate.strftime('%Y-%m-%d')
         url += '?when='+startdatestring+'/'+enddatestring
+
+        if team:
+            url += '&team={teamid}'.format(teamid=team.id)
         
         return HttpResponseRedirect(url)
 
@@ -537,16 +553,25 @@ def plannedsession_multicreate_view(request,
         'enddate':enddate
         })
 
+    if len(Team.objects.filter(manager=request.user))>=1:
+        teamform = RowerTeamForm(request.user)
+        if teamid:
+            teamform = RowerTeamForm(request.user,initial={'team':teamid})
+    else:
+        teamform = None
+
     context = {
         'ps_formset':ps_formset,
         'breadcrumbs':breadcrumbs,
         'rower':r,
         'active':'nav-plan',
         'dateform':dateform,
+        'teamform':teamform,
         'plan':trainingplan,
         'timeperiod':timeperiod,
         'teams':get_my_teams(request.user),
-        'extrasessions': extrasessions+1
+        'extrasessions': extrasessions+1,
+        'team':team,
         }