fix: 验证码逻辑错误

pull/7235/head
xinwen 2021-11-19 10:34:19 +08:00 committed by 老广
parent cae956f9a5
commit d94c515cfc
1 changed files with 6 additions and 5 deletions

View File

@ -37,12 +37,17 @@ class SendAndVerifySMSUtil:
self.code = '' self.code = ''
self.timeout = timeout or self.TIMEOUT self.timeout = timeout or self.TIMEOUT
self.key_suffix = key_suffix or str(phone) self.key_suffix = key_suffix or str(phone)
self.key = self.KEY_TMPL.format(key_suffix) self.key = self.KEY_TMPL.format(self.key_suffix)
def gen_and_send(self): def gen_and_send(self):
""" """
生成保存发送 生成保存发送
""" """
ttl = self.ttl()
if ttl > 0:
logger.error('Send sms too frequently, delay {}'.format(ttl))
raise CodeSendTooFrequently(ttl)
try: try:
code = self.generate() code = self.generate()
self.send(code) self.send(code)
@ -62,10 +67,6 @@ class SendAndVerifySMSUtil:
""" """
发送信息的方法如果有错误直接抛出 api 异常 发送信息的方法如果有错误直接抛出 api 异常
""" """
ttl = self.ttl()
if ttl > 0:
logger.error('Send sms too frequently, delay {}'.format(ttl))
raise CodeSendTooFrequently(ttl)
sms = SMS() sms = SMS()
sms.send_verify_code(self.phone, code) sms.send_verify_code(self.phone, code)
cache.set(self.key, self.code, self.timeout) cache.set(self.key, self.code, self.timeout)