瀏覽代碼

add langset by admin

chenjunkai 6 年之前
父節點
當前提交
6f66e0f31a
共有 2 個文件被更改,包括 9 次插入0 次删除
  1. 3 0
      controller/LangArea.py
  2. 6 0
      controller/LangWord.py

+ 3 - 0
controller/LangArea.py

@@ -45,6 +45,9 @@ class LangAreaView(TemplateView):
         tko = TokenObject(token)
         if tko.code == 0:
             userID = tko.userID
+            user_qs = UserModel.objects.filter(id=userID, username='admin')
+            if not user_qs.exists():
+                return response.json(403)
             if operation == 'add':
                 return self.do_add(request_dict, response)
             elif operation == 'update':

+ 6 - 0
controller/LangWord.py

@@ -46,10 +46,16 @@ class LangWordView(TemplateView):
         if tko.code == 0:
             userID = tko.userID
             if operation == 'add':
+                user_qs = UserModel.objects.filter(id=userID, username='admin')
+                if not user_qs.exists():
+                    return response.json(403)
                 return self.do_add(request_dict, response)
             elif operation == 'update':
                 return self.do_update(request_dict, response)
             elif operation == 'delete':
+                user_qs = UserModel.objects.filter(id=userID, username='admin')
+                if not user_qs.exists():
+                    return response.json(403)
                 return self.do_delete(request_dict, userID, response)
             # search index table
             elif operation == 'query':