216 lines
8.0 KiB
Python
216 lines
8.0 KiB
Python
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",
|
|
)
|