18-短信验证码接口

短信验证码接口

user/throttles.py

from rest_framework.throttling import SimpleRateThrottle

class SmsThrottling(SimpleRateThrottle):
	scope = 'sms'
	def get_cache_key(self, request, view):
		# 获取手机号码并校验是否符合格式
		telephone = request.query_params.get('telephone')
		# cache_format = 'throttle_%(scope)s_%(ident)s'
		return self.cache_format%{'scope':self.scope,'ident':telephone}

settings/dev.py

# drf配置
REST_FRAMEWORK = {
	# 自定义短信发送频率配置
	'DEFAULT_THROTTLE_RATES': {
		'sms': '1/m'  # key要跟类中的scop对应
	}
}

user/views.py

class SendSmsView(ViewSet):
	throttle_classes = (SmsThrottling,)

	# 发送短信验证码接口视图
	@action(methods=('GET',), detail=False)
	def send(self, request, *args, **kwargs):

		# 获取手机号码并校验是否符合格式
		telephone = request.query_params.get('telephone')
		if not re.match('^1[3-9][0-9]{9}$', telephone):  # 当电话号码格式不正确
			return APIResponse(code=0, msg='Error', result='号码格式不正确！')

		# 获取随机验证码，因为没有腾讯短信，所以这里就手动生成验证码
		code = get_code()
		print("当前验证码是：", code)
		# 发送手机短信
		result = send_message(telephone, code)
		if result:  # 这是发送成功
			# 发送验证码成功后，需要将验证码保存起来，方便后期登录使用
			# cache(缓存的键，缓存的值，缓存保存时间)，这里验证码先不存缓存
			cache.set(settings.PYTHON_CACHE_KEY.format(telephone), code, 180)
			return APIResponse(code=1, msg='Successful', result='验证发送成功')
		else:  # 这是发送失败
			return APIResponse(code=0, msg='Error', result='您操作太频繁了，验证发送失败')

user.urls.py

from django.urls import path, include
from rest_framework.routers import SimpleRouter
from . import views

router = SimpleRouter()
router.register('', views.SendSmsView, 'send')  # 此处第一个参数不要加上，因为视图类中使用了actions，会把该视图名添加进来

urlpatterns = [
	path('', include(router.urls)),
]