diff --git a/db.sqlite3 b/db.sqlite3 index 74ef62e..b5ae4fb 100644 Binary files a/db.sqlite3 and b/db.sqlite3 differ diff --git a/registered_user/views.py b/registered_user/views.py index 637f997..031a325 100644 --- a/registered_user/views.py +++ b/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)