diff --git a/rowers/models.py b/rowers/models.py index 2838451b..8fa1f441 100644 --- a/rowers/models.py +++ b/rowers/models.py @@ -623,6 +623,22 @@ def checkworkoutuser(user,workout): except Rower.DoesNotExist: return False +# Check if user is coach or rower +def checkaccessuser(user,rower): + try: + r = Rower.objects.get(user=user) + teams = Team.objects.filter(manager=user) + if rower == r: + return True + elif teams: + for team in teams: + if team in rower.team.all(): + return True + else: + return False + except Rower.DoesNotExist: + return False + timezones = ( (x,x) for x in pytz.common_timezones ) diff --git a/rowers/templates/list_workouts.html b/rowers/templates/list_workouts.html index 4a1963a9..82b4bba2 100644 --- a/rowers/templates/list_workouts.html +++ b/rowers/templates/list_workouts.html @@ -50,7 +50,6 @@