"""Serializers for account web""" from django.conf import settings from django.contrib.auth import password_validation as password_validators from rest_framework import serializers from account import models from account import tasks from utils import exceptions as utils_exceptions class PasswordResetSerializer(serializers.ModelSerializer): """Serializer from model PasswordReset""" email = serializers.EmailField(required=False, write_only=True) class Meta: """Meta class""" model = models.ResetPasswordToken fields = ( 'email', ) def validate(self, attrs): """Override validate method""" user = self.context.get('request').user email = attrs.get('email') if not user.is_anonymous: attrs['user'] = user else: # Check user in DB user_qs = models.User.objects.filter(email=email) if user_qs.exists(): attrs['user'] = user_qs.first() else: raise utils_exceptions.UserNotFoundError() return attrs def create(self, validated_data, *args, **kwargs): """Override create method""" user = validated_data.pop('user') ip_address = self.context.get('request').META.get('REMOTE_ADDR') obj = models.ResetPasswordToken.objects.create( user=user, ip_address=ip_address, source=models.ResetPasswordToken.WEB ) if settings.USE_CELERY: tasks.send_reset_password_email.delay(obj.id) else: tasks.send_reset_password_email(obj.id) return obj class PasswordResetConfirmSerializer(serializers.ModelSerializer): """Serializer for model User""" password = serializers.CharField(write_only=True) class Meta: """Meta class""" model = models.ResetPasswordToken fields = ('password', ) def validate(self, attrs): """Override validate method""" user = self.instance.user password = attrs.get('password') try: # Compare new password with the old ones if user.check_password(raw_password=password): raise utils_exceptions.PasswordsAreEqual() # Validate password password_validators.validate_password(password=password) except serializers.ValidationError as e: raise serializers.ValidationError(str(e)) else: return attrs def update(self, instance, validated_data): """Override update method""" # Update user password from instance instance.user.set_password(validated_data.get('password')) instance.user.save() # Overdue instance instance.overdue() return instance