mirror of
https://github.com/IRBorisov/ConceptPortal.git
synced 2025-06-26 04:50:36 +03:00
138 lines
4.2 KiB
Python
138 lines
4.2 KiB
Python
''' Serializers: User profile and Authentification. '''
|
|
from django.contrib.auth import authenticate
|
|
from django.contrib.auth.password_validation import validate_password
|
|
from rest_framework import serializers
|
|
|
|
from apps.rsform.models import Subscription
|
|
from . import models
|
|
|
|
|
|
class LoginSerializer(serializers.Serializer):
|
|
''' Serializer: User authentification by login/password. '''
|
|
# TODO: declare schema
|
|
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.get('username')
|
|
password = attrs.get('password')
|
|
user = authenticate(
|
|
request=self.context.get('request'),
|
|
username=username,
|
|
password=password
|
|
)
|
|
if not user:
|
|
msg = 'Неправильное сочетание имени пользователя и пароля.'
|
|
raise serializers.ValidationError(msg, code='authorization')
|
|
attrs['user'] = user
|
|
return attrs
|
|
|
|
def create(self, validated_data):
|
|
raise NotImplementedError('unexpected `create()` call')
|
|
|
|
def update(self, instance, validated_data):
|
|
raise NotImplementedError('unexpected `update()` call')
|
|
|
|
|
|
class AuthSerializer(serializers.Serializer):
|
|
''' Serializer: Authentication data. '''
|
|
# TODO: declare schema
|
|
def to_representation(self, instance: models.User) -> dict:
|
|
if instance.is_anonymous:
|
|
return {
|
|
'id': None,
|
|
'username': '',
|
|
'is_staff': False,
|
|
'subscriptions': []
|
|
}
|
|
else:
|
|
return {
|
|
'id': instance.pk,
|
|
'username': instance.username,
|
|
'is_staff': instance.is_staff,
|
|
'subscriptions': [sub.item.pk for sub in Subscription.objects.filter(user=instance)]
|
|
}
|
|
|
|
|
|
class UserInfoSerializer(serializers.ModelSerializer):
|
|
''' Serializer: User data. '''
|
|
class Meta:
|
|
''' serializer metadata. '''
|
|
model = models.User
|
|
fields = [
|
|
'id',
|
|
'username',
|
|
'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',
|
|
]
|
|
|
|
|
|
class ChangePasswordSerializer(serializers.Serializer):
|
|
''' Serializer: Change password. '''
|
|
old_password = serializers.CharField(required=True)
|
|
new_password = serializers.CharField(required=True)
|
|
|
|
def create(self, validated_data):
|
|
raise NotImplementedError('unexpected `create()` call')
|
|
|
|
def update(self, instance, validated_data):
|
|
raise NotImplementedError('unexpected `update()` call')
|
|
|
|
|
|
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": "Введенные пароли не совпадают"})
|
|
return attrs
|
|
|
|
def create(self, validated_data):
|
|
user = models.User.objects.create_user(
|
|
validated_data['username'], validated_data['email'], validated_data['password']
|
|
)
|
|
user.first_name = validated_data['first_name']
|
|
user.last_name = validated_data['last_name']
|
|
user.save()
|
|
return user
|