注销

注销

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())
 ]

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):
     """
     用户名字/邮箱名字 登录