106 lines
3.9 KiB
Python
106 lines
3.9 KiB
Python
# main/views.py
|
|
from django.contrib.auth import authenticate, login, logout
|
|
from rest_framework import status
|
|
# from rest_framework.views import APIView
|
|
from rest_framework.generics import GenericAPIView
|
|
from rest_framework.response import Response
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
from django.contrib.auth.models import User
|
|
from rest_framework.authtoken.models import Token
|
|
from rest_framework.permissions import AllowAny, IsAuthenticated
|
|
from rest_framework.authentication import TokenAuthentication
|
|
from rest_framework_simplejwt.authentication import JWTAuthentication
|
|
from rest_framework_simplejwt.tokens import RefreshToken
|
|
from .serializers import *
|
|
|
|
|
|
class UserRegistrationView(GenericAPIView):
|
|
permission_classes = (AllowAny,)
|
|
serializer_class = UserRegistrationSerializer
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
serializer = UserRegistrationSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
user = serializer.save()
|
|
token = RefreshToken.for_user(user)
|
|
data = serializer.data
|
|
|
|
data["tokens"] = {"refresh": str(token),
|
|
"access": str(token.access_token)}
|
|
return Response(data, status=status.HTTP_201_CREATED)
|
|
# if serializer.is_valid():
|
|
# user = serializer.save()
|
|
# return Response({"id": user.id,
|
|
# "username": user.username}, status=status.HTTP_201_CREATED)
|
|
# return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class UserLoginView(GenericAPIView):
|
|
permission_classes = (AllowAny,)
|
|
serializer_class = UserLoginSerializer
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
user = serializer.validated_data
|
|
serializer = CustomUserSerializer(user)
|
|
token = RefreshToken.for_user(user)
|
|
data = serializer.data
|
|
data["tokens"] = {"refresh": str(token),
|
|
"access": str(token.access_token)}
|
|
return Response(data, status=status.HTTP_200_OK)
|
|
# class UserLoginView(APIView):
|
|
# authentication_classes = [JWTAuthentication]
|
|
# permission_classes = [IsAuthenticated]
|
|
|
|
# def get(self, request):
|
|
# content = {'message': 'Hello, World!'}
|
|
# return Response(content)
|
|
|
|
|
|
# class UserLoginView(APIView):
|
|
# authentication_classes = [JWTAuthentication]
|
|
# permission_classes = [IsAuthenticated]
|
|
|
|
# def post(self, request):
|
|
# # Extract the token from the Authorization header
|
|
# print(request)
|
|
# auth_header = request.headers.get('Authorization')
|
|
# if auth_header is None:
|
|
# return Response({"error":
|
|
# "Authorization header missing"},
|
|
# status=status.HTTP_401_UNAUTHORIZED)
|
|
# try:
|
|
# # The token is expected to be in the format "Bearer <token>"
|
|
# token_key = auth_header.split(' ')[1]
|
|
# token = Token.objects.get(key=token_key)
|
|
# user = token.user
|
|
# return Response({"message": "Login successful", "user_id": user.id, "username": user.username}, status=status.HTTP_200_OK)
|
|
# except (Token.DoesNotExist, IndexError):
|
|
# return Response({"error": "Invalid token"}, status=status.HTTP_401_UNAUTHORIZED)
|
|
|
|
|
|
|
|
class UserLogoutView(GenericAPIView):
|
|
permission_classes = (IsAuthenticated,)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
try:
|
|
refresh_token = request.data("refresh")
|
|
token = RefreshToken(refresh_token)
|
|
token.blacklist()
|
|
return Response( status = status.HTTP_205_RESET_CONTENT)
|
|
except Exception as e:
|
|
return Response( status = status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
# def post(self, request):
|
|
# logout(request)
|
|
# return Response({"message": "Logout successful"}, status=status.HTTP_200_OK)
|
|
|
|
|
|
|
|
|
|
|
|
|