zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: login captcha

Browse Source
This commit is contained in:
wxg0103 2025-05-22 15:42:12 +08:00
parent abadff25e5
commit 9dc06909bd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -44,7 +44,7 @@ class LoginSerializer(serializers.Serializer):
username = instance.get('username') username = instance.get('username')
password = instance.get('password') password = instance.get('password')
captcha = instance.get('captcha') captcha = instance.get('captcha')
captcha_cache = cache.get(Cache_Version.CAPTCHA.get_key(captcha=captcha), captcha_cache = cache.get(Cache_Version.CAPTCHA.get_key(captcha=captcha.lower()),
version=Cache_Version.CAPTCHA.get_version()) version=Cache_Version.CAPTCHA.get_version())
if captcha_cache is None: if captcha_cache is None:
raise AppApiException(1005, _("Captcha code error or expiration")) raise AppApiException(1005, _("Captcha code error or expiration"))
@ -76,6 +76,6 @@ class CaptchaSerializer(serializers.Serializer):
image = ImageCaptcha() image = ImageCaptcha()
data = image.generate(chars) data = image.generate(chars)
captcha = base64.b64encode(data.getbuffer()) captcha = base64.b64encode(data.getbuffer())
cache.set(Cache_Version.CAPTCHA.get_key(captcha=chars), chars, cache.set(Cache_Version.CAPTCHA.get_key(captcha=chars.lower()), chars,
timeout=60, version=Cache_Version.CAPTCHA.get_version()) timeout=60, version=Cache_Version.CAPTCHA.get_version())
return {'captcha': 'data:image/png;base64,' + captcha.decode()} return {'captcha': 'data:image/png;base64,' + captcha.decode()}