网站首页 > 厂商资讯 > 环信 >

Django短信验证码的定时任务如何实现？

在Django项目中，短信验证码是常见的安全措施之一。为了提高用户体验，我们通常需要实现一个定时任务，在用户提交手机号后的一段时间内发送验证码。下面将详细介绍如何在Django中实现短信验证码的定时任务。

一、准备工作

安装Django项目

首先，确保你的环境中已经安装了Django。如果没有，请先安装Django。

配置短信发送服务

为了发送短信，你需要选择一个合适的短信发送服务提供商，如阿里云、腾讯云等。以下以阿里云短信服务为例进行说明。

（1）注册阿里云账号并开通短信服务。

（2）获取短信服务API的AccessKey和AccessKeySecret。

（3）在Django项目中配置阿里云短信服务的相关参数。

二、实现短信验证码

创建验证码模型

首先，我们需要创建一个验证码模型，用于存储验证码信息。

from django.db import models



class SMSCode(models.Model):

    mobile = models.CharField(max_length=11, verbose_name='手机号')

    code = models.CharField(max_length=6, verbose_name='验证码')

    add_time = models.DateTimeField(auto_now_add=True, verbose_name='添加时间')

    expire_time = models.DateTimeField(verbose_name='过期时间')

生成验证码

在用户提交手机号后，我们需要生成一个验证码，并将其存储到数据库中。

import random

from django.utils import timezone

from .models import SMSCode



def generate_code():

    return random.randint(100000, 999999)



def save_code(mobile):

    code = generate_code()

    expire_time = timezone.now() + timezone.timedelta(minutes=5)  # 验证码有效期为5分钟

    SMSCode.objects.create(mobile=mobile, code=code, expire_time=expire_time)

发送短信

在Django项目中，我们可以使用第三方库如django-aliyun-sms来实现短信发送功能。

首先，安装django-aliyun-sms：

pip install django-aliyun-sms

然后，在Django项目中配置阿里云短信服务的相关参数：

# settings.py



ALIYUN_SMS_ACCESS_KEY_ID = 'your_access_key_id'

ALIYUN_SMS_ACCESS_KEY_SECRET = 'your_access_key_secret'

ALIYUN_SMS_TEMPLATE_CODE = 'your_template_code'

ALIYUN_SMS_SIGN_NAME = 'your_sign_name'

最后，发送短信：

from aliyunsdkcore.client import AcsClient

from aliyunsdkcore.request import CommonRequest



def send_sms(mobile, code):

    client = AcsClient(ALIYUN_SMS_ACCESS_KEY_ID, ALIYUN_SMS_ACCESS_KEY_SECRET, 'cn-hangzhou')

    request = CommonRequest()

    request.set_accept_format('json')

    request.set_domain('dysmsapi.aliyuncs.com')

    request.set_method('POST')

    request.set_version('2017-05-25')

    request.set_action_name('SendSms')

    request.add_query_param('PhoneNumbers', mobile)

    request.add_query_param('SignName', ALIYUN_SMS_SIGN_NAME)

    request.add_query_param('TemplateCode', ALIYUN_SMS_TEMPLATE_CODE)

    request.add_query_param('TemplateParam', '{"code":"%s"}' % code)

    response = client.do_action_with_exception(request)

    return response

定时任务

为了实现定时发送验证码，我们可以使用Django的celery库来实现。

首先，安装celery：

pip install celery

然后，在Django项目中配置celery：

# settings.py



CELERY_BROKER_URL = 'redis://localhost:6379/0'

CELERY_RESULT_BACKEND = 'redis://localhost:6379/0'

接下来，创建一个celery任务：

from celery import shared_task

from .models import SMSCode

from .tasks import send_sms



@shared_task

def send_code_task(mobile):

    code = SMSCode.objects.filter(mobile=mobile, expire_time__gt=timezone.now()).first()

    if code:

        send_sms(mobile, code.code)

最后，在适当的位置调用定时任务：

from celery import Celery

from .tasks import send_code_task



app = Celery('tasks', broker='redis://localhost:6379/0')



@app.on_after_commit

def schedule_send_code_task():

    mobile = 'your_mobile_number'

    send_code_task.delay(mobile)

这样，当用户提交手机号后，Django会自动发送验证码，并在5分钟后过期。

三、总结

通过以上步骤，我们可以在Django项目中实现短信验证码的定时任务。在实际应用中，可以根据需求调整验证码的有效期、发送频率等参数。同时，也可以结合其他安全措施，如图形验证码、二次验证等，以提高系统的安全性。