import sys,json
|
from pathlib import Path
|
from datetime import datetime
|
from collections import defaultdict
|
sys.stdout.reconfigure(encoding='utf-8')
|
sys.path.insert(0,str(Path(__file__).parent/'iqc-import-20261012/deps'))
|
import pymysql
|
apply='--apply' in sys.argv
|
tables=['mes_pro_work_order','mes_pro_work_order_process','mes_pro_route_product','mes_pro_route_process','mes_md_workstation','mes_md_workstation_machine','mes_dv_machinery','mes_pro_task']
|
with pymysql.connect(host='36.213.150.237',port=10000,user='root',password=sys.stdin.readline().strip(),database='mom-hm',charset='utf8mb4',connect_timeout=15,cursorclass=pymysql.cursors.DictCursor) as c:
|
with c.cursor() as q:
|
q.execute('SELECT DATABASE() db');assert q.fetchone()['db']=='mom-hm'
|
data={}
|
for t in tables:
|
q.execute('SELECT * FROM '+t+' WHERE deleted=0 ORDER BY id'+(' FOR UPDATE' if apply else ''))
|
data[t]=q.fetchall()
|
links=defaultdict(set)
|
for r in data['mes_md_workstation_machine']:links[r['workstation_id']].add(r['machinery_id'])
|
for m in data['mes_dv_machinery']:
|
if m['workstation_id']:links[m['workstation_id']].add(m['id'])
|
needed_processes=set()
|
orders=[o for o in data['mes_pro_work_order'] if o['status']==1 and o['type']==1]
|
for o in orders:
|
processes=[p for p in data['mes_pro_work_order_process'] if p['work_order_id']==o['id']]
|
if not processes:
|
route=o['route_id']
|
if not route:
|
relations=[r for r in data['mes_pro_route_product'] if r.get('item_id')==o['product_id']]
|
route=relations[0]['route_id'] if relations else None
|
processes=[p for p in data['mes_pro_route_process'] if p['route_id']==route]
|
needed_processes.update(p['process_id'] for p in processes)
|
needed=set().union(*(links[s['id']] for s in data['mes_md_workstation'] if s['status']==0 and s['process_id'] in needed_processes))
|
active=[t for t in data['mes_pro_task'] if t['status'] not in (4,5)]
|
occupied=set().union(*(links[t['workstation_id']] for t in active)) if active else set()
|
changes=[m for m in data['mes_dv_machinery'] if m['id'] in needed-occupied and m['status']==2]
|
print(json.dumps({'confirmedOrders':len(orders),'neededMachines':len(needed),'preservedOccupied':sorted(needed&occupied),'changeToStopped':[{'id':m['id'],'name':m['name']} for m in changes]},ensure_ascii=False))
|
if apply:
|
out=Path(__file__).parent/'schedule-diagnosis-20261015';out.mkdir(exist_ok=True)
|
stamp=datetime.now().strftime('%Y%m%d%H%M%S')
|
(out/f'machines-before-{stamp}.json').write_text(json.dumps(changes,ensure_ascii=False,default=str,indent=2),encoding='utf-8')
|
for m in changes:
|
q.execute('UPDATE mes_dv_machinery SET status=1,update_time=NOW() WHERE id=%s AND deleted=0 AND status=2',(m['id'],));assert q.rowcount==1
|
q.execute('SELECT * FROM mes_dv_machinery WHERE deleted=0 ORDER BY id')
|
after={m['id']:m for m in q.fetchall()}
|
changed={m['id'] for m in changes}
|
for old in data['mes_dv_machinery']:
|
new=after[old['id']]
|
if old['id'] in changed:
|
assert new['status']==1
|
assert {k:v for k,v in old.items() if k not in ('status','update_time')}=={k:v for k,v in new.items() if k not in ('status','update_time')}
|
else:assert new==old
|
c.commit()
|
print(json.dumps({'updated':len(changes),'preservedOccupied':len(needed&occupied),'verified':True}))
|
