Tivix/django-rest-auth
1
1
Fork 0
mirror of https://github.com/Tivix/django-rest-auth.git synced 2025-07-22 13:39:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Philippe Luickx 2017-07-03 11:20:57 +03:00
parent 55fa1bbe0e
commit d6ff813e7a
1 changed files with 17 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
rest_auth/registration/views.py
View File

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