"""Raise the four October supplemental chains to 200, atomically across all modules."""
|
import sys,json,re,collections
|
from pathlib import Path
|
from decimal import Decimal,ROUND_HALF_UP
|
sys.stdout.reconfigure(encoding='utf-8')
|
sys.path.insert(0,str(Path('scripts/iqc-import-20261012/deps').resolve()))
|
import pymysql
|
P=Path(__file__).parent;D=Decimal;apply='--apply' in sys.argv
|
normalize=json.loads((P/'normalize-applied.json').read_text(encoding='utf8'))
|
manifest=json.loads((P/'result-applied.json').read_text(encoding='utf8'))
|
for link in manifest['links']:
|
for k,v in list(link.items()):
|
if isinstance(v,str) and v in normalize['mapping']:link[k]=normalize['mapping'][v]
|
schema=json.loads((P/'schema-snapshot.json').read_text(encoding='utf8'))
|
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,autocommit=False,read_timeout=90)
|
def dump(name,obj):(P/name).write_text(json.dumps(obj,ensure_ascii=False,indent=2,default=str),encoding='utf8')
|
def money(v):return D(v).quantize(D('.01'),rounding=ROUND_HALF_UP)
|
rows={};changes=[];log=[];summaries=[]
|
with c.cursor() as q:
|
try:
|
q.execute('SELECT GET_LOCK(%s,10)',('october-chain-production-floor',));assert list(q.fetchone().values())==[1]
|
for t,ids in normalize['scoped_ids'].items():
|
q.execute('SELECT * FROM `'+t+'` WHERE id IN ('+','.join(['%s']*len(ids))+') FOR UPDATE',tuple(ids));rows[t]=q.fetchall();assert len(rows[t])==len(ids)
|
dump('before-production-floor-'+('apply' if apply else 'preview')+'.json',rows)
|
byid={t:{x['id']:x for x in rr} for t,rr in rows.items()}
|
day_factors={};new_totals={}
|
for link in manifest['links']:
|
mp=byid['mes_pro_mps'][link['mps_id']];wo=byid['mes_pro_work_order'][link['work_order_id']];order=byid['erp_sale_order'][link['sale_order_id']];oi=byid['erp_sale_order_items'][link['order_item_id']]
|
old=mp['quantity'];assert old>0 and mp['quantity_produced']==wo['quantity']==wo['quantity_produced']==oi['count']==oi['out_count']==order['total_count']==order['out_count']==old
|
assert old==D(link['quantity']),('Dataset already changed',link['sale_order_no'],old)
|
target=max(old,D(200));factor=target/old;day_factors[link['date']]=factor
|
net=money(oi['product_price']*target);tax=money(net*oi['tax_percent']/100);total=net+tax
|
new_totals[link['date']]=(net,tax,total)
|
summaries.append({'sales_order':order['no'],'product':link['product'],'unit_id':link['unit_id'],'old_quantity':old,'new_quantity':target,'old_total':order['total_price'],'new_total':total})
|
link['quantity']=target;link['total']=total
|
# Scale only transaction quantities, never prices, percentages, sample readings or template limits.
|
quantity_tables={'crm_business_product','crm_sale_quotation_product','crm_contract_product','erp_sale_order','erp_sale_order_items','mes_pro_mps','mes_pro_work_order','mes_pro_work_order_bom','mes_pro_task','mes_pro_feedback','mes_pro_task_issue','mes_wm_arrival_notice_line','mes_qc_iqc','mes_qc_ipqc','mes_qc_oqc','mes_wm_item_receipt_line','mes_wm_item_receipt_detail','mes_wm_product_issue_line','mes_wm_product_issue_detail','mes_wm_item_consume_line','mes_wm_item_consume_detail','mes_wm_product_produce_line','mes_wm_product_produce_detail','mes_wm_product_receipt_line','mes_wm_product_receipt_detail','mes_wm_sales_notice_line','mes_wm_product_sales_line','mes_wm_product_sales_detail','mes_wm_stock_reserve','mes_wm_transaction','mes_wm_material_stock'}
|
for t,rr in rows.items():
|
decimals={x['Field']:int(re.search(r'decimal\(\d+,(\d+)\)',x['Type']).group(1)) for x in schema[t]['columns'] if x['Type'].startswith('decimal')}
|
for row in rr:
|
day=row['create_time'].date().isoformat();factor=day_factors.get(day,D(1));v={}
|
if factor==1:continue
|
if t in quantity_tables:
|
for k,old in row.items():
|
isquantity=k=='quantity' or k.endswith('_quantity') or k in ['count','total_count','out_count','quantity_produced','quantity_scheduled','quantity_changed']
|
if not isquantity or k.startswith(('min_','max_')) or old is None or old==0:continue
|
assert isinstance(old,(D,int)),(t,k)
|
value=D(old)*factor
|
value=value.quantize(D(1).scaleb(-decimals[k]),rounding=ROUND_HALF_UP) if k in decimals else int(value)
|
if value!=old:v[k]=value
|
net,tax,total=new_totals[day]
|
if t=='crm_business':v.update(total_product_price=net,total_price=total)
|
elif t=='crm_sale_quotation':v.update(total_product_price=net,tax_price=tax,total_price=total)
|
elif t=='crm_contract':v.update(total_product_price=total,total_price=total)
|
elif t=='erp_sale_order':v.update(total_product_price=net,total_tax_price=tax,total_price=total)
|
elif t in ['crm_business_product','crm_sale_quotation_product','erp_sale_order_items']:
|
v['total_price']=net
|
if 'tax_price' in row:v['tax_price']=tax
|
elif t=='crm_contract_product':
|
v['total_price']=total;assert money(row['contract_price']*v['count'])==total
|
elif t=='crm_receivable_plan':
|
assert row['receivable_id'] is None and row['period'] in [1,2]
|
first=money(total*D('.3'));v['price']=first if row['period']==1 else total-first
|
if not v:continue
|
sql='UPDATE `'+t+'` SET '+','.join('`'+k+'`=%s' for k in v)+',update_time=update_time WHERE id=%s'
|
args=tuple(v.values())+(row['id'],);q.execute(sql,args);assert q.rowcount==1
|
log.append(q.mogrify(sql,args)+';');changes.append({'table':t,'id':row['id'],'before':{k:row[k] for k in v},'after':v});byid[t][row['id']]={**row,**v}
|
for link in manifest['links']:
|
target=link['quantity'];day=link['date'];net,tax,total=new_totals[day]
|
for t,key,field in [('mes_pro_mps','mps_id','quantity_produced'),('mes_pro_work_order','work_order_id','quantity_produced'),('erp_sale_order','sale_order_id','out_count'),('erp_sale_order_items','order_item_id','out_count')]:assert byid[t][link[key]][field]==target
|
for tid in link['tasks']:
|
task=byid['mes_pro_task'][tid];assert task['quantity']==task['produced_quantity']==task['qualify_quantity']==target
|
for fid in link['feedbacks']:
|
fb=byid['mes_pro_feedback'][fid];assert fb['scheduled_quantity']==fb['feedback_quantity']==fb['qualified_quantity']==target
|
assert sum(byid['crm_receivable_plan'][pid]['price'] for pid in link['receivable_plans'])==total
|
assert byid['mes_qc_oqc'][link['oqc_id']]['out_quantity']==byid['mes_qc_oqc'][link['oqc_id']]['qualified_quantity']==target
|
balances=collections.defaultdict(lambda:D(0))
|
for row in byid['mes_wm_transaction'].values():balances[row['material_stock_id']]+=row['quantity']
|
assert len(balances)==62 and all(x==0 for x in balances.values()),'Inventory flows must balance'
|
for row in byid['mes_wm_material_stock'].values():assert row['quantity']==row['reserved_quantity']==0
|
# All fields outside the explicit change set must remain unchanged.
|
expected_changes={(x['table'],x['id']):x['after'] for x in changes}
|
for t,rr in rows.items():
|
for row in rr:
|
q.execute('SELECT * FROM `'+t+'` WHERE id=%s',(row['id'],));assert q.fetchone()=={**row,**expected_changes.get((t,row['id']),{})},(t,row['id'])
|
result={'applied':apply,'minimum_quantity':200,'chains':summaries,'changed_rows':len(changes),'changes':changes,'inventory_balanced':True,'unit_prices_unchanged':True}
|
dump('production-floor-'+('applied' if apply else 'preview')+'.json',result)
|
(P/('production-floor-applied.sql' if apply else 'production-floor-preview.sql')).write_text('\n'.join(log),encoding='utf8')
|
if apply:
|
c.commit();dump('result-current.json',manifest)
|
else:c.rollback()
|
print(json.dumps({k:result[k] for k in ['applied','minimum_quantity','chains','changed_rows','inventory_balanced','unit_prices_unchanged']},ensure_ascii=False,default=str))
|
except:
|
c.rollback();raise
|
finally:
|
q.execute('SELECT RELEASE_LOCK(%s)',('october-chain-production-floor',))
|
c.close()
|
