Kaynağa Gözat

add langset by admin

chenjunkai 6 yıl önce
ebeveyn
işleme
6f66e0f31a
2 değiştirilmiş dosya ile 9 ekleme ve 0 silme
  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':