gitee
/
django-vue-admin
mirror of https://gitee.com/liqianglog/django-vue-admin
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

修复验证码开关

pull/47/head
zghmvp 2022-04-11 16:21:15 +08:00
parent 75174311c8
commit 4abe392313
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/dvadmin/system/views/login.py
View File

@ -11,6 +11,7 @@ import hashlib
from datetime import datetime, timedelta from datetime import datetime, timedelta
from captcha.views import CaptchaStore, captcha_image from captcha.views import CaptchaStore, captcha_image
from django.conf import settings
from django.contrib import auth from django.contrib import auth
from django.contrib.auth import login from django.contrib.auth import login
from django.shortcuts import redirect from django.shortcuts import redirect
@ -56,7 +57,7 @@ class LoginSerializer(TokenObtainPairSerializer):
登录的序列化器: 登录的序列化器:
重写djangorestframework-simplejwt的序列化器 重写djangorestframework-simplejwt的序列化器
""" """
captcha = serializers.CharField(max_length=6) captcha = serializers.CharField(max_length=6, required=False, allow_null=True)
class Meta: class Meta:
model = Users model = Users
@ -68,6 +69,7 @@ class LoginSerializer(TokenObtainPairSerializer):
} }
def validate_captcha(self, captcha): def validate_captcha(self, captcha):
if settings.CAPTCHA_STATE is True:
self.image_code = CaptchaStore.objects.filter( self.image_code = CaptchaStore.objects.filter(
id=self.initial_data['captchaKey']).first() id=self.initial_data['captchaKey']).first()
five_minute_ago = datetime.now() - timedelta(hours=0, minutes=5, seconds=0) five_minute_ago = datetime.now() - timedelta(hours=0, minutes=5, seconds=0)