mirror of
https://github.com/lemeow125/DRF_Template.git
synced 2024-11-17 04:09:25 +08:00
102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
from accounts.models import CustomUser
|
|
from django.contrib.auth.password_validation import validate_password
|
|
from django.core import exceptions as django_exceptions
|
|
from django.core.cache import cache
|
|
from djoser.serializers import UserSerializer as BaseUserSerializer
|
|
from rest_framework import serializers
|
|
from rest_framework.serializers import ImageField, ModelSerializer
|
|
from rest_framework.settings import api_settings
|
|
from user_groups.serializers import SimpleUserGroupSerializer
|
|
|
|
# There can be multiple subject instances with the same name, only differing in course, year level, and semester. We filter them here
|
|
|
|
|
|
class SimpleCustomUserSerializer(ModelSerializer):
|
|
class Meta(BaseUserSerializer.Meta):
|
|
model = CustomUser
|
|
fields = ("id", "username", "email", "full_name")
|
|
|
|
|
|
class CustomUserSerializer(BaseUserSerializer):
|
|
avatar = ImageField()
|
|
|
|
class Meta(BaseUserSerializer.Meta):
|
|
model = CustomUser
|
|
fields = (
|
|
"id",
|
|
"username",
|
|
"email",
|
|
"avatar",
|
|
"first_name",
|
|
"is_new",
|
|
"last_name",
|
|
"user_group",
|
|
"group_member",
|
|
"group_owner",
|
|
)
|
|
read_only_fields = (
|
|
"id",
|
|
"username",
|
|
"email",
|
|
"user_group",
|
|
"group_member",
|
|
"group_owner",
|
|
)
|
|
|
|
def to_representation(self, instance):
|
|
representation = super().to_representation(instance)
|
|
representation["user_group"] = SimpleUserGroupSerializer(
|
|
instance.user_group, many=False
|
|
).data
|
|
return representation
|
|
|
|
def update(self, instance, validated_data):
|
|
cache.delete(f"user:{instance.id}")
|
|
return super().update(instance, validated_data)
|
|
|
|
|
|
class UserRegistrationSerializer(serializers.ModelSerializer):
|
|
email = serializers.EmailField(required=True)
|
|
username = serializers.CharField(required=True)
|
|
password = serializers.CharField(
|
|
write_only=True, style={"input_type": "password", "placeholder": "Password"}
|
|
)
|
|
first_name = serializers.CharField(
|
|
required=True, allow_blank=False, allow_null=False
|
|
)
|
|
last_name = serializers.CharField(
|
|
required=True, allow_blank=False, allow_null=False
|
|
)
|
|
|
|
class Meta:
|
|
model = CustomUser
|
|
fields = ["email", "username", "password", "first_name", "last_name"]
|
|
|
|
def validate(self, attrs):
|
|
user_attrs = attrs.copy()
|
|
user = self.Meta.model(**user_attrs)
|
|
password = attrs.get("password")
|
|
|
|
try:
|
|
validate_password(password, user)
|
|
except django_exceptions.ValidationError as e:
|
|
serializer_error = serializers.as_serializer_error(e)
|
|
errors = serializer_error[api_settings.NON_FIELD_ERRORS_KEY]
|
|
if len(errors) > 1:
|
|
raise serializers.ValidationError({"password": errors[0]})
|
|
else:
|
|
raise serializers.ValidationError({"password": errors})
|
|
if self.Meta.model.objects.filter(username=attrs.get("username")).exists():
|
|
raise serializers.ValidationError(
|
|
"A user with that username already exists."
|
|
)
|
|
return super().validate(attrs)
|
|
|
|
def create(self, validated_data):
|
|
user = self.Meta.model(**validated_data)
|
|
user.username = validated_data["username"]
|
|
user.is_active = False
|
|
user.set_password(validated_data["password"])
|
|
user.save()
|
|
|
|
return user
|