Procházet zdrojové kódy

add langset by admin

chenjunkai před 6 roky
rodič
revize
6f66e0f31a
2 změnil soubory, kde provedl 9 přidání a 0 odebrání
  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':