diff --git a/rowers/braintreestuff.py b/rowers/braintreestuff.py index 7d6a5fbd..ac46f4e2 100644 --- a/rowers/braintreestuff.py +++ b/rowers/braintreestuff.py @@ -172,7 +172,7 @@ def update_subscription(rower,data,method='up'): if rower.paidplan != 'coach': try: - coachgroup = coach.mycoachgroup + coachgroup = rower.mycoachgroup except CoachingGroup.DoesNotExist: coachgroup = CoachingGroup() coachgroup.save() diff --git a/rowers/models.py b/rowers/models.py index a846a2aa..500c5ad9 100644 --- a/rowers/models.py +++ b/rowers/models.py @@ -1023,7 +1023,34 @@ def checkworkoutuser(user,workout): except Rower.DoesNotExist: return False +def checkviewworkouts(user,rower): + try: + r = user.rower + if rower == r: + return True + teams = Team.objects.filter(manager=user) + print Rower.objects.filter(team__in=teams) + if rower in Rower.objects.filter(team__in=teams): + return True + except Rower.DoesNotExist: + return False +# check if user is plan and rower is in his group +def checkaccessplanuser(user,rower): + try: + r = Rower.objects.get(user=user) + if rower == r: + return True + team_managers = [t.manager for t in rower.team.all() if t.manager.rower.rowerplan in ['plan','coach']] + if user.rower.rowerplan != 'basic': + return user in team_managers + else: + return False + + return False + except Rower.DoesNotExist: + return False + # Check if user is coach or rower def checkaccessuser(user,rower): try: diff --git a/rowers/templates/list_workouts.html b/rowers/templates/list_workouts.html index 7a42b7c2..bfac39a8 100644 --- a/rowers/templates/list_workouts.html +++ b/rowers/templates/list_workouts.html @@ -189,7 +189,7 @@
- - -{% if user.is_authenticated and user|is_manager %} +{% if user.is_authenticated and user|is_planmember %}
- -{% if user|coach_rowers %} +{% if user|team_members %}
+ {% if user|coach_rowers %}