61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
"""Web account views"""
|
|
from django.utils.encoding import force_text
|
|
from django.utils.http import urlsafe_base64_decode
|
|
from rest_framework import permissions
|
|
from rest_framework import status
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework.response import Response
|
|
|
|
from account import models
|
|
from account.serializers import web as serializers
|
|
from utils import exceptions as utils_exceptions
|
|
from utils.models import gm_token_generator
|
|
from utils.views import (JWTCreateAPIView,
|
|
JWTGenericViewMixin)
|
|
|
|
|
|
# Password reset
|
|
class PasswordResetView(JWTCreateAPIView):
|
|
"""View for resetting user password"""
|
|
serializer_class = serializers.PasswordResetSerializer
|
|
queryset = models.ResetPasswordToken.objects.valid()
|
|
|
|
|
|
class PasswordResetConfirmView(JWTGenericViewMixin):
|
|
"""View for confirmation new password"""
|
|
serializer_class = serializers.PasswordResetConfirmSerializer
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method"""
|
|
return models.ResetPasswordToken.objects.valid()
|
|
|
|
def get_object(self):
|
|
"""Override get_object method
|
|
"""
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
uidb64 = self.kwargs.get('uid')
|
|
|
|
uid = force_text(urlsafe_base64_decode(uidb64))
|
|
token = self.kwargs.get('token')
|
|
|
|
filter_kwargs = {'key': token, 'user_id': uid}
|
|
obj = get_object_or_404(queryset, **filter_kwargs)
|
|
|
|
if not gm_token_generator.check_token(user=obj.user, token=token):
|
|
raise utils_exceptions.NotValidTokenError()
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, obj)
|
|
|
|
return obj
|
|
|
|
def put(self, request, *args, **kwargs):
|
|
"""Implement PUT method"""
|
|
instance = self.get_object()
|
|
serializer = self.get_serializer(instance=instance,
|
|
data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
serializer.save()
|
|
return Response(status=status.HTTP_200_OK)
|