import json,sys
|
from pathlib import Path
|
P=Path(__file__).parent
|
sys.path.insert(0,str(P/'deps'))
|
import pymysql
|
result=json.loads((P/'import-result.json').read_text(encoding='utf8'))
|
standards={s['template_id']:s['standard'] for s in json.loads((P/'national-standards-plan.json').read_text(encoding='utf8'))['schemes'] if s['status']==0}
|
c=pymysql.connect(host='36.213.150.237',port=10000,user='root',password=sys.stdin.readline().strip(),database='mom-hm',charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
|
log=[]
|
try:
|
c.begin()
|
with c.cursor() as q:
|
for iid in result['inserted_ids']:
|
q.execute('SELECT id,template_id,execute_standard FROM mes_qc_iqc WHERE id=%s AND deleted=0 FOR UPDATE',(iid,));r=q.fetchone();assert r and r['template_id'] in standards
|
standard=standards[r['template_id']];assert r['execute_standard'] in (None,'',standard)
|
sql="UPDATE mes_qc_iqc SET execute_standard=%s,updater='1' WHERE id=%s AND deleted=0";args=(standard,iid)
|
log.append(q.mogrify(sql,args)+';');q.execute(sql,args)
|
(P/'national-iqc-execute-standard.sql').write_text('START TRANSACTION;\n'+'\n'.join(log)+'\nCOMMIT;\n',encoding='utf8')
|
c.commit()
|
q.execute('SELECT * FROM mes_qc_iqc WHERE deleted=0 ORDER BY id');rows=q.fetchall()
|
for r in rows:
|
if r['id'] in result['inserted_ids']:assert r['execute_standard']==standards[r['template_id']]
|
assert (r['inspect_date']-r['receive_date']).days==15
|
assert len(rows)==40
|
(P/'post-import.json').write_text(json.dumps(rows,ensure_ascii=False,default=str,indent=2),encoding='utf8')
|
print(json.dumps({'total_iqc':len(rows),'execute_standard_verified':len(result['inserted_ids']),'receive_date_verified':True}))
|
except Exception:c.rollback();raise
|
finally:c.close()
|
