auth works now
This commit is contained in:
@ -4,6 +4,7 @@ from rest_framework.response import Response
|
||||
from rest_framework.request import Request
|
||||
from rest_framework.views import APIView
|
||||
from django.contrib.auth.models import User
|
||||
from rest_framework_simplejwt.tokens import RefreshToken
|
||||
|
||||
class CustomAuthToken(ObtainAuthToken):
|
||||
|
||||
@ -12,12 +13,19 @@ class CustomAuthToken(ObtainAuthToken):
|
||||
context={'request': request})
|
||||
serializer.is_valid(raise_exception=True)
|
||||
user = serializer.validated_data['user']
|
||||
token, created = Token.objects.get_or_create(user=user)
|
||||
return Response({
|
||||
'token': token.key,
|
||||
'user_id': user.pk,
|
||||
'username': user.username
|
||||
})
|
||||
refresh = RefreshToken.for_user(user)
|
||||
|
||||
return {
|
||||
'refresh': str(refresh),
|
||||
'access': str(refresh.access_token),
|
||||
}
|
||||
#
|
||||
# token, created = Token.objects.get_or_create(user=user)
|
||||
# return Response({
|
||||
# 'token': token.key,
|
||||
# 'user_id': user.pk,
|
||||
# 'username': user.username
|
||||
# })
|
||||
|
||||
|
||||
class RegisterView(APIView):
|
||||
@ -45,11 +53,16 @@ class RegisterView(APIView):
|
||||
username=username,
|
||||
password=password
|
||||
)
|
||||
token, created = Token.objects.get_or_create(user=user)
|
||||
return Response({
|
||||
'token': token.key,
|
||||
'user_id': user.pk,
|
||||
'username': user.username
|
||||
}, status=200)
|
||||
refresh = RefreshToken.for_user(user)
|
||||
return {
|
||||
'refresh': str(refresh),
|
||||
'access': str(refresh.access_token),
|
||||
}
|
||||
# token, created = Token.objects.get_or_create(user=user)
|
||||
# return Response({
|
||||
# 'token': token.key,
|
||||
# 'user_id': user.pk,
|
||||
# 'username': user.username
|
||||
# }, status=200)
|
||||
except Exception as e:
|
||||
return Response({"message": "Something went wrong, registration is not completed", "error": e}, status=500)
|
||||
|
||||
Reference in New Issue
Block a user