Private
Public Access
1
0

Merge branch 'hotfix/v1.93'

This commit is contained in:
Sander Roosendaal
2017-04-07 08:25:21 +02:00
2 changed files with 23 additions and 10 deletions

View File

@@ -37,8 +37,12 @@ class WorkoutViewSet(viewsets.ModelViewSet):
serializer_class = WorkoutSerializer
def get_queryset(self):
r = Rower.objects.get(user=self.request.user)
return Workout.objects.filter(user=r).order_by("-date","-starttime")
try:
r = Rower.objects.get(user=self.request.user)
return Workout.objects.filter(user=r).order_by("-date","-starttime")
except TypeError:
return []
permission_classes = (
@@ -52,8 +56,11 @@ class RowerViewSet(viewsets.ModelViewSet):
#queryset = Rower.objects.all()
def get_queryset(self):
r = Rower.objects.filter(user=self.request.user)
return r
try:
r = Rower.objects.filter(user=self.request.user)
return r
except TypeError:
return []
permission_classes = (
IsOwnerOrNot,
@@ -68,8 +75,11 @@ class FavoriteChartViewSet(viewsets.ModelViewSet):
#queryset = FavoriteChart.objects.all()
def get_queryset(self):
r = Rower.objects.filter(user=self.request.user)
return FavoriteChart.objects.filter(user=r)
try:
r = Rower.objects.filter(user=self.request.user)
return FavoriteChart.objects.filter(user=r)
except TypeError:
return []
permission_classes = (
IsOwnerOrNot,

View File

@@ -173,10 +173,13 @@ def rdata(file,rower=rrower()):
# Query to get teams managed and member of
def get_my_teams(user):
therower = Rower.objects.get(user=user)
teams1 = therower.team.all()
teams2 = Team.objects.filter(manager=user)
teams = list(set(teams1).union(set(teams2)))
try:
therower = Rower.objects.get(user=user)
teams1 = therower.team.all()
teams2 = Team.objects.filter(manager=user)
teams = list(set(teams1).union(set(teams2)))
except TypeError:
teams = []
return teams