mirror of
https://github.com/ramvignesh-b/pi-ku.git
synced 2026-05-04 15:56:56 +00:00
22 lines
597 B
Python
22 lines
597 B
Python
from django.contrib.auth import get_user_model
|
|
from rest_framework import generics, permissions
|
|
|
|
from .serializers import UserSerializer
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class RegisterView(generics.CreateAPIView):
|
|
queryset = User.objects.all()
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = UserSerializer
|
|
|
|
|
|
class MeView(generics.RetrieveAPIView):
|
|
serializer_class = UserSerializer
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
|
|
def get_object(self):
|
|
# Returns the user associated with the JWT token in the request
|
|
return self.request.user
|