This commit is contained in:
Philippe Luickx 2017-07-03 11:20:57 +03:00
parent 55fa1bbe0e
commit d6ff813e7a

View File

@ -36,14 +36,7 @@ class RegisterView(CreateAPIView):
@sensitive_post_parameters_m
def dispatch(self, *args, **kwargs):
# Check if registration is open
if get_adapter(self.request).is_open_for_signup(self.request):
return super(RegisterView, self).dispatch(*args, **kwargs)
else:
return Response(
data={'message': 'Registration is not open.'},
status=status.HTTP_403_FORBIDDEN,
)
def get_response_data(self, user):
if allauth_settings.EMAIL_VERIFICATION == \
@ -60,6 +53,8 @@ class RegisterView(CreateAPIView):
return TokenSerializer(user.auth_token).data
def create(self, request, *args, **kwargs):
# Check if registration is open
if get_adapter(self.request).is_open_for_signup(self.request):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
user = self.perform_create(serializer)
@ -68,6 +63,11 @@ class RegisterView(CreateAPIView):
return Response(self.get_response_data(user),
status=status.HTTP_201_CREATED,
headers=headers)
else:
return Response(
data={'message': 'Registration is not open.'},
status=status.HTTP_403_FORBIDDEN,
)
def perform_create(self, serializer):
user = serializer.save(self.request)
@ -124,16 +124,5 @@ class SocialLoginView(LoginView):
serializer_class = SocialLoginSerializer
@sensitive_post_parameters_m
def dispatch(self, *args, **kwargs):
# Check if registration is open
if get_adapter(self.request).is_open_for_signup(self.request):
return super(SocialLoginView, self).dispatch(*args, **kwargs)
else:
return Response(
data={'message': 'Registration is not open.'},
status=status.HTTP_403_FORBIDDEN,
)
def process_login(self):
get_adapter(self.request).login(self.request, self.user)