diff --git a/account/views/oj.py b/account/views/oj.py index fdf2c31..b1cef63 100644 --- a/account/views/oj.py +++ b/account/views/oj.py @@ -66,8 +66,8 @@ class UserRegisterAPI(APIView): """ data = request.data captcha = Captcha(request) - # if not captcha.check(data["captcha"]): - # return self.error("Invalid captcha") + if not captcha.check(data["captcha"]): + return self.error("Invalid captcha") try: User.objects.get(username=data["username"]) return self.error("Username already exists") diff --git a/account/views/user.py b/account/views/user.py index 1f9ee24..d0b8393 100644 --- a/account/views/user.py +++ b/account/views/user.py @@ -14,9 +14,8 @@ from utils.shortcuts import rand_str from ..decorators import login_required from ..models import User, UserProfile -from ..serializers import (EditUserSerializer, SSOSerializer, - TwoFactorAuthCodeSerializer, UserSerializer, - UserProfileSerializer, UserInfoSerializer, +from ..serializers import (SSOSerializer, TwoFactorAuthCodeSerializer, + UserSerializer, UserProfileSerializer, EditUserProfileSerializer, AvatarUploadForm) @@ -33,7 +32,7 @@ class UserInfoAPI(APIView): return self.error("User does not exist") profile = UserProfile.objects.get(user=user) dit = UserProfileSerializer(profile).data - dit['user'] = UserSerializer(user).data + dit["user"] = UserSerializer(user).data return self.success(dit) @@ -49,7 +48,7 @@ class UserProfileAPI(APIView): return self.error("User does not exist") profile = UserProfile.objects.get(user=user) dit = UserProfileSerializer(profile).data - dit['user'] = UserSerializer(user).data + dit["user"] = UserSerializer(user).data return self.success(dit) @validate_serializer(EditUserProfileSerializer)