（生鲜项目）18. 用户手机号注册 (code验证 + 信号量 + 和前端Vue联调)

from rest_framework import serializers
from django.contrib.auth import get_user_model
from rest_framework.validators import UniqueValidator  # 给username字段做查重用

User = get_user_model()  # 可以获取数据库的userprofile表
# 验证码校验
class UserRegSerializer(serializers.ModelSerializer):
    # 由于userprofile并没有code字段,所以先加进来
    code = serializers.CharField(required=True, max_length=4, min_length=4, label="短信验证码",
                                 # 自定义的错误提示信息
                                 error_messages={
                                     "blank":"你的验证码哪儿去了",  # blank针对的是有字段名,但没有字段值
                                     "required":"请输入验证码",  # required针对的是连字段名都没有
                                     "max_length":"验证码长度错误",
                                     "min_length":"验证码长度错误",
                                 },
                                 help_text="验证码")

    # 虽然在发送验证码阶段, 已经验证了手机号码是否存在,但在这里由于前端post的是用户名,所以也要对用户名进行验证,校验是否已经存在
    username = serializers.CharField(required=True,allow_blank=False, label="用户名",
                                     validators=[UniqueValidator(queryset=User.objects.all(),message="用户已经存在")])

    # style可以将密码改成密文格式显示
    password = serializers.CharField(
        style={'input_type': 'password'},
        label="密码"
    )

    # 局部钩子, 只对code进行校验
    def validate_code(self, code):
        # POST过来的数据封装在self.initial_data里, 且传递过来手机号的key是username
        # 一定要按时间排序,因为我们只会取最后一条code来验证, order_by默认是升序
        verify_records = VerifyCode.objects.filter(mobile=self.initial_data["username"]).order_by("-add_time")

        # 如果验证码存在
        if verify_records:
            # 取最新的那一条验证码
            last_records = verify_records[0]

            # 验证码有效期为5min
            five_minute_ago = datetime.now() - timedelta(hours=0, minutes=5, seconds=0)
            if five_minute_ago > last_records.add_time:
                raise serializers.ValidationError("验证码过期")

            # 验证码核对
            if last_records.code != code:
                raise serializers.ValidationError("验证码错误")

        # 如果验证码不存在, 则直接报错
        else:
            raise serializers.ValidationError("验证码错误")

    # 全局钩子, 目的是过滤掉code字段(code其实只用于验证,而不用存在userprofile表中)
    # attrs是局部钩子清洗后的所有字段的dict
    def validate(self, attrs):
        attrs["mobile"] = attrs["username"]
        del attrs["code"]
        return attrs

    class Meta:
        model = User
        fields = ("username", "code", "mobile","password")

# 用户注册逻辑
class UserViewset(CreateModelMixin,viewsets.GenericViewSet):
    serializer_class=UserRegSerializer

# 用户注册接口(包含code验证,username去重验证)
router.register(r'users', UserViewset, basename="users")

code = serializers.CharField(...
                                 write_only=True,  # 有了这个就解决了'UserProfile' object has no attribute 'code' 的问题
                                                   # 原理就是执行serializer.data时,该字段不会被序列化
                            ...)

    password = serializers.CharField(
　　　　 ...
        write_only=True,  # 让密码不返回
    )

    # 让密码在保存的时候加密 (拦截create方法, 在其屁股后面加上密码加密)
    def create(self, validated_data):
        user = super(UserRegSerializer, self).create(validated_data=validated_data)
        user.set_password(validated_data["password"])
        user.save()
        return user

from django.db.models.signals import post_save
from django.dispatch import receiver
from rest_framework.authtoken.models import Token
from django.contrib.auth import get_user_model

User = get_user_model()  # 可以获取数据库的userprofile表


@receiver(post_save, sender=User)  # sender即我们接收哪个Model传递过来的
def create_auth_token(sender, instance=None, created=False, **kwargs):
    if created:  # created即只有当检测到是新建用户时, 才执行以下逻辑
        password = instance.password
        instance.set_password(password)
        instance.save()
        # Token.objects.create(user=instance) # 我们已经使用了JWT模式, 就无须再创建Token

from django.apps import AppConfig


class UsersConfig(AppConfig):
    name = 'users'
    verbose_name = "用户管理"

    def ready(self):
        import users.signals

from rest_framework_jwt.serializers import jwt_payload_handler, jwt_encode_handler

# 用户注册逻辑
class UserViewset(CreateModelMixin, viewsets.GenericViewSet):
    serializer_class = UserRegSerializer

    # 重写create函数,使得前端注册成功后,后端可以返回该用户的token,好让前端直接处于一个登陆的状态
    # 同时,要明白后端是如何根据user生成token的
    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        user = self.perform_create(serializer)

        # 给serializer.data封装上token字段,返给前端用的
        re_dict = serializer.data
        payload = jwt_payload_handler(user)
        re_dict["token"] = jwt_encode_handler(payload)
        re_dict["name"] = user.name if user.name else user.username

        headers = self.get_success_headers(serializer.data)
        return Response(re_dict, status=status.HTTP_201_CREATED, headers=headers)

    # 源码中的该函数并没有返回值,而这里我们需要其返回user
    def perform_create(self, serializer):
        return serializer.save() # 知识点: serializer.save()返回的就是它对应的实例,这里即user