Flask项目中实现短信验证码和邮箱验证码功能

以下是“Flask项目中实现短信验证码和邮箱验证码功能”的完整攻略：

短信验证码功能的实现

调用短信API接口

首先，需要找到一个靠谱的短信API接口。可以通过第三方短信服务商提供的短信API接口来发送短信验证码。

以阿里云为例，可以借助阿里云的短信服务平台实现。

import json
from aliyunsdkcore.client import AcsClient
from aliyunsdkcore.request import CommonRequest

# 阿里云短信API接口的配置信息
ACCESS_KEY_ID = 'your_access_key_id'
ACCESS_KEY_SECRET = 'your_access_key_secret'
REGION_ID = 'cn-hangzhou'
SIGN_NAME = 'your_sms_sign_name'
TEMPLATE_CODE = 'your_template_code'

# 定义发送短信验证码的函数
def send_sms_code(mobile, code):
    # 创建ACS客户端实例
    client = AcsClient(ACCESS_KEY_ID, ACCESS_KEY_SECRET, REGION_ID)
    # 构造用于调用发送短信API的请求
    request = CommonRequest()
    request.set_method('POST')
    request.set_domain('dysmsapi.aliyuncs.com')
    request.set_version('2017-05-25')
    request.set_action_name('SendSms')
    request.add_query_param('RegionId', REGION_ID)
    request.add_query_param('PhoneNumbers', mobile)
    request.add_query_param('SignName', SIGN_NAME)
    request.add_query_param('TemplateCode', TEMPLATE_CODE)
    request.add_query_param('TemplateParam', json.dumps({'code': code}))
    # 发送短信
    response = client.do_action(request)
    return response

生成随机验证码

可以借助Python的随机数模块random来实现验证码的生成。生成随机数之后，需要将验证码同时保存到后端服务器和发送给用户。

import random

# 定义生成随机验证码的函数
def generate_random_code(length=6):
    # length参数表示验证码的位数，默认为6位
    return ''.join([str(random.choice(range(10))) for i in range(length)])

发送验证码到用户手机

将随机生成的验证码传入send_sms_code()函数，然后再通过短信API接口调用发送验证码到用户手机。

from flask import Flask, request, jsonify

app = Flask(__name__)

# 定义短信验证码API接口
@app.route('/api/sms_code', methods=['POST'])
def sms_code():
    mobile = request.form.get('mobile')
    code = generate_random_code()
    # TODO: 将code保存到后端服务器
    result = send_sms_code(mobile, code)
    return jsonify({'success': True, 'message': '验证码已发送'})

if __name__ == '__main__':
    app.run()

邮箱验证码功能的实现

调用邮件API接口

邮件API接口可以使用SMTP协议发送邮件，也可以使用第三方邮件服务商提供的API发送邮件。以SMTP协议发送邮件为例，可以通过Python内置的smtplib模块实现。

import smtplib
from email.mime.text import MIMEText
from email.header import Header

# 邮件服务的配置信息
SMTP_SERVER = 'smtp.exmail.qq.com'
SMTP_PORT = 465
SMTP_USER = 'your_smtp_user'
SMTP_PASSWORD = 'your_smtp_password'
FROM_ADDRESS = 'your_from_address'

# 定义发送验证邮件的函数
def send_email_code(email, code):
    try:
        # 构建邮件对象
        message = MIMEText('您的验证码是：{}'.format(code), 'plain', 'utf-8')
        message['From'] = Header('xxx', 'utf-8')
        message['To'] = Header(email, 'utf-8')
        message['Subject'] = Header('验证码', 'utf-8')
        # 建立SMTP连接并发送邮件
        smtp_server = smtplib.SMTP_SSL(SMTP_SERVER, SMTP_PORT)
        smtp_server.login(SMTP_USER, SMTP_PASSWORD)
        smtp_server.sendmail(FROM_ADDRESS, [email], message.as_string())
        smtp_server.quit()
        return True
    except Exception as e:
        print(str(e))
        return False

生成随机验证码

同短信验证码功能的实现，可以借助Python的随机数模块random来生成邮箱验证码。

import random

# 定义生成随机验证码的函数
def generate_random_code(length=6):
    # length参数表示验证码的位数，默认为6位
    return ''.join([str(random.choice(range(10))) for i in range(length)])

发送验证码到用户邮箱

与发送短信验证码类似，将随机生成的验证码传入send_email_code()函数，然后再通过邮件API接口调用发送验证码到用户邮箱。

from flask import Flask, request, jsonify

app = Flask(__name__)

# 定义邮箱验证码API接口
@app.route('/api/email_code', methods=['POST'])
def email_code():
    email = request.form.get('email')
    code = generate_random_code()
    # TODO: 将code保存到后端服务器
    result = send_email_code(email, code)
    if result:
        return jsonify({'success': True, 'message': '验证码已发送'})
    else:
        return jsonify({'success': False, 'message': '验证码发送失败'})

if __name__ == '__main__':
    app.run()