added some model forms

rowers/models.py

@@ -678,6 +678,17 @@ class TrainingTarget(models.Model):
         default=timezone.now()+datetime.timedelta(days=182))
     notes = models.TextField(max_length=300,blank=True)
 
+class TrainingTargetForm(ModelForm):
+    class Meta:
+        model = TrainingTarget
+        fields = ['name','date','notes']
+
+	widgets = {
+	    'date': SelectDateWidget(
+                years=range(
+                    timezone.now().year-1,timezone.now().year+1)),
+	    }
+
 # SportTracks has a TrainingGoal like this
 #class TrainingGoal(models.Model):
 #    rower = models.ForeignKey(Rower)
@@ -701,6 +712,21 @@ class TrainingPlan(models.Model):
     enddate = models.DateField(
         default=timezone.now()+datetime.timedelta(days=182))
 
+class TrainingPlanForm(ModelForm):
+    class Meta:
+        model = TrainingPlan
+        fields = ['name','target','startdate','enddate']
+
+	widgets = {
+	    'startdate': SelectDateWidget(
+                years=range(
+                    timezone.now().year-1,timezone.now().year+1)),
+	    'enddate': SelectDateWidget(
+                years=range(
+                    timezone.now().year-1,timezone.now().year+1)),
+	    }
+
+    
 cycletypechoices = (
     ('filler','System Defined'),
     ('userdefined','User Defined')

rowers/plannedsessions.py

@@ -82,14 +82,24 @@ def is_session_complete(ps):
 def rank_results(ps):
     return 1
 
-def add_team(t,ps):
+def add_team_session(t,ps):
     ps.team.add(t)
     ps.save()
 
     return 1
 
-def add_rower(r,ps):
+def add_rower_session(r,ps):
     ps.rower.add(r)
     ps.save()
     
     return 1
+
+def remove_team_session(t,ps):
+    ps.team.remove(t)
+
+    return 1
+
+def remove_rower_session(r,ps):
+    ps.rower.remove(r)
+
+    return 1

templates/base.html

@@ -3,6 +3,13 @@
 {% load rowerfilters %}
 {% endblock %}
 
+{% block plans %}
+{% if user.rower.rowerplan == 'pro' or user.rower.rowerplan == 'coach' %}
+<p><a class="button gray small" href="/rowers/sessions/">Plans</a></p>
+{% else %}
+<p><a class="button gray small" href="/rowers/promembership/">Plans</a></p>
+{% endif %}
+{% endblock %}
 
 
 {% block teams %}

templates/basebase.html

@@ -201,6 +201,14 @@
           {% block teams %}
           {% endblock %}
         </div>
+        <div class="grid_1 tooltip">
+          {% block plans %}
+          {% endblock %}
+        </div>
+        <div class="grid_1 tooltip">
+          {% block challenges %}
+          {% endblock %}
+        </div>
       </div>