team expiry date added to Rower model

rowers/teams.py

@@ -33,6 +33,20 @@ def remove_team(id):
     t = Team.objects.get(id=id)
     return t.delete()
 
+def set_teamplanexpires(rower):
+    ts = Team.objects.filter(rower=rower)
+
+    texp = datetime.date(timezone.now())
+
+    for t in ts:
+        mr = Rower.objects.get(user=t.manager)
+        if mr.teamplanexpires > texp:
+            rower.teamplanexpires = mr.teamplanexpires
+
+    t.save()
+
+    return (1,'Updated rower team expiry')
+
 def add_member(id,rower):
     t= Team.objects.get(id=id)
     rower.team.add(t)
@@ -40,7 +54,9 @@ def add_member(id,rower):
     ws = Workout.objects.filter(user=rower)
     for w in ws:
         w.team.add(t)
-    
+
+    set_teamplanexpires(rower)
+        
     return (1,'Member added')
 
 def remove_member(id,rower):
@@ -51,6 +67,7 @@ def remove_member(id,rower):
     for w in ws:
         w.team.remove(t)
 
+    set_teamplanexpires(rower)
     return (1,'Member removed')
 
 def mgr_remove_member(id,manager,rower):