github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-05 17:30:30 +00:00
Code Issues Projects Releases Wiki Activity

perf: Custom SMS (files) support obtaining more user information. (#14486)

Browse Source 
* perf: Custom SMS (files) support obtaining more user information.

* perf: Remove the useless modules

* perf: modify

---------

Co-authored-by: jiangweidong <1053570670@qq.com>
This commit is contained in:
fit2bot
2024-11-20 10:29:14 +08:00
committed by GitHub
parent f92c557235
commit e2904ab042
7 changed files with 44 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
apps/common/utils/verify_code.py
View File

@@ -20,16 +20,20 @@ logger = get_logger(__file__)
be executed to send SMS messages"""
)
)
def send_sms_async(target, code):
SMS().send_verify_code(target, code)
def send_sms_async(target, code, user_info):
SMS().send_verify_code(target, code, user_info=user_info)
class SendAndVerifyCodeUtil(object):
KEY_TMPL = 'auth-verify-code-{}'
def __init__(self, target, code=None, key=None, backend='email', timeout=None, **kwargs):
def __init__(
self, target, code=None, key=None, backend='email',
user_info=None, timeout=None, **kwargs
):
self.code = code
self.target = target
self.user_info = user_info
self.backend = backend
self.key = key or self.KEY_TMPL.format(target)
self.timeout = settings.VERIFY_CODE_TTL if timeout is None else timeout
@@ -78,7 +82,7 @@ class SendAndVerifyCodeUtil(object):
return code
def __send_with_sms(self):
send_sms_async.apply_async(args=(self.target, self.code), priority=100)
send_sms_async.apply_async(args=(self.target, self.code, self.user_info), priority=100)
def __send_with_email(self):
subject = self.other_args.get('subject', '')