From ca5f853a619306779f466367a1dd9d876ef879f4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=BD=95=E5=85=A8?= Date: Wed, 27 Mar 2019 17:06:45 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B3=A8=E9=94=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 注销 --- system/urls.py | 4 ++-- system/views.py | 10 +++++++--- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/system/urls.py b/system/urls.py index 8b57a8f..6f1e5e4 100644 --- a/system/urls.py +++ b/system/urls.py @@ -1,5 +1,5 @@ from django.urls import path -from system.views import login_view, UserPasswordUpdateView, logout_view, UserInfo +from system.views import login_view, UserPasswordUpdateView, logout_view, UserInfo, UserLogout app_name = "system" @@ -10,5 +10,5 @@ urlpatterns = [ path('api/user_info', UserInfo.as_view()), - path('api/logout', UserInfo.as_view()) + path('api/logout', UserLogout.as_view()) ] diff --git a/system/views.py b/system/views.py index c76a917..11e9252 100644 --- a/system/views.py +++ b/system/views.py @@ -16,6 +16,8 @@ from rest_framework.authentication import TokenAuthentication from rest_framework.authtoken.models import Token from rest_framework.views import APIView from rest_framework.response import Response +from rest_framework import permissions +from rest_framework import generics logger = logging.getLogger('system') @@ -24,6 +26,8 @@ class UserInfo(APIView): """ 获取用户信息 """ + permission_classes = (permissions.IsAuthenticated,) + def post(self, request): token = (json.loads(request.body))['token'] obj = Token.objects.get(key=token).user @@ -37,10 +41,9 @@ class UserInfo(APIView): return HttpResponse(json.dumps(result)) -class UserLogin(APIView): - """ +class UserLogout(APIView): + permission_classes = (permissions.IsAuthenticated,) - """ def post(self, request): token = (json.loads(request.body))['token'] obj = Token.objects.get(key=token) @@ -50,6 +53,7 @@ class UserLogin(APIView): } return HttpResponse(json.dumps(result)) + class CustomBackend(ModelBackend): """ 用户名字/邮箱名字 登录