Python实战项目-7发送验证码/短信登录/短信注册接口

腾讯云短信开发申请

API和SDK,有SDK优先用SDK
SDK:
3.0版本云操作：云操作的sdk，不仅仅有发送短信，还有云功能的其他功能
2.0版本，简单，只有发送短信功能

# 安装sdk
	-方式一：pip install tencentcloud-sdk-python
    -方式二源码安装：
    	-下载源码
        -执行 python setup.py install

# 发送短信测试

查询手机号是否存在接口

class UserView(ViewSet):
	@action(methods=['get'], detail=False)
    def check_mobile(self, request):
        try:
            # 如果布报错就说明手机号存在
            mobile = request.query_params.get('mobile')
            UserInfo.objects.get(mobile=mobile)
            return APIResponse('手机号存在')
        except Exception as e:
            raise APIException('手机号不存在！')

短信验证码功能

发短信的功能是公共的,我们可以将这个功能封装到公共的目录下面
libs目录下：
send_tx_sms（包）
init.py
settings.py
sms.py
sms.py

# 获取n位随机数字验证码
import json

from utils.common_log import logger
import random
def get_code(num=4): # // 获取随机验证码
    code = ''
    for i in range(num):
        random_num = random.randint(0, 9)
        code += str(random_num)
    return code

from tencentcloud.common import credential
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
import os
# 导入对应产品模块的client models。
from tencentcloud.sms.v20210111 import sms_client, models

# 导入可选配置类
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
# 发送短信的函数
from . import settings
def send_phone_sms(mobile,code): # 将它封装成函数,方便我们在调用的时候使用它
    try:
        # 必要步骤：
        # 实例化一个认证对象，入参需要传入腾讯云账户密钥对secretId，secretKey。
        # 这里采用的是从环境变量读取的方式，需要在环境变量中先设置这两个值。
        # 你也可以直接在代码中写死密钥对，但是小心不要将代码复制、上传或者分享给他人，
        # 以免泄露密钥对危及你的财产安全。
        # SecretId、SecretKey 查询: https://console.cloud.tencent.com/cam/capi
        cred = credential.Credential(settings.SECRET_ID, settings.SECRET_KEY)
        # cred = credential.Credential(
        #     os.environ.get("secretId"),
        #     os.environ.get("secretKey")  #我们可以将 sercetId 和 sercetKey放在环境变量中
                                          # 这样我们的安全性更高
        # )

        # 实例化一个http选项，可选的，没有特殊需求可以跳过。
        httpProfile = HttpProfile()
        # 如果需要指定proxy访问接口，可以按照如下方式初始化hp（无需要直接忽略）
        # httpProfile = HttpProfile(proxy="http://用户名:密码@代理IP:代理端口")
        httpProfile.reqMethod = "POST"  # post请求(默认为post请求)
        httpProfile.reqTimeout = 30  # 请求超时时间，单位为秒(默认60秒)
        httpProfile.endpoint = "sms.tencentcloudapi.com"  # 指定接入地域域名(默认就近接入)

        # 非必要步骤:
        # 实例化一个客户端配置对象，可以指定超时时间等配置
        clientProfile = ClientProfile()
        clientProfile.signMethod = "TC3-HMAC-SHA256"  # 指定签名算法
        clientProfile.language = "en-US"
        clientProfile.httpProfile = httpProfile

        # 实例化要请求产品(以sms为例)的client对象
        # 第二个参数是地域信息，可以直接填写字符串ap-guangzhou，支持的地域列表参考 https://cloud.tencent.com/document/api/382/52071#.E5.9C.B0.E5.9F.9F.E5.88.97.E8.A1.A8
        client = sms_client.SmsClient(cred, "ap-guangzhou", clientProfile)

        # 实例化一个请求对象，根据调用的接口和实际情况，可以进一步设置请求参数
        # 你可以直接查询SDK源码确定SendSmsRequest有哪些属性可以设置
        # 属性可能是基本类型，也可能引用了另一个数据结构
        # 推荐使用IDE进行开发，可以方便的跳转查阅各个接口和数据结构的文档说明
        req = models.SendSmsRequest()

        # 基本类型的设置:
        # SDK采用的是指针风格指定参数，即使对于基本类型你也需要用指针来对参数赋值。
        # SDK提供对基本类型的指针引用封装函数
        # 帮助链接：
        # 短信控制台: https://console.cloud.tencent.com/smsv2
        # 腾讯云短信小助手: https://cloud.tencent.com/document/product/382/3773#.E6.8A.80.E6.9C.AF.E4.BA.A4.E6.B5.81

        # 短信应用ID: 短信SdkAppId在 [短信控制台] 添加应用后生成的实际SdkAppId，示例如1400006666
        # 应用 ID 可前往 [短信控制台](https://console.cloud.tencent.com/smsv2/app-manage) 查看
        req.SmsSdkAppId = settings.AppID
        # 短信签名内容: 使用 UTF-8 编码，必须填写已审核通过的签名
        # 签名信息可前往 [国内短信](https://console.cloud.tencent.com/smsv2/csms-sign) 或 [国际/港澳台短信](https://console.cloud.tencent.com/smsv2/isms-sign) 的签名管理查看
        req.SignName = settings.SignName
        # 模板 ID: 必须填写已审核通过的模板 ID
        # 模板 ID 可前往 [国内短信](https://console.cloud.tencent.com/smsv2/csms-template) 或 [国际/港澳台短信](https://console.cloud.tencent.com/smsv2/isms-template) 的正文模板管理查看
        req.TemplateId = settings.TEMPLATE_ID
        # 模板参数: 模板参数的个数需要与 TemplateId 对应模板的变量个数保持一致，，若无模板参数，则设置为空
        req.TemplateParamSet = [code, '30']
        # 下发手机号码，采用 E.164 标准，+[国家或地区码][手机号]
        # 示例如：+8613711112222， 其中前面有一个+号 ，86为国家码，13711112222为手机号，最多不要超过200个手机号
        req.PhoneNumberSet = ["+86"+ mobile]
        # 用户的 session 内容（无需要可忽略）: 可以携带用户侧 ID 等上下文信息，server 会原样返回
        req.SessionContext = ""
        # 短信码号扩展号（无需要可忽略）: 默认未开通，如需开通请联系 [腾讯云短信小助手]
        req.ExtendCode = ""
        # 国际/港澳台短信 senderid（无需要可忽略）: 国内短信填空，默认未开通，如需开通请联系 [腾讯云短信小助手]
        req.SenderId = ""

        resp = client.SendSms(req)

        # 输出json格式的字符串回包
        # print(resp.to_json_string(indent=2))
        logger.warning(f'发送短信{code},给手机号{mobile}成功！')
        res = json.loads(resp.to_json_string(indent=2))
        # 当出现以下错误码时，快速解决方案参考
        # - [FailedOperation.SignatureIncorrectOrUnapproved](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Afailedoperation.signatureincorrectorunapproved-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
        # - [FailedOperation.TemplateIncorrectOrUnapproved](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Afailedoperation.templateincorrectorunapproved-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
        # - [UnauthorizedOperation.SmsSdkAppIdVerifyFail](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Aunauthorizedoperation.smssdkappidverifyfail-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
        # - [UnsupportedOperation.ContainDomesticAndInternationalPhoneNumber](https://cloud.tencent.com/document/product/382/9558#.E7.9F.AD.E4.BF.A1.E5.8F.91.E9.80.81.E6.8F.90.E7.A4.BA.EF.BC.9Aunsupportedoperation.containdomesticandinternationalphonenumber-.E5.A6.82.E4.BD.95.E5.A4.84.E7.90.86.EF.BC.9F)
        # - 更多错误，可咨询[腾讯云助手](https://tccc.qcloud.com/web/im/index.html#/chat?webAppId=8fa15978f85cb41f7e2ea36920cb3ae1&title=Sms)
        return res
    except TencentCloudSDKException as err:
        print(err)
        return False

settings.py

我们将sms.py中的配置项放在settings.py文件中
SECRET_ID = ''  ## 当然可以采用环境变量放置给其赋值
SECRET_KEY = ''
APP_ID = ''
SIGN_NAME = ''
TEMPLATE_ID = ''

发送验证码接口

view视图层

from rest_framework.viewsets import ViewSet
from rest_framework.decorators import action
from libs.send_tx_sms import send_phone_sms, get_code
from utils.response import APIResponse
class UserView(ViewSet):
    @action(method=['get'],detail=False)  # 我们使用action装饰器自动生成路由
    def send_sms(self,request) # 不一定是GET请求我们也可以更换action装饰器变成post请求
    # 出了在手机号取值方面不一样,其余都是一样的
    	mobile = request.query_params.get('mobile')
    	code = get_code() # 获取验证码
    	'''
        我们虽然拿到了验证码,但是等一下我们需要对其进行登录校验
        所以我们暂时先将验证码放在缓存中(django项目重启,缓存中内容就会消失)
     	'''
    	cache.set('sms_code_%s' % mobile, code) # 内存中的Key需要唯一,如果不唯一,key会被覆盖！
    
    	res = send_phone_sms(mobile,code)
    	print(res.get('SendStatusSet')[0]['Code']) #我们可以通过判断 这个值
    	if res:  # 我在发送短信函数中失败处返回了False这样我们可以对res进行判断
            return APIresponse(msg='发送成功')
        else:
            return APIresponse(msg='发送验证码失败',code=1002)

这样我们就完成了短信发送的接口,测试一下！

logger.warning(f'发送短信{code},给手机号{mobile}成功！') 我在发送短信函数中增加了日志信息

短信登录接口

需求分析:
短信登录需要填写 手机号验证码
前端返回格式:{mobile:1213121,code:123}
思路:
先获取用户输入手机号和验证码
对验证码进行校验,是否与缓存中验证码一致
在通过手机号查询用户信息,判断用户是否存在，如果存在签发token
短信登录接口其实与多方式登录接口思路是一致的,校验逻辑我们可以放在序列化类中
而视图层只需要几行代码即可
view视图层

class UserView(ViewSet):
    @action(methods=['post'], detail=False) # 我们可以通过Post请求携带数据
    def mobile_login(self,request):
        # 产生序列化对象,通过序列化类来进行判断
        ser_obj = self._get_serializer(data=self.request.data)
        ser_obj.is_valid(raise_exception=True)
        token = ser_obj.context.get('token')
        icon = ser_obj.context.get('icon')
        username = ser_obj.context.get('username')
        return APIResponse(token=token, icon=icon, username=username)

至此我们可以发现,短信登录与多方式登录视图层的代码都是一样的,除了序列化类不一样而已
所以我们就可以专门编写替换序列化类的函数,判断他们的区别，让他们获取不同的序列化类对象

获取序列化类

class UserView(ViewSet):
    def _get_serializer(self,data):
        # 因为登录的请求的路径是不同的,我们可以通过请求路径进行判断
        if res == '/api/v1/userinfo/user/mobile_login/':
            return MobileLoginSerializer(data=data)# 如果手机登录返回它
        return UserMulLoginSerializer(data=data)


class UserView(ViewSet):
    def _get_serializer(self,data)：
    # 因为请求的action装饰器不同，我们也可以通过action装饰器的值进行判断
    	if self.action == 'login'
        	return UserMulLoginSerializer(data=data)
		else:
        	return MobileLoginSerializer(data=data)

现在我们就可以将多方式登录和短信登录汇总到一起

def common_login(self):
    ser_obj = self._get_serializer(data=self.request.data) # 将我们所需要验证的数据传入获取不同的序列化类即可
    ser_obj.is_valid(raise_exception=True)
    token = ser_obj.context.get('token')
    icon = ser_obj.context.get('icon')
    username = ser_obj.context.get('username')
    return APIResponse(token=token, icon=icon, username=username)

至此我们的视图类就变成了

class UserView(ViewSet):
    @action(methods=['post'], detail=False, )
    def login(self, request):
        return self.common_login()

    @action(methods=['post'], detail=False)
    def mobile_login(self, request):
        return self.common_login()

让我们在看一下序列化类中的变化

序列化类
我的思路是进行了三合一的封装,封装的有些多..
总体来说:
多方式登录序列化类/手机登录序列化类签发Token步骤/序列化全局校验步骤都是一样的
所以我们可以将token部分/全局钩子校验封装起来
除了_get_user方法是登录部分上的不同,其余基本一致，

import re
from django.contrib.auth import authenticate
from rest_framework.exceptions import ValidationError,APIException
from django.core.cache import cache
from rest_framework_jwt.serializers import jwt_payload_handler, jwt_encode_handler
from user.models import UserInfo


class Verification():
    def _get_token(self, user_obj):
        pay_load = jwt_payload_handler(user_obj)
        token = jwt_encode_handler(pay_load)
        return token
    def _validate(self,attrs):
        user_obj = self._get_user(attrs)
        token = self._get_token(user_obj)
        self.context['token'] = token  # self 是 序列化类对象 所以可以点context传值进去
        self.context['username'] = user_obj.username
        self.context['icon'] = 'http://127.0.0.1:8000/media/' + str(user_obj.icon)
        return attrs
    def _get_user(self, attrs):
        username = attrs.get('username')
        password = attrs.get('password')
        mobile = attrs.get('mobile')
        if mobile:
            if not re.match(r'^1[3-9][0-9]{9}$', mobile):
                raise ValidationError('手机号格式输入错误')
            code = attrs.get('code')
            old_code = cache.get('sms_code_%s' % mobile)
            if code == old_code:
                try:
                    user_obj = UserInfo.objects.get(mobile=mobile)
                    return user_obj
                except Exception as e:
                    raise APIException('手机号不存在')
            raise APIException('验证码错误')

        if re.match(r'^1[3-9][0-9]{9}$', username):
            user_obj = UserInfo.objects.filter(mobile=username).first()
            if user_obj and user_obj.check_password(password):
                return user_obj
            raise APIException('用户名或密码错误')
            # user_obj = authenticate(mobile=username, password=password)
        elif re.match(r'^.+@.+$', username):
            user_obj = authenticate(email=username, password=password)
        else:
            user_obj = authenticate(username=username, password=password)
        if not user_obj:
            raise APIException('用户名或密码错误')
        return user_obj

因为这个类也是单独抽出来的重复代码,所以我将它封装到了utils下的common_serializer.py中
这样我们在序列化类中直接调用它(并继承)即可

from rest_framework import serializers
from .models import UserInfo
from utils.common_serializer import Verification
from rest_framework.exceptions import APIException,ValidationError
from django.core.cache import cache


class UserMulLoginSerializer(serializers.ModelSerializer, Verification):
    username = serializers.CharField()

    class Meta:
        model = UserInfo
        fields = ['username', 'password']

    def validate(self, attrs):
        attrs = self._validate(attrs)
        return attrs


class MobileLoginSerializer(serializers.ModelSerializer, Verification):
    mobile = serializers.CharField()  # 需要重写这两个字段，因为数据库中mobile字段携带unique 并且code在内存中
    code = serializers.CharField()

    class Meta:
        model = UserInfo
        fields = ['mobile', 'code']  # code存储再内存中

    def validate(self, attrs):
        attrs = self._validate(attrs)
        return attrs

短信注册接口

短信注册接口分析:
前端返回格式:{mobile:123123123,code:1232,password:12312}
我们需要先获取前端发送的数据,
与登录同理,我们也在序列化类中进行校验
先判断手机号是否存在，并且判断验证码是否与内存中验证码一致
如果一致，我们需要调用auth的create_user方法,因为我们的密码没有进行加密，所以通过此方法进行加密处理
序列化类:

from rest_framework import serializers
from .models import UserInfo
from utils.common_serializer import Verification
from rest_framework.exceptions import APIException,ValidationError
from django.core.cache import cache

class MobileRegisterSerializer(serializers.ModelSerializer):
    mobile = serializers.CharField()
    code = serializers.CharField(max_length=4,min_length=4)
    class Meta:
        model = UserInfo
        fields = ['mobile', 'code', 'password']

    def validate(self, attrs):
        mobile = attrs.get('mobile')
        code = attrs.get('code')
        password = attrs.get('password')
        user_obj = UserInfo.objects.filter(username=mobile)
        if user_obj:
            raise APIException('手机号已存在')
        old_code = cache.get('sms_code_%s' % mobile)
        if code != old_code:
            raise APIException('验证码错误')
        UserInfo.objects.create_user(username=mobile, password=password,mobile=mobile)
        self.context['username'] = mobile
        return attrs

** 视图层**

class UserView(ViewSet):
	@action(methods=['post'], detail=False)
    def mobile_register(self, request):
        ser_obj = MobileRegisterSerializer(data=request.data)
        ser_obj.is_valid(raise_exception=True)
        return APIResponse(msg='注册成功', code=108, username=ser_obj.context.get('username'))

__EOF__

本文作者： 随风dd
本文链接： https://www.cnblogs.com/ddsuifeng/p/17184193.html
关于博主： 评论和私信会在第一时间回复。或者直接私信我。
版权声明： 本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主： 如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。