from rest_framework import viewsets
from rest_framework.permissions import IsAuthenticated
from xenua.django.views import UserScopedViewMixin

from xenua.django.rest.models import Token


class TokenViewSet(UserScopedViewMixin, viewsets.ModelViewSet):
    permission_classes = (IsAuthenticated,)
    queryset = Token.objects.none()  # hack for scope support

    def get_queryset(self):
        return Token.objects.filter(user=self.request.user)

    def get_serializer_class(self):
        from xenua.django.rest.serializers import TokenSerializer
        return TokenSerializer