panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: refactor captcha generation logic in login.py

v3.2
wxg0103 2025-09-25 13:46:09 +08:00
parent fc05d26eaf
commit 94823b2a6a
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
apps/users/serializers/login.py
View File

@ -142,42 +142,43 @@ class CaptchaSerializer(serializers.Serializer):
def generate(username: str, type: str = 'system'): def generate(username: str, type: str = 'system'):
auth_setting = LoginSerializer.get_auth_setting() auth_setting = LoginSerializer.get_auth_setting()
max_attempts = auth_setting.get("max_attempts", 1) max_attempts = auth_setting.get("max_attempts", 1)
need_captcha = False need_captcha = False
if max_attempts == -1: if max_attempts == -1:
need_captcha = False need_captcha = False
elif max_attempts > 0: elif max_attempts > 0:
fail_count = cache.get(system_get_key(f'{type}_{username}'), version=system_version) or 0 fail_count = cache.get(system_get_key(f'system_{username}'), version=system_version) or 0
need_captcha = fail_count >= max_attempts need_captcha = fail_count >= max_attempts
if need_captcha: return CaptchaSerializer._generate_captcha_if_needed(username, type, need_captcha)
chars = get_random_chars()
image = ImageCaptcha()
data = image.generate(chars)
captcha = base64.b64encode(data.getbuffer())
cache.set(Cache_Version.CAPTCHA.get_key(captcha=f'{type}_{username}'), chars.lower(),
timeout=300, version=Cache_Version.CAPTCHA.get_version())
return {'captcha': 'data:image/png;base64,' + captcha.decode()}
return {'captcha': ''}
@staticmethod @staticmethod
def chat_generate(username: str, type: str = 'chat', access_token: str = ''): def chat_generate(username: str, type: str = 'chat', access_token: str = ''):
auth_setting = {}
application_access_token = ApplicationAccessToken.objects.filter( application_access_token = ApplicationAccessToken.objects.filter(
access_token=access_token access_token=access_token
).first() ).first()
if not application_access_token: if not application_access_token:
raise AppApiException(1005, _('Invalid access token')) raise AppApiException(1005, _('Invalid access token'))
if application_access_token:
auth_setting = application_access_token.authentication_value auth_setting = application_access_token.authentication_value
max_attempts = auth_setting.get("max_attempts", 1) max_attempts = auth_setting.get("max_attempts", 1)
need_captcha = False need_captcha = False
if max_attempts == -1: if max_attempts == -1:
need_captcha = False need_captcha = False
elif max_attempts > 0: elif max_attempts > 0:
fail_count = cache.get(system_get_key(f'{type}_{username}'), version=system_version) or 0 fail_count = cache.get(system_get_key(f'system_{username}'), version=system_version) or 0
need_captcha = fail_count >= max_attempts need_captcha = fail_count >= max_attempts
return CaptchaSerializer._generate_captcha_if_needed(username, type, need_captcha)
@staticmethod
def _generate_captcha_if_needed(username: str, type: str, need_captcha: bool):
"""
提取的公共验证码生成方法
"""
if need_captcha: if need_captcha:
chars = get_random_chars() chars = get_random_chars()
image = ImageCaptcha() image = ImageCaptcha()
@ -187,6 +188,3 @@ class CaptchaSerializer(serializers.Serializer):
timeout=300, version=Cache_Version.CAPTCHA.get_version()) timeout=300, version=Cache_Version.CAPTCHA.get_version())
return {'captcha': 'data:image/png;base64,' + captcha.decode()} return {'captcha': 'data:image/png;base64,' + captcha.decode()}
return {'captcha': ''} return {'captcha': ''}