views.py 文件源码

def get(self, request, *args, **kwargs):
        try:
            provided_token = request.META['HTTP_AUTHORIZATION']
            provided_token = provided_token.replace('Token ', '')
            token = Token.objects.select_related('user').get(key=provided_token)
        except Exception as e:
            # modify the original exception response
            raise exceptions.APIException("Token not valid.")

        if not token.user.is_active:
            # can also modify this exception message
            return Response(error_response("User inactive or deleted."))

        sport = int(request.query_params.get('sport', ''))

        if sport == 0:
            swimmer = Swimmer.objects.get(type=request.user.id)
            user_serialized = SwimmerSerializer(swimmer)
        elif sport == 1:
            runner = Runner.objects.get(type=request.user.id)
            user_serialized = RunnerSerializer(runner)

        return Response(success_response(user_serialized.data))