import sys,json
from pathlib import Path
from datetime import datetime
sys.stdout.reconfigure(encoding='utf-8')
ROOT=Path(__file__).resolve().parent
sys.path.insert(0,str(ROOT.parent/'iqc-import-20261012/deps'))
import pymysql
manifest=json.loads((ROOT/'backlog-inserted-ids.json').read_text(encoding='utf-8'))
ids=[r['follow_id'] for r in manifest['records']]
assert len(ids)==1773 and len(set(ids))==len(ids)
args=dict(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,connect_timeout=15,read_timeout=60)
conn=pymysql.connect(**args)
try:
    with conn.cursor() as q:
        q.execute('SELECT DATABASE() db,NOW() now');state=q.fetchone();assert state['db']=='mom-hm'
        clause=','.join(['%s']*len(ids))
        q.execute('SELECT * FROM crm_follow_up_record WHERE id IN ('+clause+') ORDER BY id FOR UPDATE',ids)
        before=q.fetchall();assert len(before)==len(ids)
        for r in before:
            assert r['creator']=='0' and r['type']==5
            assert all(r[field] in ('[]','') for field in ('business_ids','contact_ids','pic_urls'))
        backup=ROOT/('followup-list-repair-backup-'+state['now'].strftime('%Y%m%d-%H%M%S')+'.json')
        backup.write_text(json.dumps(before,ensure_ascii=False,indent=2,default=str),encoding='utf-8')
        q.execute("UPDATE crm_follow_up_record SET business_ids=CASE WHEN business_ids='[]' THEN '' ELSE business_ids END, contact_ids=CASE WHEN contact_ids='[]' THEN '' ELSE contact_ids END, pic_urls=CASE WHEN pic_urls='[]' THEN '' ELSE pic_urls END, update_time=update_time WHERE id IN ("+clause+") AND (business_ids='[]' OR contact_ids='[]' OR pic_urls='[]')",ids)
        changed=q.rowcount
        q.execute('SELECT * FROM crm_follow_up_record WHERE id IN ('+clause+') ORDER BY id',ids)
        after=q.fetchall()
        for old,current in zip(before,after):
            assert all(current[k]==v for k,v in old.items() if k not in ('business_ids','contact_ids','pic_urls'))
            assert all(current[field]=='' for field in ('business_ids','contact_ids','pic_urls'))
        q.execute('SELECT COUNT(*) count FROM system_operate_log WHERE trace_id=%s',(manifest['batch'],));assert q.fetchone()['count']==1773
        conn.commit()
    with pymysql.connect(**args) as verify:
        with verify.cursor() as q:
            q.execute('SELECT * FROM crm_follow_up_record WHERE id IN ('+clause+') ORDER BY id',ids)
            assert q.fetchall()==after
    result=dict(database='mom-hm',repaired_records=changed,records_verified=len(after),committed=True,verified_after_commit=True,backup=str(backup))
    (ROOT/'followup-list-repair-result.json').write_text(json.dumps(result,ensure_ascii=False,indent=2),encoding='utf-8')
    print(json.dumps(result,ensure_ascii=False))
except Exception:
    conn.rollback();raise
finally:conn.close()