phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

apply user role

Browse Source
This commit is contained in:
Kuroshini 2020-02-05 14:23:37 +03:00
parent c9cf2672b6
commit 6d556665e0
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/account/views/web.py
View File

@ -1,5 +1,6 @@
"""Web account views""" """Web account views"""
from django.conf import settings from django.conf import settings
from django.db import transaction
from django.shortcuts import get_object_or_404 from django.shortcuts import get_object_or_404
from django.utils.encoding import force_text from django.utils.encoding import force_text
from django.utils.http import urlsafe_base64_decode from django.utils.http import urlsafe_base64_decode
@ -61,10 +62,18 @@ class PasswordResetConfirmView(JWTGenericViewMixin, generics.GenericAPIView):
def patch(self, request, *args, **kwargs): def patch(self, request, *args, **kwargs):
"""Implement PATCH method""" """Implement PATCH method"""
instance = self.get_object() instance = self.get_object()
serializer = self.get_serializer(instance=instance, with transaction.atomic():
data=request.data) serializer = self.get_serializer(instance=instance,
serializer.is_valid(raise_exception=True) data=request.data)
serializer.save() serializer.is_valid(raise_exception=True)
serializer.save()
# apply requested user_role if 'role' in query_params
if 'role' in request.query_params:
user_role = get_object_or_404(klass=models.UserRole, pk=request.query_params['role'])
user_role.state = models.UserRole.VALIDATED
user_role.save()
# Create tokens # Create tokens
tokens = instance.create_jwt_tokens() tokens = instance.create_jwt_tokens()
return self._put_cookies_in_response( return self._put_cookies_in_response(