Luffy /5/ 短信登陆接口&短信注册接口

短信登录接口

视图

class LoginView(GenericViewSet):
    serializer_class = MulLoginSerializer
    queryset = User

    # 两个登陆方式都写在这里面（多方式，一个是验证码登陆）
    # login不是保存，但是用post，咱们的想法是把验证逻辑写到序列化类中
    @action(methods=["post"], detail=False)
    def mul_login(self, request):
        return self._common_login(request)


    # 127.0.0.1:8000/api/v1/user/login/sms_login
    @action(methods=["post"], detail=False)
    def sms_login(self, request):
        # 默认情况下使用的序列化类使用的是MulLoginSerializer---》多方式登陆的逻辑-->不符合短信登陆逻辑
        # 再新写一个序列化类，给短信登陆用
        return self._common_login(request)
    def get_serializer_class(self):
        # 方式一：
        # if 'mul_login' in self.request.path:
        #     return self.serializer_class
        # else:
        #     return SmsLoginSerializer
        # 方式二
        if self.action=='mul_login':
            return self.serializer_class
        else:
            return SmsLoginSerializer


    def _common_login(self,request):
        try:
            # 序列化类在变
            ser = self.get_serializer(data=request.data, context={'request': request})
            ser.is_valid(raise_exception=True)  # 如果校验失败，直接抛异常，不需要加if判断了
            token = ser.context.get('token')
            username = ser.context.get('username')
            icon = ser.context.get('icon')
            return APIResponse(token=token, username=username, icon=icon)  # {code:100,msg:成功，token:dsadsf,username:lqz}
        except Exception as e:
            raise APIException(str(e))

序列化类

# 只用来做反序列化，短信登陆
class SmsLoginSerializer(serializers.ModelSerializer):
    code = serializers.CharField(max_length=4, min_length=4)  # 字段自己的规则
    mobile = serializers.CharField(max_length=11, min_length=11)  # 一定要重写，不重写，字段自己的校验过不去，就到不了全局钩子

    class Meta:
        model = User
        fields = ['mobile', 'code']  # code不在表中，它是验证码，要重新

    def validate(self, attrs):
        # 1 验证手机号是否和合法 验证code是否合法---》去缓存中取出来判断
        self._check_code(attrs)
        # 2 根据手机号获取用户---》需要密码吗？不需要
        user = self._get_user(attrs)
        # 3 签发token
        token = self._get_token(user)
        # 4 把token，username，icon放到context中
        request = self.context['request']
        self.context['token'] = token
        self.context['username'] = user.username
        self.context['icon'] = 'http://%s/media/%s' % (request.META['HTTP_HOST'], str(user.icon))
        return attrs

    def _check_code(self, attrs):
        mobile = attrs.get('mobile')
        new_code = attrs.get('code')
        if mobile:
            # 验证验证码是否正确
            old_code = cache.get('sms_cache_%s' % mobile)
            if new_code != old_code:
                raise ValidationError('验证码错误')

        else:
            raise ValidationError('手机号没有带')

    def _get_user(self, attrs):
        mobile = attrs.get('mobile')
        # return User.objects.get(mobile=mobile)
        user = User.objects.filter(mobile=mobile).first()
        if user:
            return user
        else:
            raise ValidationError("该用户不存在")

    def _get_token(self, user):
        # jwt模块中提供的
        from rest_framework_jwt.serializers import jwt_payload_handler, jwt_encode_handler
        payload = jwt_payload_handler(user)
        token = jwt_encode_handler(payload)
        return token

短信注册接口

路由

# 127.0.0.1:8000/api/v1/user/register --- >post请求
router.register('register',RegisterView , 'register')

视图类

from rest_framework.mixins import CreateModelMixin
class RegisterView(GenericViewSet,CreateModelMixin):
    serializer_class = RegisterSerializer
    queryset = User.objects.all()
    def create(self, request, *args, **kwargs):
        # 方式一：
        super().create(request, *args, **kwargs)  # 小问题,code不是表的字段，需要用write_only

        # 方式二：
        # serializer = self.get_serializer(data=request.data)
        # serializer.is_valid(raise_exception=True)
        # # self.perform_create(serializer)
        # serializer.save()

        return APIResponse(msg='注册成功')

序列化类

# 主要用来做反序列化，数据校验----》其实序列化是用不到的，但是create源码中只要写了serializer.data，就会用序列化
class RegisterSerializer(serializers.ModelSerializer):
    code = serializers.CharField(max_length=4, min_length=4,write_only=True)

    class Meta:
        model = User
        fields = ['mobile', 'code', 'password']
        extra_kwargs = {
            'password': {'write_only': True},
        }

    def validate(self, attrs):
        # 1 校验手机号和验证码
        self._check_code(attrs)
        # 2 就可以新增了---》User中字段很多，现在只带了俩字段，
        #   username必填随机生成，code不存表，剔除，
        #  存user表，不能使用默认的create，一定要重写create方法
        self._per_save(attrs)
        return attrs

    # 校验手机号
    def validate_mobile(self, value):  # 局部钩子
        if not re.match(r'^1[3-9][0-9]{9}$', value):
            raise ValidationError('手机号不合法')
        return value

    # 入库前准备
    def _per_save(self, attrs):
        # 剔除code，
        attrs.pop('code')
        # 新增username-->用手机号作为用户名
        attrs['username'] = attrs.get('mobile')

    # 写成公共函数，传入手机号，就校验验证码
    # 经常公司中为了省短信，回留万能验证码，8888
    def _check_code(self, attrs):
        # 校验code
        new_code = attrs.get('code')
        mobile = attrs.get('mobile')
        old_code = cache.get('sms_cache_%s' % mobile)
        if new_code != old_code:
            raise ValidationError("验证码错误")

    def create(self, validated_data):
        # 如果补充些，密码不是密文
        user = User.objects.create_user(**validated_data)
        return user

登陆注册前端

# 127.0.0.1:8080 和localhost:8080 和 192.168.31.226:8080


# 前端可以存数据的地方
	-存到cookie中，js操作，在vue中可以借助vue-cookies第三方插件
    下载：
  	-cnpm install vue-cookies -S
    导入：
	-main.js中
    import cookies from 'vue-cookies'
	Vue.prototype.$cookies = cookies;
    -怎么使用？
    this.$cookies.set()
    this.$cookies.get()

  -localStorage，永久存储
      localStorage.setItem('key', 'value');
    	localStorage.key = "value"
    	localStorage["key"] = "value"
  -sessionStorage，临时存储，关闭浏览器就没了
  	sessionStorage.setItem("age",'19')
  

Header.vue

login.vue

register.vue