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() 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": "فرمت شماره موبایل نادرست است."}) if password != re_password: raise serializers.ValidationError({"password": "رمز عبور مطابقت ندارد."}) 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("شماره موبایل معتبر نیست.") 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("فرمت شماره موبایل نادرست است.") 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("فرمت شماره موبایل نادرست است.") 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("فرمت شماره موبایل نادرست است.") 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(self, data): if data.get("password") != data.get("re_password"): raise serializers.ValidationError({"password": "رمز عبور مطابقت ندارد."}) 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": "رمز عبور جدید و تکرار آن مطابقت ندارند."}) 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", )