diff --git a/api/views.py b/api/views.py index acda60c..a8d84b5 100644 --- a/api/views.py +++ b/api/views.py @@ -1,3 +1,4 @@ +from pandas.io.feather_format import pd from .serializers import ClubSerializer, CourtSerializer, DateIntervalSerializer, TournamentSerializer, UserSerializer, ChangePasswordSerializer, EventSerializer, RoundSerializer, GroupStageSerializer, MatchSerializer, TeamScoreSerializer, TeamRegistrationSerializer, PlayerRegistrationSerializer, LiveMatchSerializer, PurchaseSerializer, UserUpdateSerializer, FailedApiCallSerializer, LogSerializer, DeviceTokenSerializer from tournaments.models import Club, Tournament, CustomUser, Event, Round, GroupStage, Match, TeamScore, TeamRegistration, PlayerRegistration, Court, DateInterval, Purchase, FailedApiCall, Log, DeviceToken @@ -194,7 +195,8 @@ class TeamScoreViewSet(viewsets.ModelViewSet): def get_queryset(self): tournament_id = self.request.query_params.get('tournament') if tournament_id: - return self.queryset.filter(team_registration__tournament=tournament_id) + q = Q(team_registration__tournament=tournament_id) | Q(match__group_stage__tournament=tournament_id) | Q(match__round__tournament=tournament_id) + return self.queryset.filter(q) if self.request.user: return self.queryset.filter(team_registration__tournament__event__creator=self.request.user) return []