ConceptPortal-public/rsconcept/backend/apps/users/serializers.py

''' Serializers: User profile and Authorization. '''
from django.contrib.auth import authenticate
from django.contrib.auth.password_validation import validate_password
from rest_framework import serializers

from apps.library.models import Editor
from shared import messages as msg

from . import models


class NonFieldErrorSerializer(serializers.Serializer):
    ''' Serializer: list of non-field errors. '''
    non_field_errors = serializers.ListField(
        child=serializers.CharField()
    )


class LoginSerializer(serializers.Serializer):
    ''' Serializer: User authentication by login/password. '''
    username = serializers.CharField(
        label='Имя пользователя',
        write_only=True
    )
    password = serializers.CharField(
        label='Пароль',
        style={'input_type': 'password'},
        trim_whitespace=False,
        write_only=True
    )

    def validate(self, attrs):
        username = attrs['username']
        if '@' in username:
            user = models.User.objects.filter(email=username)
            if not user.exists() or user.count() > 1:
                raise serializers.ValidationError(
                    msg.passwordAuthFailed(),
                    code='authorization'
                )
            username = user.first().username
        password = attrs['password']
        authenticated = authenticate(
            request=self.context.get('request'),
            username=username,
            password=password
        )
        if not authenticated:
            raise serializers.ValidationError(
                msg.passwordAuthFailed(),
                code='authorization'
            )
        attrs['user'] = authenticated
        return attrs


class AuthSerializer(serializers.Serializer):
    ''' Serializer: Authorization data. '''
    id = serializers.IntegerField()
    username = serializers.CharField()
    is_staff = serializers.BooleanField()

    def to_representation(self, instance: models.User) -> dict:
        if instance.is_anonymous:
            return {
                'id': None,
                'username': '',
                'is_staff': False,
                'editor': []
            }
        else:
            return {
                'id': instance.pk,
                'username': instance.username,
                'is_staff': instance.is_staff,
                'editor': [edit.item.pk for edit in Editor.objects.filter(editor=instance)]
            }


class UserInfoSerializer(serializers.ModelSerializer):
    ''' Serializer: User data. '''
    class Meta:
        ''' serializer metadata. '''
        model = models.User
        fields = [
            'id',
            'first_name',
            'last_name',
        ]


class UserSerializer(serializers.ModelSerializer):
    ''' Serializer: User data. '''
    id = serializers.IntegerField(read_only=True)

    class Meta:
        ''' serializer metadata. '''
        model = models.User
        fields = [
            'id',
            'username',
            'email',
            'first_name',
            'last_name',
        ]

    def validate(self, attrs):
        attrs = super().validate(attrs)
        if 'email' in attrs:
            maybe_user = models.User.objects.filter(email=attrs['email'])
            if maybe_user.exists():
                if maybe_user.count() > 1 or maybe_user.first().pk != self.context['request'].user.pk:
                    raise serializers.ValidationError({
                        'email': msg.emailAlreadyTaken()
                    })
        return attrs


class ChangePasswordSerializer(serializers.Serializer):
    ''' Serializer: Change password. '''
    old_password = serializers.CharField(required=True)
    new_password = serializers.CharField(required=True)


class SignupSerializer(serializers.ModelSerializer):
    ''' Serializer: Create user profile. '''
    id = serializers.IntegerField(read_only=True)
    password = serializers.CharField(write_only=True, required=True, validators=[validate_password])
    password2 = serializers.CharField(write_only=True, required=True)

    class Meta:
        ''' serializer metadata. '''
        model = models.User
        fields = [
            'id',
            'username',
            'email',
            'first_name',
            'last_name',
            'password',
            'password2'
        ]

    def validate(self, attrs):
        if attrs['password'] != attrs['password2']:
            raise serializers.ValidationError({
                'password': msg.passwordsNotMatch()
            })
        if models.User.objects.filter(email=attrs['email']).exists():
            raise serializers.ValidationError({
                'email': msg.emailAlreadyTaken()
            })
        return attrs

    def create(self, validated_data):
        user = models.User.objects.create_user(
            username=validated_data['username'],
            email=validated_data['email'],
            password=validated_data['password']
        )
        user.first_name = validated_data['first_name']
        user.last_name = validated_data['last_name']
        user.save()
        return user