passing tests - teams can be created

rowers/teams.py

@@ -61,6 +61,14 @@ def update_team(t,name,manager,private,notes,viewing):
 
 def create_team(name,manager,private='open',notes='',viewing='allmembers'):
     # needs some error testing
+    if manager.rower.rowerplan == 'basic':
+        if manager.rower.protrialexpires < timezone.now().date() and manager.rower.plantrialexpires < timezone.now().date():
+            return (0,'You need to upgrade to a paid plan to establish a team')
+    elif manager.rower.rowerplan != 'coach':
+        ts = Team.objects.filter(manager=manager)
+        if len(ts)>=1:
+            return (0,'You need to upgrade to the Coach plan to have more than one team')
+    
     try:
         t = Team(name=name,manager=manager,notes=notes,
                  private=private,viewing=viewing)
@@ -77,17 +85,18 @@ def remove_team(id):
         send_team_delete_mail(t,r)
     return t.delete()
 
-def set_teamplanexpires(rower):
-    ts = Team.objects.filter(rower=rower)
+#def set_teamplanexpires(rower):
+#    ts = Team.objects.filter(rower=rower)
 
-    texp = datetime.date(timezone.now())
+#    texp = datetime.date(timezone.now())
 
-    for t in ts:
-        mr = Rower.objects.get(user=t.manager)
-        if mr.teamplanexpires > texp:
-            rower.teamplanexpires = mr.teamplanexpires
+#    for t in ts:
+#        print t.name
+#        mr = Rower.objects.get(user=t.manager)
+#        if mr.teamplanexpires > texp:
+#            rower.teamplanexpires = mr.teamplanexpires
 
-        t.save()
+#        t.save()
 
     return (1,'Updated rower team expiry')
 
@@ -99,7 +108,7 @@ def add_member(id,rower):
     
     res = handle_add_workouts_team(ws,t)
 
-    set_teamplanexpires(rower)
+#    set_teamplanexpires(rower)
         
     return (id,'Member added')
 
@@ -111,7 +120,7 @@ def remove_member(id,rower):
 
     res = handle_remove_workouts_team(ws,t)
             
-    set_teamplanexpires(rower)
+#    set_teamplanexpires(rower)
     return (id,'Member removed')
 
 def mgr_remove_member(id,manager,rower):