import sys, json
|
from pathlib import Path
|
from datetime import datetime
|
from decimal import Decimal
|
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
|
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 conn:
|
with conn.cursor() as q:
|
q.execute('SELECT DATABASE() db')
|
assert q.fetchone()['db'] == 'mom-hm'
|
q.execute('SELECT * FROM crm_receivable_plan WHERE deleted=0 AND id IN (43,44,45,46,47,48,49,50) ORDER BY id' + (' FOR UPDATE' if apply else ''))
|
before=q.fetchall()
|
assert len(before)==8
|
changes=[]
|
for row in before:
|
q.execute('SELECT COALESCE(SUM(price),0) paid FROM crm_receivable WHERE deleted=0 AND audit_status=20 AND (plan_id=%s OR id=%s)', (row['id'],row['receivable_id']))
|
paid=Decimal(q.fetchone()['paid'])
|
remain=max(Decimal(row['price'])-paid,Decimal('0'))
|
due=row['return_time'].strftime('%Y-%m-%d') if row['return_time'] else None
|
date_text=f'计划回款日期{due}' if due else '计划日期待确认'
|
remark=f'尚未回款余额{remain:.2f}元；已回款{paid:.2f}元。{date_text}；实际到账情况待核对'
|
changes.append({'id':row['id'],'before':row['remark'],'after':remark})
|
print(json.dumps({'changes':changes},ensure_ascii=False))
|
if apply:
|
assert all(not (x['remark'] or '').strip() for x in before), '仅填充此前清空的8条备注，避免覆盖新内容'
|
out=Path(__file__).parent/'receivable-plan-remark-20261015'
|
out.mkdir(exist_ok=True)
|
stamp=datetime.now().strftime('%Y%m%d%H%M%S')
|
(out/f'fill-before-{stamp}.json').write_text(json.dumps(before,ensure_ascii=False,default=str,indent=2),encoding='utf-8')
|
for change in changes:
|
q.execute('UPDATE crm_receivable_plan SET remark=%s,update_time=update_time WHERE id=%s AND deleted=0', (change['after'],change['id']))
|
assert q.rowcount==1
|
q.execute('SELECT * FROM crm_receivable_plan WHERE deleted=0 AND id IN (43,44,45,46,47,48,49,50) ORDER BY id')
|
after=q.fetchall()
|
assert len(after)==len(before)
|
for a,b,change in zip(after,before,changes):
|
assert a['remark']==change['after']
|
assert {k:v for k,v in a.items() if k!='remark'}=={k:v for k,v in b.items() if k!='remark'}
|
conn.commit()
|
(out/f'fill-result-{stamp}.json').write_text(json.dumps(changes,ensure_ascii=False,indent=2),encoding='utf-8')
|
print(json.dumps({'updated':len(changes),'onlyRemarkChanged':True}))
|
