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

45
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 return super(RegisterView, self).dispatch(*args, **kwargs)
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): def get_response_data(self, user):
if allauth_settings.EMAIL_VERIFICATION == \ if allauth_settings.EMAIL_VERIFICATION == \
@ -60,14 +53,21 @@ 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):
serializer = self.get_serializer(data=request.data) # Check if registration is open
serializer.is_valid(raise_exception=True) if get_adapter(self.request).is_open_for_signup(self.request):
user = self.perform_create(serializer) serializer = self.get_serializer(data=request.data)
headers = self.get_success_headers(serializer.data) serializer.is_valid(raise_exception=True)
user = self.perform_create(serializer)
headers = self.get_success_headers(serializer.data)
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)
@ -117,23 +117,12 @@ class SocialLoginView(LoginView):
class FacebookLogin(SocialLoginView): class FacebookLogin(SocialLoginView):
adapter_class = FacebookOAuth2Adapter adapter_class = FacebookOAuth2Adapter
client_class = OAuth2Client client_class = OAuth2Client
callback_url = 'localhost:8000' callback_url = 'localhost:8000'
------------- -------------
""" """
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)