diff --git a/safesite/tasks.py b/safesite/tasks.py index daf5a61b..91e46fcc 100644 --- a/safesite/tasks.py +++ b/safesite/tasks.py @@ -10,184 +10,345 @@ import calendar import pandas as pd from sklearn import linear_model from django.db.models import Sum -from django.conf import settings +from django.conf import settings from .safespider import getTzzs, getAqzs import time -dirname = settings.BASE_DIR +'/safesite/' -def getcs(companyid):#获取公司相关参数设置 +dirname = settings.BASE_DIR + "/safesite/" + + +def getcs(companyid): # 获取公司相关参数设置 + qs = Yjsetup.objects.filter(usecomp__partid=companyid) + if qs.exists(): + pass + else: + Yjsetup.objects.create( + monthyhavg=100, + safev=50, + warnv=75, + alertv=100, + maxv=200, + yhpgqz=4, + yhdjqz=4, + yhsblqz=1, + yhzglqz=4, + pxdjqz=3, + pxxgqz=4, + yldjqz=3, + ylxgqz=3, + usecomp=Partment.objects.get(partid=companyid), + ) return Yjsetup.objects.filter(usecomp__partid=companyid).values()[0] + def gettime(x=datetime.datetime.now()): days_num = calendar.monthrange(x.year, x.month)[1] - first_day = datetime.date(x.year,x.month,1) - first_day_of_next_month = first_day + datetime.timedelta(days = days_num) - return first_day,first_day_of_next_month + first_day = datetime.date(x.year, x.month, 1) + first_day_of_next_month = first_day + datetime.timedelta(days=days_num) + return first_day, first_day_of_next_month + @shared_task def get_wtoken(): - r=requests.get('https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=wx4d75244c1b980b5b&secret=edf7d23a457fb4df5ad47eeb9d4b7a39').text - token=json.loads(r)['access_token'] - with open('D:/safesite/safesite/token.txt','w') as f: - f.write(token) + r = requests.get( + "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=wx4d75244c1b980b5b&secret=edf7d23a457fb4df5ad47eeb9d4b7a39" + ).text + token = json.loads(r)["access_token"] + with open("D:/safesite/safesite/token.txt", "w") as f: + f.write(token) + @shared_task -def send_wechatmsgs(postdict,tolist): - with open(dirname + 'token.txt','r',encoding= 'utf-8') as f: - token=f.read() +def send_wechatmsgs(postdict, tolist): + with open(dirname + "token.txt", "r", encoding="utf-8") as f: + token = f.read() try: for i in tolist: - postdict['touser']=i - requests.post('https://api.weixin.qq.com/cgi-bin/message/template/send?access_token='+token,data=json.dumps(postdict)) + postdict["touser"] = i + requests.post( + "https://api.weixin.qq.com/cgi-bin/message/template/send?access_token=" + + token, + data=json.dumps(postdict), + ) except: pass @shared_task def send_wechatmsg(postdict): - with open(dirname + 'token.txt','r',encoding= 'utf-8') as f: - token=f.read() + with open(dirname + "token.txt", "r", encoding="utf-8") as f: + token = f.read() try: - v = requests.post('https://api.weixin.qq.com/cgi-bin/message/template/send?access_token='+token,data=json.dumps(postdict)) + v = requests.post( + "https://api.weixin.qq.com/cgi-bin/message/template/send?access_token=" + + token, + data=json.dumps(postdict), + ) print(v.text()) except: pass + @shared_task def yjjs(companyid, first_day=gettime()[0], first_day_of_next_month=gettime()[1]): - #隐患算分 - a = Trouble.objects.filter(fxsj__range=(first_day, first_day_of_next_month),usecomp__partid=companyid,deletemark=1) - #隐患按时上报率 - if a.count()==0: + # 隐患算分 + a = Trouble.objects.filter( + fxsj__range=(first_day, first_day_of_next_month), + usecomp__partid=companyid, + deletemark=1, + ) + # 隐患按时上报率 + if a.count() == 0: sbjs = 1 else: sbjs = 0 - #隐患整改率 + # 隐患整改率 if a.count() != 0: - zgjs = a.filter(yhzt__in=[4,5,6]).count()/a.count() + zgjs = a.filter(yhzt__in=[4, 5, 6]).count() / a.count() else: zgjs = 4 if zgjs == 1: zgjs = 0 - elif zgjs>=0.8 and zgjs <1: + elif zgjs >= 0.8 and zgjs < 1: zgjs = 0.1 - elif zgjs>=0.5 and zgjs<0.8: + elif zgjs >= 0.5 and zgjs < 0.8: zgjs = 0.3 - elif zgjs>=0.3 and zgjs<0.5: + elif zgjs >= 0.3 and zgjs < 0.5: zgjs = 0.5 else: - zgjs = 1*4 + zgjs = 1 * 4 cs = getcs(companyid) - yhsw = a.filter(yhpg__dickeyid=53).count() #死亡隐患数 + yhsw = a.filter(yhpg__dickeyid=53).count() # 死亡隐患数 yhzs = a.filter(yhpg__dickeyid=52).count() yhqs = a.filter(yhpg__dickeyid=51).count() yhqws = a.filter(yhpg__dickeyid=50).count() yhwsh = a.filter(yhpg__dickeyid=49).count() - score = (((yhsw*1 + yhzs*0.6 + yhqs*0.3+ yhqws*0.2 + yhwsh*0.1)/cs['monthyhavg'] )* cs['yhpgqz']+ #隐患评估 - (a.filter(yhdj__dickeyid=55).count()*0.3+a.filter(yhdj__dickeyid=56).count()*1)* cs['yhdjqz']+ #隐患等级 - sbjs*cs['yhsblqz']+ #隐患上报率 - zgjs*cs['yhzglqz']) #隐患整改率 - #print(score) + score = ( + ( + (yhsw * 1 + yhzs * 0.6 + yhqs * 0.3 + yhqws * 0.2 + yhwsh * 0.1) + / cs["monthyhavg"] + ) + * cs["yhpgqz"] # 隐患评估 + + ( + a.filter(yhdj__dickeyid=55).count() * 0.3 + + a.filter(yhdj__dickeyid=56).count() * 1 + ) + * cs["yhdjqz"] # 隐患等级 + + sbjs * cs["yhsblqz"] # 隐患上报率 + + zgjs * cs["yhzglqz"] + ) # 隐患整改率 + # print(score) year = first_day.year month = first_day.month - obj,created = Yjyc.objects.get_or_create(usecomp__partid=companyid,year=year,month=month,defaults={'usecomp':Partment.objects.get(partid=companyid) ,'year':year,'month':month}) + obj, created = Yjyc.objects.get_or_create( + usecomp__partid=companyid, + year=year, + month=month, + defaults={ + "usecomp": Partment.objects.get(partid=companyid), + "year": year, + "month": month, + }, + ) obj.troublevalue = score - obj.yjz = round(obj.troublevalue + obj.trainvalue + obj.drillvalue + obj.missvalue + obj.observevalue,2) + obj.yjz = round( + obj.troublevalue + + obj.trainvalue + + obj.drillvalue + + obj.missvalue + + obj.observevalue, + 2, + ) obj.save() + @shared_task def yjjs_px(companyid, first_day=gettime()[0], first_day_of_next_month=gettime()[1]): - #培训算分 - a = Train.objects.filter(starttime__range=(first_day, first_day_of_next_month),usecomp__partid=companyid,state=1,deletemark=1) #有效培训 - #计算培训效果 + # 培训算分 + a = Train.objects.filter( + starttime__range=(first_day, first_day_of_next_month), + usecomp__partid=companyid, + state=1, + deletemark=1, + ) # 有效培训 + # 计算培训效果 participantnum = Trainuser.objects.filter(train__in=a).count() - knownum = a.aggregate(knownum = Sum('knownum'))['knownum'] + knownum = a.aggregate(knownum=Sum("knownum"))["knownum"] if participantnum != 0: - x = knownum/participantnum + x = knownum / participantnum else: x = 0 if x == 1: pxxg = 1 - elif 1>x>= 0.9: + elif 1 > x >= 0.9: pxxg = 0.8 - elif 0.9>x>=0.6: + elif 0.9 > x >= 0.6: pxxg = 0.5 else: pxxg = 0.1 - #培训等级 + # 培训等级 cs = getcs(companyid) - score = ((a.filter(trainlevel__dicid=5).count()*0.5 + a.filter(trainlevel__dicid=6).count()*0.8 + a.filter(trainlevel__dicid=7).count()*0.5+ a.filter(trainlevel__dicid=42).count()*0.5)*cs['pxdjqz'] + - pxxg*cs['pxxgqz'] ) #培训效果 + score = ( + a.filter(trainlevel__dicid=5).count() * 0.5 + + a.filter(trainlevel__dicid=6).count() * 0.8 + + a.filter(trainlevel__dicid=7).count() * 0.5 + + a.filter(trainlevel__dicid=42).count() * 0.5 + ) * cs["pxdjqz"] + pxxg * cs["pxxgqz"] # 培训效果 year = first_day.year month = first_day.month - obj,created = Yjyc.objects.get_or_create(usecomp__partid=companyid,year=year,month=month,defaults={'usecomp':Partment.objects.get(partid=companyid) ,'year':year,'month':month}) + obj, created = Yjyc.objects.get_or_create( + usecomp__partid=companyid, + year=year, + month=month, + defaults={ + "usecomp": Partment.objects.get(partid=companyid), + "year": year, + "month": month, + }, + ) obj.trainvalue = -score - obj.yjz = round(obj.troublevalue + obj.trainvalue + obj.drillvalue + obj.missvalue + obj.observevalue,2) + obj.yjz = round( + obj.troublevalue + + obj.trainvalue + + obj.drillvalue + + obj.missvalue + + obj.observevalue, + 2, + ) obj.save() + @shared_task def yjjs_gc(companyid, first_day=gettime()[0], first_day_of_next_month=gettime()[1]): - #观察算分 - a = Observe.objects.filter(looktime__range=(first_day, first_day_of_next_month),usecomp__partid=companyid,deletemark=1) + # 观察算分 + a = Observe.objects.filter( + looktime__range=(first_day, first_day_of_next_month), + usecomp__partid=companyid, + deletemark=1, + ) num = a.count() - if num>40: - score = (num-40)*0.04 + 40*0.4 + if num > 40: + score = (num - 40) * 0.04 + 40 * 0.4 else: - score = num*0.4 + score = num * 0.4 year = first_day.year month = first_day.month - obj,created = Yjyc.objects.get_or_create(usecomp__partid=companyid,year=year,month=month,defaults={'usecomp':Partment.objects.get(partid=companyid) ,'year':year,'month':month}) + obj, created = Yjyc.objects.get_or_create( + usecomp__partid=companyid, + year=year, + month=month, + defaults={ + "usecomp": Partment.objects.get(partid=companyid), + "year": year, + "month": month, + }, + ) obj.observevalue = -score - obj.yjz = round(obj.troublevalue + obj.trainvalue + obj.drillvalue + obj.missvalue + obj.observevalue,2) + obj.yjz = round( + obj.troublevalue + + obj.trainvalue + + obj.drillvalue + + obj.missvalue + + obj.observevalue, + 2, + ) obj.save() + @shared_task def yjjs_ws(companyid, first_day=gettime()[0], first_day_of_next_month=gettime()[1]): - #未遂算分 - a = Miss.objects.filter(misstime__range=(first_day, first_day_of_next_month),usecomp__partid=companyid,deletemark=1) # - score = (a.count())*0.6 + # 未遂算分 + a = Miss.objects.filter( + misstime__range=(first_day, first_day_of_next_month), + usecomp__partid=companyid, + deletemark=1, + ) # + score = (a.count()) * 0.6 year = first_day.year month = first_day.month - obj,created = Yjyc.objects.get_or_create(usecomp__partid=companyid,year=year,month=month,defaults={'usecomp':Partment.objects.get(partid=companyid) ,'year':year,'month':month}) + obj, created = Yjyc.objects.get_or_create( + usecomp__partid=companyid, + year=year, + month=month, + defaults={ + "usecomp": Partment.objects.get(partid=companyid), + "year": year, + "month": month, + }, + ) obj.missvalue = score - obj.yjz = round(obj.troublevalue + obj.trainvalue + obj.drillvalue + obj.missvalue + obj.observevalue,2) + obj.yjz = round( + obj.troublevalue + + obj.trainvalue + + obj.drillvalue + + obj.missvalue + + obj.observevalue, + 2, + ) obj.save() + @shared_task def yjjs_yl(companyid, first_day=gettime()[0], first_day_of_next_month=gettime()[1]): - a = Drill.objects.filter(starttime__range=(first_day, first_day_of_next_month),usecomp__partid=companyid,state=1,deletemark=1) #有效演练 - #计算演练效果 + a = Drill.objects.filter( + starttime__range=(first_day, first_day_of_next_month), + usecomp__partid=companyid, + state=1, + deletemark=1, + ) # 有效演练 + # 计算演练效果 participantnum = Drill.objects.filter(drillid__in=a).count() - knownum = a.aggregate(knownum = Sum('knownum'))['knownum'] + knownum = a.aggregate(knownum=Sum("knownum"))["knownum"] if participantnum != 0: - x = knownum/participantnum + x = knownum / participantnum else: x = 0 if x == 1: ylxg = 1 - elif 1>x>= 0.9: + elif 1 > x >= 0.9: ylxg = 0.8 - elif 0.9>x>=0.6: + elif 0.9 > x >= 0.6: ylxg = 0.5 else: ylxg = 0.1 - #演练等级 + # 演练等级 cs = getcs(companyid) - score = ((a.filter(drilllevel__dicid=20).count()*1 + a.filter(drilllevel__dicid=21).count()*0.5 + a.filter(drilllevel__dicid=22).count()*1 + a.filter(drilllevel__dicid=43).count()*0.5)*cs['yldjqz'] + - ylxg*cs['ylxgqz'] ) #演练效果 + score = ( + a.filter(drilllevel__dicid=20).count() * 1 + + a.filter(drilllevel__dicid=21).count() * 0.5 + + a.filter(drilllevel__dicid=22).count() * 1 + + a.filter(drilllevel__dicid=43).count() * 0.5 + ) * cs["yldjqz"] + ylxg * cs["ylxgqz"] # 演练效果 year = first_day.year month = first_day.month - obj,created = Yjyc.objects.get_or_create(usecomp__partid=companyid,year=year,month=month,defaults={'usecomp':Partment.objects.get(partid=companyid) ,'year':year,'month':month}) + obj, created = Yjyc.objects.get_or_create( + usecomp__partid=companyid, + year=year, + month=month, + defaults={ + "usecomp": Partment.objects.get(partid=companyid), + "year": year, + "month": month, + }, + ) obj.drillvalue = -score - obj.yjz = round(obj.troublevalue + obj.trainvalue + obj.drillvalue,2) + obj.yjz = round(obj.troublevalue + obj.trainvalue + obj.drillvalue, 2) obj.save() + @shared_task -def correct_yjz(year:int, month:int, companyid: int = 0): +def correct_yjz(year: int = 0, month: int = 0, companyid: int = 0): # 重新更新某个月的预警值 + if year and month: + pass + else: + now = datetime.datetime.now() + now_1_month_before = now - datetime.timedelta(days=now.day + 1) + year = now_1_month_before.year + month = now_1_month_before.month days_num = calendar.monthrange(year, month)[1] first_day = datetime.date(year, month, 1) - first_day_of_next_month = first_day + datetime.timedelta(days = days_num) + first_day_of_next_month = first_day + datetime.timedelta(days=days_num) if companyid == 0: companys = Partment.objects.filter(iscompany=1) else: @@ -198,42 +359,50 @@ def correct_yjz(year:int, month:int, companyid: int = 0): yjjs_gc(i.partid, first_day, first_day_of_next_month) yjjs_ws(i.partid, first_day, first_day_of_next_month) yjjs_yl(i.partid, first_day, first_day_of_next_month) + ycjs(first_day.year, first_day.month) + @shared_task -def ycjs(): - #print('正在执行预测计算。。。') +def ycjs(nowyear=datetime.datetime.now().year, nowmonth=datetime.datetime.now().month): + # print('正在执行预测计算。。。') companys = Partment.objects.filter(iscompany=1) - nowyear = datetime.datetime.now().year - nowmonth = datetime.datetime.now().month - objs = Yjyc.objects.exclude(yjz=0,year=nowyear,month=nowmonth) + objs = Yjyc.objects.exclude(yjz=0, year=nowyear, month=nowmonth) for x in companys: - obj = objs.filter(usecomp=x).order_by('-yjycid')[0:5] # 只取半年内的 - objv = obj.values('yjycid','yjz') - #print(objv) - if len(obj)>1: + obj = objs.filter(usecomp=x).order_by("-yjycid")[0:5] # 只取半年内的 + objv = obj.values("yjycid", "yjz") + # print(objv) + if len(obj) > 1: vl = list(objv) for i in range(len(vl)): - vl[i]['num'] = i+1 + vl[i]["num"] = i + 1 if i == 0: - vl[i]['yjzs'] = vl[i]['yjz'] - if i > 0 : - vl[i]['yjzs'] = vl[i-1]['yjzs'] + vl[i]['yjz'] + vl[i]["yjzs"] = vl[i]["yjz"] + if i > 0: + vl[i]["yjzs"] = vl[i - 1]["yjzs"] + vl[i]["yjz"] df = pd.DataFrame(vl) # 建立线性回归模型 regr = linear_model.LinearRegression() # 拟合 - regr.fit(df['num'].values.reshape(-1, 1), df['yjzs']) # 注意此处.reshape(-1, 1)，因为X是一维的 + regr.fit( + df["num"].values.reshape(-1, 1), df["yjzs"] + ) # 注意此处.reshape(-1, 1)，因为X是一维的 # 得到直线的斜率、截距 a, b = regr.coef_, regr.intercept_ # 更新本月的预测值 # first_day,first_day_of_next_month = gettime() # year = first_day_of_next_month.year # month = first_day_of_next_month.month - objnew,created = Yjyc.objects.get_or_create(usecomp=x,year=nowyear,month=nowmonth,defaults={'usecomp':x,'year':nowyear,'month':nowmonth}) - objnew.ycz = ("%.2f" % a) - objnew.b = ("%.2f" % b) + objnew, created = Yjyc.objects.get_or_create( + usecomp=x, + year=nowyear, + month=nowmonth, + defaults={"usecomp": x, "year": nowyear, "month": nowmonth}, + ) + objnew.ycz = "%.2f" % a + objnew.b = "%.2f" % b objnew.save() - #print('执行完毕!') + # print('执行完毕!') + @shared_task def risktask(): @@ -241,28 +410,34 @@ def risktask(): for i in Risk.objects.exclude(tasktype=0): if i.tasktype == 1: if i.tasktime: - if (nowtime - i.tasktime).seconds>=28800: + if (nowtime - i.tasktime).seconds >= 28800: taskexpire = nowtime + datetime.timedelta(seconds=28800) Risktask.objects.filter(risk=i).update(usable=0) - Risktask.objects.create(risk=i,group=i.group,taskexpire = taskexpire,taskadd=nowtime) + Risktask.objects.create( + risk=i, group=i.group, taskexpire=taskexpire, taskadd=nowtime + ) i.tasktime = nowtime i.save() elif i.tasktype == 2: if i.tasktime: - if (nowtime - i.tasktime).days>=1: + if (nowtime - i.tasktime).days >= 1: taskexpire = nowtime + datetime.timedelta(days=1) Risktask.objects.filter(risk=i).update(usable=0) - Risktask.objects.create(risk=i,group=i.group,taskexpire = taskexpire,taskadd=nowtime) + Risktask.objects.create( + risk=i, group=i.group, taskexpire=taskexpire, taskadd=nowtime + ) i.tasktime = nowtime i.save() elif i.tasktype == 3: if i.tasktime: - if (nowtime - i.tasktime).days>=7: + if (nowtime - i.tasktime).days >= 7: taskexpire = nowtime + datetime.timedelta(days=7) Risktask.objects.filter(risk=i).update(usable=0) - Risktask.objects.create(risk=i,group=i.group,taskexpire = taskexpire,taskadd=nowtime) + Risktask.objects.create( + risk=i, group=i.group, taskexpire=taskexpire, taskadd=nowtime + ) i.tasktime = nowtime - i.save() + i.save() # elif i.tasktype == 4: # if i.tasktime: # if (nowtime - i.tasktime).weeks>=4: @@ -270,35 +445,44 @@ def risktask(): # Risktask.objects.filter(risk=i).update(usable=0) # Risktask.objects.create(risk=i,group=i.group,taskexpire = taskexpire) # i.tasktime = taskexpire - # i.save() + # i.save() + + @shared_task def riskacttask(): nowtime = datetime.datetime.now() for i in RiskAct.objects.exclude(tasktype=0): if i.tasktype == 1: if i.tasktime: - if (nowtime - i.tasktime).seconds>=28800: + if (nowtime - i.tasktime).seconds >= 28800: taskexpire = nowtime + datetime.timedelta(seconds=28800) - RiskActTask.objects.filter(riskact=i,istask=1).update(usable=0) - RiskActTask.objects.create(riskact=i,taskexpire = taskexpire,taskadd=nowtime,istask=1) + RiskActTask.objects.filter(riskact=i, istask=1).update(usable=0) + RiskActTask.objects.create( + riskact=i, taskexpire=taskexpire, taskadd=nowtime, istask=1 + ) i.tasktime = nowtime i.save() elif i.tasktype == 2: if i.tasktime: - if (nowtime - i.tasktime).days>=1: + if (nowtime - i.tasktime).days >= 1: taskexpire = nowtime + datetime.timedelta(days=1) - RiskActTask.objects.filter(riskact=i,istask=1).update(usable=0) - RiskActTask.objects.create(riskact=i,taskexpire = taskexpire,taskadd=nowtime,istask=1) + RiskActTask.objects.filter(riskact=i, istask=1).update(usable=0) + RiskActTask.objects.create( + riskact=i, taskexpire=taskexpire, taskadd=nowtime, istask=1 + ) i.tasktime = nowtime i.save() elif i.tasktype == 3: if i.tasktime: - if (nowtime - i.tasktime).days>=7: + if (nowtime - i.tasktime).days >= 7: taskexpire = nowtime + datetime.timedelta(days=7) - RiskActTask.objects.filter(riskact=i,istask=1).update(usable=0) - RiskActTask.objects.create(riskact=i,taskexpire = taskexpire,taskadd=nowtime,istask=1) + RiskActTask.objects.filter(riskact=i, istask=1).update(usable=0) + RiskActTask.objects.create( + riskact=i, taskexpire=taskexpire, taskadd=nowtime, istask=1 + ) i.tasktime = nowtime - i.save() + i.save() + @shared_task def gridtasksend(): @@ -309,31 +493,42 @@ def gridtasksend(): for i in GridTaskSet.objects.filter(is_paused=False, start_task____lte=time2): sendGridtask(i) + from dateutil.relativedelta import * + def sendGridtask(obj): nowtime = datetime.datetime.now() schedule = obj.schedule - if 'interval' in schedule and schedule['interval']: - number, type_ = schedule['interval'].split(',') + if "interval" in schedule and schedule["interval"]: + number, type_ = schedule["interval"].split(",") start_task = obj.start_task - m = start_task + relativedelta({type_:int(number)}) + m = start_task + relativedelta({type_: int(number)}) if nowtime >= m: - newm = nowtime + relativedelta({type_:int(number)}) + newm = nowtime + relativedelta({type_: int(number)}) RiskActTask.objects.filter(taskset=obj, usable=1).update(usable=0) - RiskActTask.objects.create(riskact=obj.riskact,taskexpire = newm,taskadd=nowtime,istask=1,taskset=obj,tasknote=obj.note,user=obj.user) + RiskActTask.objects.create( + riskact=obj.riskact, + taskexpire=newm, + taskadd=nowtime, + istask=1, + taskset=obj, + tasknote=obj.note, + user=obj.user, + ) obj.start_task = nowtime obj.count = obj.count + 1 obj.save() else: pass + @shared_task def sendGridtask2(**kwargs): - obj = GridTaskSet.objects.get(id=kwargs['gridtaskset']) + obj = GridTaskSet.objects.get(id=kwargs["gridtaskset"]) if obj.trouble: if obj.trouble.yhzt == 6: - #如果隐患关闭了就直接停止计划任务 + # 如果隐患关闭了就直接停止计划任务 obj.periodictask.enabled = False obj.periodictask.save() else: @@ -341,11 +536,26 @@ def sendGridtask2(**kwargs): nowtime = datetime.datetime.now() if obj.expire: newm = nowtime + datetime.timedelta(hours=int(obj.expire)) - TroubleFollowTask.objects.create(trouble=obj.trouble,taskexpire = newm,taskadd=nowtime,taskset=obj,tasknote=obj.note,user=obj.user,gridlevel=obj.gridlevel) + TroubleFollowTask.objects.create( + trouble=obj.trouble, + taskexpire=newm, + taskadd=nowtime, + taskset=obj, + tasknote=obj.note, + user=obj.user, + gridlevel=obj.gridlevel, + ) else: - TroubleFollowTask.objects.create(trouble=obj.trouble,taskadd=nowtime,taskset=obj,tasknote=obj.note,user=obj.user,gridlevel=obj.gridlevel) + TroubleFollowTask.objects.create( + trouble=obj.trouble, + taskadd=nowtime, + taskset=obj, + tasknote=obj.note, + user=obj.user, + gridlevel=obj.gridlevel, + ) obj.last_run_at = nowtime - obj.count = obj.count+1 + obj.count = obj.count + 1 obj.save() elif obj.riskact: @@ -353,114 +563,188 @@ def sendGridtask2(**kwargs): nowtime = datetime.datetime.now() if obj.expire: newm = nowtime + datetime.timedelta(hours=int(obj.expire)) - RiskActTask.objects.create(riskact=obj.riskact,taskexpire = newm,taskadd=nowtime,istask=1,taskset=obj,tasknote=obj.note,user=obj.user) + RiskActTask.objects.create( + riskact=obj.riskact, + taskexpire=newm, + taskadd=nowtime, + istask=1, + taskset=obj, + tasknote=obj.note, + user=obj.user, + ) else: - RiskActTask.objects.create(riskact=obj.riskact,taskadd=nowtime,istask=1,taskset=obj,tasknote=obj.note,user=obj.user) + RiskActTask.objects.create( + riskact=obj.riskact, + taskadd=nowtime, + istask=1, + taskset=obj, + tasknote=obj.note, + user=obj.user, + ) obj.last_run_at = nowtime - obj.count = obj.count+1 + obj.count = obj.count + 1 obj.save() + @shared_task def expireRiskacttask(): - RiskActTask.objects.filter(taskexpire__lte = datetime.datetime.now(), usable=1).update(usable=0) + RiskActTask.objects.filter( + taskexpire__lte=datetime.datetime.now(), usable=1 + ).update(usable=0) @shared_task def expireTroublefollowtask(): - TroubleFollowTask.objects.filter(taskexpire__lte = datetime.datetime.now(), usable=1).update(usable=0) + TroubleFollowTask.objects.filter( + taskexpire__lte=datetime.datetime.now(), usable=1 + ).update(usable=0) + @shared_task def closeExamtest(): # 自动关闭到期的考试 - ExamTest.objects.filter(state=1, endtime__lt = datetime.datetime.now()).update(state=0) + ExamTest.objects.filter(state=1, endtime__lt=datetime.datetime.now()).update( + state=0 + ) + @shared_task def checktask(): nowtime = datetime.datetime.now() - for i in Checktask.objects.filter(deletemark=1,taskstate=1): - if i.tasktype==1: + for i in Checktask.objects.filter(deletemark=1, taskstate=1): + if i.tasktype == 1: if i.checktime: - - if (nowtime - i.checktime).days>=1: + if (nowtime - i.checktime).days >= 1: endtimes = nowtime + datetime.timedelta(days=1) for j in i.checkname.all(): - x=User.objects.get(userid=j.userid) - - Checkjob.objects.filter(checktask=i,checkname=x).update(jobstate=2) - Checkjob.objects.create(checktask=i,checkname=x,taskstate=4,starttime=nowtime,endtime = endtimes,usecomp=i.usecomp) - + x = User.objects.get(userid=j.userid) + + Checkjob.objects.filter(checktask=i, checkname=x).update( + jobstate=2 + ) + Checkjob.objects.create( + checktask=i, + checkname=x, + taskstate=4, + starttime=nowtime, + endtime=endtimes, + usecomp=i.usecomp, + ) + i.checktime = nowtime i.save() - elif i.tasktype==2: + elif i.tasktype == 2: if i.checktime: - - if (nowtime - i.checktime).days>=7: + if (nowtime - i.checktime).days >= 7: endtimes = nowtime + datetime.timedelta(days=7) for j in i.checkname.all(): - x=User.objects.get(userid=j.userid) - - Checkjob.objects.filter(checktask=i,checkname=x).update(jobstate=2) - Checkjob.objects.create(checktask=i,checkname=x,taskstate=4,starttime=nowtime,endtime = endtimes,usecomp=i.usecomp) - + x = User.objects.get(userid=j.userid) + + Checkjob.objects.filter(checktask=i, checkname=x).update( + jobstate=2 + ) + Checkjob.objects.create( + checktask=i, + checkname=x, + taskstate=4, + starttime=nowtime, + endtime=endtimes, + usecomp=i.usecomp, + ) + i.checktime = nowtime i.save() - elif i.tasktype==3: + elif i.tasktype == 3: if i.checktime: - - if (nowtime - i.checktime).days>=30: + if (nowtime - i.checktime).days >= 30: endtimes = nowtime + datetime.timedelta(days=30) for j in i.checkname.all(): - x=User.objects.get(userid=j.userid) - - Checkjob.objects.filter(checktask=i,checkname=x).update(jobstate=2) - Checkjob.objects.create(checktask=i,checkname=x,taskstate=4,starttime=nowtime,endtime = endtimes,usecomp=i.usecomp) - + x = User.objects.get(userid=j.userid) + + Checkjob.objects.filter(checktask=i, checkname=x).update( + jobstate=2 + ) + Checkjob.objects.create( + checktask=i, + checkname=x, + taskstate=4, + starttime=nowtime, + endtime=endtimes, + usecomp=i.usecomp, + ) + i.checktime = nowtime i.save() - elif i.tasktype==4: + elif i.tasktype == 4: if i.checktime: - - if (nowtime - i.checktime).days>=120: + if (nowtime - i.checktime).days >= 120: endtimes = nowtime + datetime.timedelta(days=120) for j in i.checkname.all(): - x=User.objects.get(userid=j.userid) - - Checkjob.objects.filter(checktask=i,checkname=x).update(jobstate=2) - Checkjob.objects.create(checktask=i,checkname=x,taskstate=4,starttime=nowtime,endtime = endtimes,usecomp=i.usecomp) - + x = User.objects.get(userid=j.userid) + + Checkjob.objects.filter(checktask=i, checkname=x).update( + jobstate=2 + ) + Checkjob.objects.create( + checktask=i, + checkname=x, + taskstate=4, + starttime=nowtime, + endtime=endtimes, + usecomp=i.usecomp, + ) + i.checktime = nowtime i.save() - elif i.tasktype==5: + elif i.tasktype == 5: if i.checktime: - - if (nowtime - i.checktime).days>=182: + if (nowtime - i.checktime).days >= 182: endtimes = nowtime + datetime.timedelta(days=182) for j in i.checkname.all(): - x=User.objects.get(userid=j.userid) - - Checkjob.objects.filter(checktask=i,checkname=x).update(jobstate=2) - Checkjob.objects.create(checktask=i,checkname=x,taskstate=4,starttime=nowtime,endtime = endtimes,usecomp=i.usecomp) - + x = User.objects.get(userid=j.userid) + + Checkjob.objects.filter(checktask=i, checkname=x).update( + jobstate=2 + ) + Checkjob.objects.create( + checktask=i, + checkname=x, + taskstate=4, + starttime=nowtime, + endtime=endtimes, + usecomp=i.usecomp, + ) + i.checktime = nowtime i.save() - elif i.tasktype==6: + elif i.tasktype == 6: if i.checktime: - - if (nowtime - i.checktime).days>=365: + if (nowtime - i.checktime).days >= 365: endtimes = nowtime + datetime.timedelta(days=365) for j in i.checkname.all(): - x=User.objects.get(userid=j.userid) - - Checkjob.objects.filter(checktask=i,checkname=x).update(jobstate=2) - Checkjob.objects.create(checktask=i,checkname=x,taskstate=4,starttime=nowtime,endtime = endtimes,usecomp=i.usecomp) - + x = User.objects.get(userid=j.userid) + + Checkjob.objects.filter(checktask=i, checkname=x).update( + jobstate=2 + ) + Checkjob.objects.create( + checktask=i, + checkname=x, + taskstate=4, + starttime=nowtime, + endtime=endtimes, + usecomp=i.usecomp, + ) + i.checktime = nowtime i.save() @shared_task def expireTzzs(): - queryset = Socertificate.objects.filter(yfsrq__lte = datetime.datetime.now()+datetime.timedelta(days=185)) + queryset = Socertificate.objects.filter( + yfsrq__lte=datetime.datetime.now() + datetime.timedelta(days=185) + ) for x in queryset: try: x.zszt = guoqi(x.yfsrq) @@ -468,9 +752,12 @@ def expireTzzs(): except: pass + @shared_task def expireAqzs(): - queryset = Safecert.objects.filter(yfsrq__lte = datetime.datetime.now()+datetime.timedelta(days=185)) + queryset = Safecert.objects.filter( + yfsrq__lte=datetime.datetime.now() + datetime.timedelta(days=185) + ) for x in queryset: try: x.zszt = guoqi(x.yfsrq) @@ -478,43 +765,52 @@ def expireAqzs(): except: pass + @shared_task def updateTzzs(): - queryset = Socertificate.objects.filter(yfsrq__lte = datetime.datetime.now()+datetime.timedelta(days=200)) + queryset = Socertificate.objects.filter( + yfsrq__lte=datetime.datetime.now() + datetime.timedelta(days=200) + ) for x in queryset: - data = getTzzs(x.cardnum,x.realname) + data = getTzzs(x.cardnum, x.realname) time.sleep(10) if data: for i in data: - updated_values={ - 'realname':i['姓名'], - 'gender':i['性别'], - 'zylb':i['作业类别'], - 'czxm':i['操作项目'], - 'fzjg':i['发证机关'], - 'ccfzrq':i['初次发证日期'] if i['初次发证日期'] else None, - 'yfsrq':i['应复审日期'] if i['应复审日期'] else None, - 'yxqkssj':i['有效期开始时间'] if i['有效期开始时间'] else None, - 'yxqjssj':i['有效期结束时间'] if i['有效期结束时间'] else None, - 'sjfssj':i['实际复审时间'] if i['实际复审时间'] else None, - 'user':x.user, - 'usecomp':x.usecomp + updated_values = { + "realname": i["姓名"], + "gender": i["性别"], + "zylb": i["作业类别"], + "czxm": i["操作项目"], + "fzjg": i["发证机关"], + "ccfzrq": i["初次发证日期"] if i["初次发证日期"] else None, + "yfsrq": i["应复审日期"] if i["应复审日期"] else None, + "yxqkssj": i["有效期开始时间"] if i["有效期开始时间"] else None, + "yxqjssj": i["有效期结束时间"] if i["有效期结束时间"] else None, + "sjfssj": i["实际复审时间"] if i["实际复审时间"] else None, + "user": x.user, + "usecomp": x.usecomp, } try: - updated_values['zszt'] = guoqi(i['应复审日期']) + updated_values["zszt"] = guoqi(i["应复审日期"]) except: pass # obj, created = Socertificate.objects.update_or_create( # cardnum=x.cardnum, czxm=i['操作项目'], defaults=updated_values) try: - obj = Socertificate.objects.get(cardnum=x.cardnum, czxm=i['操作项目']) - if obj.yxqjssj.strftime('%Y-%m-%d')now1: + if datetime.datetime.strptime(x, "%Y-%m-%d") > now1: return 1 - elif now1 > datetime.datetime.strptime(x,'%Y-%m-%d')>now: + elif now1 > datetime.datetime.strptime(x, "%Y-%m-%d") > now: return 2 else: - return 3 + return 3 + @shared_task def updateAqzs(): - queryset = Safecert.objects.filter(yfsrq__lte = datetime.datetime.now()+datetime.timedelta(days=200)) + queryset = Safecert.objects.filter( + yfsrq__lte=datetime.datetime.now() + datetime.timedelta(days=200) + ) for x in queryset: - data = getAqzs(x.cardnum,x.realname) + data = getAqzs(x.cardnum, x.realname) time.sleep(10) if data: for i in data: - updated_values={ - 'realname':i['姓名'], - 'gender':i['性别'], - 'zglx':i['资格类型'], - 'dwlx':i['单位类型'], - 'fzjg':i['发证机关'], - 'yfsrq':i['应复审日期'] if i['应复审日期'] else None, - 'yxqkssj':i['有效期开始时间'] if i['有效期开始时间'] else None, - 'yxqjssj':i['有效期结束时间'] if i['有效期结束时间'] else None, - 'user':x.user, - 'usecomp':x.usecomp + updated_values = { + "realname": i["姓名"], + "gender": i["性别"], + "zglx": i["资格类型"], + "dwlx": i["单位类型"], + "fzjg": i["发证机关"], + "yfsrq": i["应复审日期"] if i["应复审日期"] else None, + "yxqkssj": i["有效期开始时间"] if i["有效期开始时间"] else None, + "yxqjssj": i["有效期结束时间"] if i["有效期结束时间"] else None, + "user": x.user, + "usecomp": x.usecomp, } try: - updated_values['zszt'] = guoqi(i['有效期结束时间']) + updated_values["zszt"] = guoqi(i["有效期结束时间"]) except: pass # obj, created = Safecert.objects.update_or_create( # cardnum=x.cardnum, zglx=i['资格类型'], dwlx=i['单位类型'], defaults=updated_values) try: - obj = Safecert.objects.get(cardnum=x.cardnum, zglx=i['资格类型'], dwlx=i['单位类型']) - if obj.yxqjssj.strftime('%Y-%m-%d')