26 lines
1014 B
Python
26 lines
1014 B
Python
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from .serializers import RegisterSerializer, CustomTokenObtainPairSerializer
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework_simplejwt.views import TokenObtainPairView
|
|
|
|
class RegisterView(APIView):
|
|
def post(self, request):
|
|
serializer = RegisterSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
user = serializer.save()
|
|
return Response({"message": "User registered successfully."}, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class MeView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get(self, request):
|
|
user = request.user
|
|
serializer = RegisterSerializer(user)
|
|
return Response(serializer.data)
|
|
|
|
class CustomTokenObtainPairView(TokenObtainPairView):
|
|
serializer_class = CustomTokenObtainPairSerializer |