Django短信验证码的存储方式有哪些？

在Django框架中，短信验证码的存储方式主要有以下几种：

使用数据库存储

使用数据库存储短信验证码是最常见的方式，因为数据库具有强大的查询、存储和管理功能。以下是使用数据库存储短信验证码的步骤：

（1）创建一个模型（Model）来存储验证码信息，包括手机号码、验证码、过期时间等字段。

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='验证码')

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

（2）在发送验证码时，将验证码信息保存到数据库中。

from django.utils import timezone

from .models import SMSCode



def send_sms_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)

（3）在验证验证码时，从数据库中查询验证码信息，并进行校验。

from django.utils import timezone

from .models import SMSCode



def check_sms_code(mobile, code):

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

    if sms_code:

        return True

    return False

使用缓存存储

使用缓存存储短信验证码可以减少数据库的压力，提高系统性能。以下是使用缓存存储短信验证码的步骤：

（1）在Django项目中配置缓存。

# settings.py

CACHES = {

    'default': {

        'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',

        'LOCATION': 'unique-snowflake',

    }

}

（2）使用缓存存储验证码信息。

from django.core.cache import cache

from django.utils import timezone

import random



def send_sms_code(mobile):

    code = str(random.randint(100000, 999999))

    expire_time = timezone.now() + timezone.timedelta(minutes=5)

    cache.set(f'sms_code_{mobile}', code, timeout=expire_time.total_seconds())

（3）在验证验证码时，从缓存中获取验证码信息，并进行校验。

from django.utils import timezone



def check_sms_code(mobile, code):

    cached_code = cache.get(f'sms_code_{mobile}')

    if cached_code and cached_code == code:

        return True

    return False

使用文件存储

使用文件存储短信验证码适用于小型项目或测试环境，但不太适合生产环境，因为文件存储存在性能和安全性问题。以下是使用文件存储短信验证码的步骤：

（1）创建一个文件夹用于存储验证码信息。

（2）在发送验证码时，将验证码信息保存到文件中。

import os

import random



def send_sms_code(mobile):

    code = str(random.randint(100000, 999999))

    file_path = os.path.join('sms_codes', f'{mobile}.txt')

    with open(file_path, 'w') as f:

        f.write(code)

（3）在验证验证码时，从文件中读取验证码信息，并进行校验。

def check_sms_code(mobile, code):

    file_path = os.path.join('sms_codes', f'{mobile}.txt')

    if os.path.exists(file_path):

        with open(file_path, 'r') as f:

            cached_code = f.read()

            if cached_code == code:

                return True

    return False

综上所述，Django短信验证码的存储方式主要有数据库存储、缓存存储和文件存储。在实际项目中，应根据需求选择合适的存储方式，以达到性能、安全性和可维护性的平衡。

猜你喜欢：企业智能办公场景解决方案