mirror of
https://github.com/Tivix/django-rest-auth.git
synced 2024-11-11 03:36:35 +03:00
105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
from django.conf import settings
|
|
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework.permissions import AllowAny
|
|
from rest_framework.generics import CreateAPIView
|
|
from rest_framework import status
|
|
from rest_framework.exceptions import MethodNotAllowed
|
|
|
|
from allauth.account.views import ConfirmEmailView
|
|
from allauth.account.utils import complete_signup
|
|
from allauth.account import app_settings as allauth_settings
|
|
|
|
from rest_auth.app_settings import (TokenSerializer,
|
|
JWTSerializer,
|
|
create_token)
|
|
from rest_auth.registration.serializers import (SocialLoginSerializer,
|
|
VerifyEmailSerializer)
|
|
from rest_auth.views import LoginView
|
|
from rest_auth.models import TokenModel
|
|
from .app_settings import RegisterSerializer
|
|
|
|
from rest_auth.utils import jwt_encode
|
|
|
|
class RegisterView(CreateAPIView):
|
|
serializer_class = RegisterSerializer
|
|
permission_classes = (AllowAny, )
|
|
token_model = TokenModel
|
|
|
|
def get_response_data(self, user):
|
|
if allauth_settings.EMAIL_VERIFICATION == \
|
|
allauth_settings.EmailVerificationMethod.MANDATORY:
|
|
return {}
|
|
|
|
if getattr(settings, 'REST_USE_JWT', False):
|
|
data = {
|
|
'user': user,
|
|
'token': self.token
|
|
}
|
|
return JWTSerializer(data).data
|
|
else:
|
|
return TokenSerializer(user.auth_token).data
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.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), status=status.HTTP_201_CREATED, headers=headers)
|
|
|
|
def perform_create(self, serializer):
|
|
user = serializer.save(self.request)
|
|
if getattr(settings, 'REST_USE_JWT', False):
|
|
self.token = jwt_encode(user)
|
|
else:
|
|
create_token(self.token_model, user, serializer)
|
|
complete_signup(self.request._request, user,
|
|
allauth_settings.EMAIL_VERIFICATION,
|
|
None)
|
|
return user
|
|
|
|
|
|
class VerifyEmailView(APIView, ConfirmEmailView):
|
|
|
|
permission_classes = (AllowAny,)
|
|
allowed_methods = ('POST', 'OPTIONS', 'HEAD')
|
|
|
|
def get(self, *args, **kwargs):
|
|
raise MethodNotAllowed('GET')
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
serializer = VerifyEmailSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
self.kwargs['key'] = serializer.validated_data['key']
|
|
confirmation = self.get_object()
|
|
confirmation.confirm(self.request)
|
|
return Response({'message': 'ok'}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class SocialLoginView(LoginView):
|
|
"""
|
|
class used for social authentications
|
|
example usage for facebook with access_token
|
|
-------------
|
|
from allauth.socialaccount.providers.facebook.views import FacebookOAuth2Adapter
|
|
|
|
class FacebookLogin(SocialLoginView):
|
|
adapter_class = FacebookOAuth2Adapter
|
|
-------------
|
|
|
|
example usage for facebook with code
|
|
|
|
-------------
|
|
from allauth.socialaccount.providers.facebook.views import FacebookOAuth2Adapter
|
|
from allauth.socialaccount.providers.oauth2.client import OAuth2Client
|
|
|
|
class FacebookLogin(SocialLoginView):
|
|
adapter_class = FacebookOAuth2Adapter
|
|
client_class = OAuth2Client
|
|
callback_url = 'localhost:8000'
|
|
-------------
|
|
"""
|
|
|
|
serializer_class = SocialLoginSerializer |