edit and create team sessions

rowers/forms.py

@@ -1,6 +1,6 @@
 from django import forms
 from django.contrib.admin.widgets import FilteredSelectMultiple
-from rowers.models import Workout
+from rowers.models import Workout,Rower,Team
 from rowers.rows import validate_file_extension,must_be_csv,validate_image_extension
 from django.contrib.auth.forms import UserCreationForm
 from django.contrib.auth.models import User
@@ -629,3 +629,26 @@ class WorkoutSessionSelectForm(forms.Form):
             widget = forms.CheckboxSelectMultiple,
             )
             
+class PlannedSessionTeamForm(forms.Form):
+    team = forms.ModelMultipleChoiceField(
+        queryset=Team.objects.all(),
+        required=False,
+        widget=forms.CheckboxSelectMultiple())
+
+    def __init__(self, user, *args, **kwargs):
+        super(PlannedSessionTeamForm,self).__init__(*args, **kwargs)
+
+        self.fields['team'].queryset = Team.objects.filter(manager=user)
+
+
+class PlannedSessionTeamMemberForm(forms.Form):
+    members = forms.ModelMultipleChoiceField(
+        queryset=Rower.objects.all(),
+        widget=forms.CheckboxSelectMultiple())
+    
+    def __init__(self, thesession, *args, **kwargs):
+        super(PlannedSessionTeamMemberForm,self).__init__(*args,**kwargs)
+
+        self.fields['members'].queryset = thesession.rower.all()
+        
+