PC新增初始化密码接口

Controller/PctestController.py

@@ -88,6 +88,8 @@ class PcTest(View):
                     return self.logquery(request_dict, userID, response)
                 elif operation == 'token/fullInfo':
                     return self.fullInfo(request_dict, userID, response)
+                elif operation == 'staff/initPass':
+                    return self.initPass(request_dict, userID, response)
                 else:
                     return response.json(404)
 
@@ -294,6 +296,7 @@ class PcTest(View):
     def staffadd(self, request_dict, userID, response):
         username = request_dict.get('username', None)
         password = request_dict.get('password', None)
+        id = request_dict.get('id', None)
         job_id = request_dict.get('job_id', None)
         param_flag = CommonService.get_param_flag(
             data=[username, password, job_id])
@@ -301,12 +304,33 @@ class PcTest(View):
             return response.json(444)
         # 判断员工与岗位是否存在，员工存在返回174
         user_list = PctestuserModel.objects.filter(username=username)
+
         job_list = PctestjobModel.objects.filter(id=job_id)
-        if not job_list.exists() or user_list.exists():
+        if not job_list.exists():
             return response.json(174)
-        else:
+
+        if not id:
+            if user_list.exists():
+                return response.json(174)
             PctestuserModel.objects.create(username=username, password=password, job_id=job_id)
             return response.json(0)
+        else:
+            PctestuserModel.objects.filter(id=id).update(username=username, password=password, job_id=job_id)
+
+
+    def initPass(self, request_dict, userID, response):
+
+        id = request_dict.get('id', None)
+        param_flag = CommonService.get_param_flag(
+            data=[id])
+        if param_flag is not True:
+            return response.json(444)
+
+        # 判断员工与岗位是否存在，员工存在返回174
+        user_list = PctestuserModel.objects.filter(id=id)
+        if user_list.exists():
+            PctestuserModel.objects.filter(id=id).update(password=555)
+
 
     def staffquery(self, request_dict, userID, response):
         user_list = PctestuserModel.objects.filter(id__gt=1).values('id', 'username', 'job__jobname', 'job__jobcode')