WIP

registered_user/views.py

@@ -10,6 +10,7 @@ from rest_framework.authentication import TokenAuthentication
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.decorators import authentication_classes
 from rest_framework.decorators import permission_classes
+from django.shortcuts import get_object_or_404
 import json
 
 
@@ -53,9 +54,9 @@ def login_registered_user(request):
     user = authenticate(username=email, password=password)
 
     if user is not None:
-        registered_user = RegisteredUser.objects.get(user=user)
+        registered_user = get_object_or_404(RegisteredUser, user=user)
         serializer = RegisteredUserSerializer(registered_user, many=False)
-        token = Token.objects.get(user=registered_user.user).key
+        token = get_object_or_404(Token, user=registered_user.user).key
         data = serializer.data
         data["token"] = token
         return Response(data,status=status.HTTP_200_OK)