Browse Source

PC新增设备删除,岗位删除,职能删除

chenshibin 4 years ago
parent
commit
a9bf4efb30
1 changed files with 55 additions and 24 deletions
  1. 55 24
      Controller/PctestController.py

+ 55 - 24
Controller/PctestController.py

@@ -51,30 +51,36 @@ class PcTest(View):
                 # 获取访问者的id和岗位
                 userID = tko.id
                 jobID = tko.job
-                if jobID == 1:
-                    # 管理员可访问的接口
-                    if operation == 'job/add':
-                        return self.jobadd(request_dict, userID, response)
-                    elif operation == 'job/query':
-                        return self.jobquery(request_dict, userID, response)
-                    elif operation == 'device/add':
-                        return self.deviceadd(request_dict, userID, response)
-                    elif operation == 'device/query':
-                        return self.devicequery(request_dict, userID, response)
-                    elif operation == 'function/add':
-                        return self.functionadd(request_dict, userID, response)
-                    elif operation == 'function/query':
-                        return self.functionquery(request_dict, userID, response)
-                    elif operation == 'job/device/add':
-                        return self.jobdeviceadd(request_dict, userID, response)
-                    elif operation == 'device/function/add':
-                        return self.devicefunctionadd(request_dict, userID, response)
-                    elif operation == 'staff/add':
-                        return self.staffadd(request_dict, userID, response)
-                    elif operation == 'staff/query':
-                        return self.staffquery(request_dict, userID, response)
-                    elif operation == 'staff/delete':
-                        return self.staffdelete(request_dict, userID, response)
+                # if jobID == 1:
+                #     # 管理员可访问的接口
+                if operation == 'job/add':
+                    return self.jobadd(request_dict, userID, response)
+                elif operation == 'job/query':
+                    return self.jobquery(request_dict, userID, response)
+                elif operation == 'job/delete':
+                    return self.jobdelete(request_dict, userID, response)
+                elif operation == 'device/add':
+                    return self.deviceadd(request_dict, userID, response)
+                elif operation == 'device/query':
+                    return self.devicequery(request_dict, userID, response)
+                elif operation == 'device/delete':
+                    return self.devicedelete(request_dict, userID, response)
+                elif operation == 'function/add':
+                    return self.functionadd(request_dict, userID, response)
+                elif operation == 'function/query':
+                    return self.functionquery(request_dict, userID, response)
+                elif operation == 'function/delete':
+                    return self.functiondelete(request_dict, userID, response)
+                elif operation == 'job/device/add':
+                    return self.jobdeviceadd(request_dict, userID, response)
+                elif operation == 'device/function/add':
+                    return self.devicefunctionadd(request_dict, userID, response)
+                elif operation == 'staff/add':
+                    return self.staffadd(request_dict, userID, response)
+                elif operation == 'staff/query':
+                    return self.staffquery(request_dict, userID, response)
+                elif operation == 'staff/delete':
+                    return self.staffdelete(request_dict, userID, response)
 
 
                 # 公共访问的接口
@@ -155,6 +161,14 @@ class PcTest(View):
         count = device_qs.count()
         return response.json(0, {'datas': data, 'count': count})
 
+    def jobdelete(self, request_dict, userID, response):
+        id = request_dict.get('id', None)
+        job_qs = PctestjobModel.objects.filter(id=id)
+        if job_qs.exists:
+            job_qs.delete()
+            PctestjobdeviceModel.objects.filter(job__in=job_qs).delete()
+        return response.json(0)
+
     def deviceadd(self, request_dict, userID, response):
         deviceid = request_dict.get('deviceid', None)
         devicename = request_dict.get('devicename', None)
@@ -206,6 +220,15 @@ class PcTest(View):
         count = device_qs.count()
         return response.json(0, {'datas': data, 'count': count})
 
+    def devicedelete(self, request_dict, userID, response):
+        id = request_dict.get('id', None)
+        device_qs = PctestdeviceModel.objects.filter(id=id)
+        if device_qs.exists:
+            device_qs.delete()
+            PctestModel.objects.filter(device__in=device_qs).delete()
+
+        return response.json(0)
+
     def functionadd(self, request_dict, userID, response):
         functionname = request_dict.get('functionname', None)
         functioncode = request_dict.get('functioncode', None)
@@ -225,6 +248,14 @@ class PcTest(View):
 
         return response.json(0)
 
+    def functiondelete(self, request_dict, userID, response):
+        id = request_dict.get('id', None)
+        function_qs = PctestfunctionModel.objects.filter(id=id)
+
+        if function_qs.exists:
+            function_qs.delete()
+        return response.json(0)
+
     def functionquery(self, request_dict, userID, response):
         function_list = PctestfunctionModel.objects.all()
         count = function_list.count()