67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
from rest_framework import generics, permissions
|
|
from rest_framework.exceptions import PermissionDenied
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from .models import VendorProfile
|
|
from .serializers import (
|
|
CustomerRegistrationSerializer,
|
|
UserMeSerializer,
|
|
VendorProfileSerializer,
|
|
VendorRegistrationSerializer,
|
|
)
|
|
|
|
|
|
class VendorRegistrationView(generics.CreateAPIView):
|
|
serializer_class = VendorRegistrationSerializer
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
user = serializer.save()
|
|
return Response(UserMeSerializer(user).data, status=201)
|
|
|
|
|
|
class CustomerRegistrationView(generics.CreateAPIView):
|
|
serializer_class = CustomerRegistrationSerializer
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
user = serializer.save()
|
|
return Response(UserMeSerializer(user).data, status=201)
|
|
|
|
|
|
class PasswordResetRequestView(APIView):
|
|
"""Accepts email for UX parity; outbound email is not wired yet."""
|
|
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
def post(self, request):
|
|
return Response(
|
|
{
|
|
"detail": "If an account exists for this email, you will receive password reset instructions once email delivery is enabled."
|
|
},
|
|
status=200,
|
|
)
|
|
|
|
|
|
class MeView(APIView):
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
|
|
def get(self, request):
|
|
return Response(UserMeSerializer(request.user).data)
|
|
|
|
|
|
class VendorProfileMeView(generics.RetrieveUpdateAPIView):
|
|
serializer_class = VendorProfileSerializer
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
|
|
def get_object(self):
|
|
user = self.request.user
|
|
if not user.is_vendor:
|
|
raise PermissionDenied("Only vendors can access vendor profile setup.")
|
|
return VendorProfile.objects.get(user=user)
|