qlockify-backend-deployment/apps/users/api/serializers.py

from django.contrib.auth import get_user_model
from drf_spectacular.utils import extend_schema_serializer
from rest_framework import serializers

from core.serializers.base import BaseModelSerializer

User = get_user_model()

INVALID_MOBILE_FORMAT_MESSAGE = "فرمت شماره موبایل نادرست است."
INVALID_MOBILE_NUMBER_MESSAGE = "شماره موبایل معتبر نیست."
PASSWORD_MISMATCH_MESSAGE = "رمز عبور مطابقت ندارد."
NEW_PASSWORD_MISMATCH_MESSAGE = "رمز عبور جدید و تکرار آن مطابقت ندارند."


class UserProfilePictureSerializer(BaseModelSerializer):
    class Meta:
        model = User
        fields = BaseModelSerializer.Meta.fields + ("profile_picture",)


class UserListSerializer(BaseModelSerializer):
    full_name = serializers.CharField(read_only=True)

    class Meta:
        model = User
        fields = BaseModelSerializer.Meta.fields + (
            "mobile",
            "full_name",
            "profile_picture",
        )


class RegisterSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    code = serializers.CharField(max_length=6)
    password = serializers.CharField(write_only=True)
    re_password = serializers.CharField(write_only=True)
    first_name = serializers.CharField(max_length=100, required=False, allow_blank=True)
    last_name = serializers.CharField(max_length=100, required=False, allow_blank=True)

    def validate(self, data):
        mobile = data.get("mobile", "")
        password = data.get("password", "")
        re_password = data.get("re_password", "")

        if not (mobile.isdigit() and len(mobile) == 11):
            raise serializers.ValidationError({"mobile": INVALID_MOBILE_FORMAT_MESSAGE})

        if password != re_password:
            raise serializers.ValidationError({"password": PASSWORD_MISMATCH_MESSAGE})

        return data


@extend_schema_serializer(component_name="UsersSendOTP")
class SendOTPSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    mode = serializers.ChoiceField(choices=["register", "login", "forget_password"])

    def validate_mobile(self, value):
        if not value.isdigit() or len(value) != 11 or not value.startswith("09"):
            raise serializers.ValidationError(INVALID_MOBILE_NUMBER_MESSAGE)
        return value


@extend_schema_serializer(component_name="UsersLoginOtp")
class LoginOtpSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    code = serializers.CharField(max_length=6)

    def validate_mobile(self, value):
        if not (value.isdigit() and len(value) == 11):
            raise serializers.ValidationError(INVALID_MOBILE_FORMAT_MESSAGE)
        return value


class LoginSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    password = serializers.CharField(write_only=True)

    def validate_mobile(self, value):
        if not (value.isdigit() and len(value) == 11):
            raise serializers.ValidationError(INVALID_MOBILE_FORMAT_MESSAGE)
        return value


class GoogleOAuthFlowSerializer(serializers.Serializer):
    flow = serializers.CharField()


class GoogleOAuthCompleteSerializer(serializers.Serializer):
    flow = serializers.CharField()
    mobile = serializers.CharField(max_length=11)

    def validate_mobile(self, value):
        normalized = "".join(ch for ch in value if ch.isdigit())
        if len(normalized) != 11 or not normalized.startswith("09"):
            raise serializers.ValidationError(INVALID_MOBILE_FORMAT_MESSAGE)
        return normalized


class GoogleOAuthClaimVerifySerializer(serializers.Serializer):
    flow = serializers.CharField()
    code = serializers.CharField(max_length=6)


class ResetPasswordSerializer(serializers.Serializer):
    mobile = serializers.CharField(max_length=11)
    code = serializers.CharField(max_length=6)
    password = serializers.CharField(write_only=True)
    re_password = serializers.CharField(write_only=True)

    def validate_mobile(self, value):
        if not value.isdigit() or len(value) != 11 or not value.startswith("09"):
            raise serializers.ValidationError(INVALID_MOBILE_NUMBER_MESSAGE)
        return value

    def validate(self, data):
        if data.get("password") != data.get("re_password"):
            raise serializers.ValidationError({"password": PASSWORD_MISMATCH_MESSAGE})
        return data


class ChangePasswordSerializer(serializers.Serializer):
    old_password = serializers.CharField(required=True, write_only=True)
    new_password = serializers.CharField(required=True, write_only=True)
    re_password = serializers.CharField(required=True, write_only=True)

    def validate(self, data):
        if data.get("new_password") != data.get("re_password"):
            raise serializers.ValidationError({"new_password": NEW_PASSWORD_MISMATCH_MESSAGE})

        return data


class LogoutSerializer(serializers.Serializer):
    refresh = serializers.CharField()


class TokenPairSerializer(serializers.Serializer):
    access = serializers.CharField()
    refresh = serializers.CharField()


class RegisterWithPasswordSerializer(serializers.Serializer):
    mobile = serializers.CharField()
    password = serializers.CharField()


class UserProfileSerializer(BaseModelSerializer):
    full_name = serializers.ReadOnlyField()
    age = serializers.ReadOnlyField()

    class Meta:
        model = User
        fields = BaseModelSerializer.Meta.fields + (
            "mobile",
            "email",
            "first_name",
            "last_name",
            "description",
            "profile_picture",
            "birth_date",
            "is_verified",
            "full_name",
            "age",
        )
        read_only_fields = BaseModelSerializer.Meta.fields + ("mobile", "is_verified")


class UserSearchSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = (
            "id",
            "first_name",
            "last_name",
            "mobile",
            "profile_picture",
        )