import json,collections,unicodedata
|
from pathlib import Path
|
import xml.etree.ElementTree as ET
|
from decimal import Decimal
|
P=Path('scripts/quotation-import')
|
plan=json.loads((P/'local-plan.json').read_text(encoding='utf-8'))
|
db=ET.parse(P/'remote-snapshot.xml')
|
def rows(tag):return [{e.tag:e.text or '' for e in x} for x in db.findall('./'+tag+'/row')]
|
norm=lambda v:unicodedata.normalize('NFKC',v).casefold().strip()
|
materials={x['code']:x for x in rows('materials')}
|
units={x['name'].strip():x for x in rows('units')}
|
customers=collections.defaultdict(list)
|
for x in rows('customers'):
|
if x['deleted']=='0':customers[norm(x['name'])].append(x)
|
owners=[x for x in rows('users') if x['username']=='zhuli']
|
assert len(owners)==1 and owners[0]['nickname'].strip()=='朱莉' and owners[0]['status']=='0'
|
owner=owners[0]
|
new={x['code']:x for x in plan['new_products']}
|
matched=set();created=set()
|
for q in plan['quotes']:
|
matches=customers[norm(q['customer'])]
|
assert len(matches)<=1,(q['customer'],matches)
|
if matches:
|
q['customer']=matches[0]['name'];matched.add(q['customer'])
|
assert matches[0]['owner_user_id']==owner['id'],('Existing customer owner differs',q['customer'])
|
else:created.add(q['customer'])
|
for x in q['items']:
|
if x['name']=='黑果枸杞' and x['specification']=='120克（4g/罐×30）':
|
candidate=materials['TC0020']
|
assert '黑果枸杞' in candidate['name'] and '120' in (candidate['name']+candidate['specification'])
|
x['item_code']='TC0020'
|
code=x['item_code'];unit=units[x['unit']]
|
x['item_unit_id']=int(unit['id'])
|
if code in materials:
|
m=materials[code];assert m['status']=='0'
|
x.update(item_id=int(m['id']),item_name=m['name'],expected_base_unit=int(m['unit_measure_id']))
|
else:
|
assert code in new,code
|
x.update(item_id=0,expected_base_unit=int(unit['id']))
|
new[code]['unit_id']=int(unit['id'])
|
needed={x['item_code'] for q in plan['quotes'] for x in q['items'] if not x['item_id']}
|
plan['new_products']=[x for code,x in new.items() if code in needed]
|
plan.update(target='remote',owner_remote_id=int(owner['id']),matched_customers=len(matched),new_customers=len(created))
|
assert len(plan['quotes'])==345 and sum(len(q['items']) for q in plan['quotes'])==1702
|
assert sum(Decimal(q['total']) for q in plan['quotes'])==Decimal('15392805.71')
|
(P/'remote-plan.json').write_text(json.dumps(plan,ensure_ascii=False,indent=2),encoding='utf-8')
|
print('Remote prepared: 345 quotes, 1702 lines;',len(matched),'existing customers,',len(created),'new customers,',len(needed),'new products; owner',owner['username'],owner['id'])
|
