from django.contrib.auth import get_user_model, password_validation from django.core.exceptions import ValidationError as DjangoValidationError 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 = "\u0641\u0631\u0645\u062a \u0634\u0645\u0627\u0631\u0647 \u0645\u0648\u0628\u0627\u06cc\u0644 \u0646\u0627\u062f\u0631\u0633\u062a \u0627\u0633\u062a." INVALID_MOBILE_NUMBER_MESSAGE = "\u0634\u0645\u0627\u0631\u0647 \u0645\u0648\u0628\u0627\u06cc\u0644 \u0645\u0639\u062a\u0628\u0631 \u0646\u06cc\u0633\u062a." PASSWORD_MISMATCH_MESSAGE = "\u0631\u0645\u0632 \u0639\u0628\u0648\u0631 \u0645\u0637\u0627\u0628\u0642\u062a \u0646\u062f\u0627\u0631\u062f." NEW_PASSWORD_MISMATCH_MESSAGE = "\u0631\u0645\u0632 \u0639\u0628\u0648\u0631 \u062c\u062f\u06cc\u062f \u0648 \u062a\u06a9\u0631\u0627\u0631 \u0622\u0646 \u0645\u0637\u0627\u0628\u0642\u062a \u0646\u062f\u0627\u0631\u0646\u062f." PASSWORD_REUSE_MESSAGE = "\u0631\u0645\u0632 \u0639\u0628\u0648\u0631 \u062c\u062f\u06cc\u062f \u0646\u0628\u0627\u06cc\u062f \u0628\u0627 \u0631\u0645\u0632 \u0639\u0628\u0648\u0631 \u0642\u0628\u0644\u06cc \u06cc\u06a9\u0633\u0627\u0646 \u0628\u0627\u0634\u062f." def _raise_password_validation_error(password: str, *, user, field_name: str) -> None: try: password_validation.validate_password(password, user=user) except DjangoValidationError as exc: raise serializers.ValidationError({field_name: exc.messages[0] if len(exc.messages) == 1 else exc.messages}) 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): mobile = data.get("mobile", "") password = data.get("password", "") re_password = data.get("re_password", "") if password != re_password: raise serializers.ValidationError({"password": PASSWORD_MISMATCH_MESSAGE}) user = User.objects.filter(mobile=mobile).only("password", "mobile", "first_name", "last_name", "email").first() if user is not None: _raise_password_validation_error(password, user=user, field_name="password") if user.check_password(password): raise serializers.ValidationError({"password": PASSWORD_REUSE_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): old_password = data.get("old_password", "") new_password = data.get("new_password", "") re_password = data.get("re_password", "") if new_password != re_password: raise serializers.ValidationError({"new_password": NEW_PASSWORD_MISMATCH_MESSAGE}) request = self.context.get("request") user = getattr(request, "user", None) if old_password and old_password == new_password: raise serializers.ValidationError({"new_password": PASSWORD_REUSE_MESSAGE}) if user is not None and getattr(user, "is_authenticated", False): _raise_password_validation_error(new_password, user=user, field_name="new_password") 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", )