1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
| from pathlib import Path
| import sys,re,json
| sys.stdout.reconfigure(encoding='utf-8')
| sys.path.insert(0,str(Path('scripts/iqc-import-20261012/deps').resolve()))
| import pymysql
| s=Path('yudao-server/src/main/resources/application-local.yaml').read_text(encoding='utf-8')
| x=s[s.index('jdbc:mysql://localhost:3306/'):]
| pw=re.search(r'password:\s*([^\r\n#]+)',x).group(1).strip().strip('\"\'')
| c=pymysql.connect(host='127.0.0.1',port=3306,user='root',password=pw,database='mom-hm',charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor,autocommit=False)
| try:
| with c.cursor() as q:
| if '--seed' in sys.argv:
| from decimal import Decimal
| out=Path('scripts/local-production-20261015');out.mkdir(exist_ok=True)
| assert not (out/'result.json').exists(), 'Already seeded; inspect result.json instead of duplicating'
| q.execute('select count(*) n from mes_pro_task where deleted=0 and (status is null or status not in (4,5))')
| assert q.fetchone()['n']==0, 'Active tasks require resource review'
| q.execute('select distinct m.* from mes_dv_machinery m left join mes_md_workstation_machine wm on wm.machinery_id=m.id and wm.deleted=0 where m.deleted=0 and (m.workstation_id in (11,12,13) or wm.workstation_id in (11,12,13)) for update')
| machines=q.fetchall()
| assert machines and all(m['status'] in (1,2) and m['production_efficiency'] and m['production_efficiency']>0 for m in machines)
| (out/'equipment-before.json').write_text(json.dumps(machines,ensure_ascii=False,indent=2,default=str),encoding='utf-8')
| ids=[]; summary=[]
| for product,route,qty in [(497,36,200),(507,34,300),(497,36,500)]:
| q.execute('select * from mes_md_item where id=%s and deleted=0 and status=0',(product,));item=q.fetchone();assert item
| q.execute('select * from mes_pro_route_process where route_id=%s and deleted=0 order by sort',(route,));processes=q.fetchall();assert len(processes)==3
| for p in processes:
| q.execute('select count(*) n from mes_pro_process where id=%s and deleted=0 and status=0',(p['process_id'],));assert q.fetchone()['n']==1
| q.execute('select count(*) n from mes_pd_process_param_template_detail where template_id=%s and deleted=0',(p['param_template_id'],));assert q.fetchone()['n']>0
| q.execute("select result from mes_md_auto_code_record where rule_id=30 and result like 'WO_20261015%' union select code result from mes_pro_work_order where code like 'WO_20261015%'")
| serial=max([int(r['result'][-3:]) for r in q.fetchall()] or [0])+1
| code=f'WO_20261015{serial:03d}'
| q.execute('insert into mes_pro_work_order(code,name,type,order_source_type,product_id,quantity,quantity_produced,quantity_changed,quantity_scheduled,route_id,auto_finish_flag,request_date,status,creator,updater,is_feedback_approve,parent_id) values(%s,%s,1,2,%s,%s,0,0,0,%s,1,%s,1,%s,%s,0,0)',(code,item['name']+'生产工单',product,qty,route,'2026-10-23 18:00:00','244','244'))
| wid=q.lastrowid;ids.append(wid)
| q.execute('insert into mes_md_auto_code_record(rule_id,result,serial_no,creator,updater) values(30,%s,%s,%s,%s)',(code,serial,'244','244'))
| cols=['process_id','param_template_id','yield_rate','sort','next_process_id','link_type','prepare_time','wait_time','color_code','key_flag','check_flag','backflush_flag','output_item_id','remark']
| for p in processes:
| q.execute('insert into mes_pro_work_order_process(work_order_id,'+','.join(cols)+',creator,updater) values('+','.join(['%s']*(len(cols)+3))+')',[wid]+[p[k] for k in cols]+['244','244'])
| q.execute('select bom_item_id,quantity from mes_md_product_bom where item_id=%s and deleted=0',(product,));bom=q.fetchall();assert bom
| for b in bom:
| q.execute('select id from mes_md_item where id=%s and deleted=0',(b['bom_item_id'],));assert q.fetchone()
| q.execute('insert into mes_pro_work_order_bom(work_order_id,item_id,quantity,creator,updater) values(%s,%s,%s,%s,%s)',(wid,b['bom_item_id'],b['quantity']*Decimal(qty),'244','244'))
| summary.append(dict(id=wid,code=code,product=item['name'],quantity=qty,route_id=route,process_count=len(processes),bom_count=len(bom)))
| for m in machines:
| if m['status']==2:q.execute('update mes_dv_machinery set status=1 where id=%s and status=2',(m['id'],));assert q.rowcount==1
| q.execute('select count(*) n from mes_pro_task where deleted=0 and work_order_id in ('+','.join(['%s']*len(ids))+')',ids);assert q.fetchone()['n']==0
| c.commit()
| (out/'result.json').write_text(json.dumps({'database':'localhost:3306/mom-hm','orders':summary,'equipment_ids':[m['id'] for m in machines],'tasks_created':0},ensure_ascii=False,indent=2),encoding='utf-8')
| print(json.dumps(summary,ensure_ascii=False));sys.exit(0)
| for sql in sys.argv[1:]:
| q.execute(sql);print(json.dumps(q.fetchall(),ensure_ascii=False,default=str))
| finally:
| c.rollback();c.close()
|
|
