From 1da0e12e3be4ec279f0d064fddb8f30bdab4be06 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Wed, 14 Oct 2020 13:55:56 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AB=8B=E5=8D=B3=E6=95=B4=E6=94=B9=E7=9A=84?=
 =?UTF-8?q?=E6=95=B4=E6=94=B9=E9=83=A8=E9=97=A8=E4=B8=8E=E5=8F=91=E7=8E=B0?=
 =?UTF-8?q?=E9=83=A8=E9=97=A8=E4=B8=80=E8=87=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 safesite/views.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/safesite/views.py b/safesite/views.py
index b8561845..64f7973b 100644
--- a/safesite/views.py
+++ b/safesite/views.py
@@ -990,6 +990,9 @@ def addyh(req):
         if zgbm and zgr:
             a.zgbm = Partment.objects.get(partid=zgbm)
             a.zgr = User.objects.get(userid=zgr)
+        else:
+            a.zgbm = a.fxbm
+            a.zgr = a.fxr
         a.save()
         m = TroubleAccess(troubleid=a, clr=user, yhzt=0,
                           action='新增隐患已整改', result=1)
@@ -1600,7 +1603,6 @@ def accessyh(req):
         send_wechatmsg.delay(postdict)
         return JsonResponse({"code": 1})
 
-
 def parttree2(req):
     userid = req.session['userid']
     companyid = getcompany(userid)
@@ -4956,7 +4958,11 @@ def apitool(req):
         from .tasks import updateAqzs
         updateAqzs()
         return JsonResponse({"code":1})
-
+    elif a == 'correct_ljzg':
+        objs = Trouble.objects.filter(yhzt=6, deletemark=1, zgbm__isnull=True, zgr__isnull=True)
+        # print(Trouble.objects.filter(deletemark=1).count(), objs.count())
+        objs.update(zgbm=F('fxbm'), zgr=F('fxr'))
+        return JsonResponse({"code":1})