import json,collections,xml.etree.ElementTree as ET,sys
|
from pathlib import Path
|
from decimal import Decimal
|
P=Path('scripts/quotation-import')
|
target=sys.argv[1] if len(sys.argv)>1 else 'local'
|
assert target in ['local','remote']
|
plan=json.loads((P/f'{target}-plan.json').read_text(encoding='utf-8'))
|
source=json.loads((P/'source.json').read_text(encoding='utf-8'))
|
r=ET.parse(P/f'{target}-post-snapshot.xml')
|
def records(tag):return [{e.tag:e.text or '' for e in x} for x in r.findall('./'+tag+'/row')]
|
customers={x['id']:x for x in records('customers')}
|
materials={x['id']:x for x in records('materials')}
|
units={x['id']:x for x in records('units')}
|
users={x['username']:x for x in records('users')}
|
owner=users['zhuli'];assert owner['nickname'].strip()=='朱莉' and owner['status']=='0'
|
headers={x['no']:x for x in records('quotations') if x['deleted']=='0'}
|
lines=collections.defaultdict(list)
|
for x in records('lines'):
|
if x['deleted']=='0':lines[x['quotation_id']].append(x)
|
permissions={(x['biz_type'],x['biz_id'],x['user_id'],x['level']) for x in records('permissions') if x['deleted']=='0'}
|
assert len(headers)==345 and sum(len(v) for v in lines.values())==1702
|
assert len({h['customer_id'] for h in headers.values()})==344
|
original={q['no']:q for q in source['quotes']}
|
total=Decimal(0)
|
for q in plan['quotes']:
|
h=headers[q['no']];s=original[q['no']]
|
assert h['owner_user_id']==owner['id'] and h['status']=='0'
|
assert customers[h['customer_id']]['name']==q['customer']
|
assert customers[h['customer_id']]['owner_user_id']==owner['id']
|
assert h['quotation_time']==q['quotation_time'] and h['valid_until']==q['valid_until']
|
assert Decimal(h['total_product_price'])==Decimal(q['total'])==Decimal(h['total_price'])
|
assert h['tax_rate']==h['tax_price']==''
|
assert Decimal(h['discount_percent'])==Decimal(h['discount_price'])==0
|
assert h['business_id']==h['contact_id']==''
|
assert h['remark']==q['remark']
|
assert ('9',h['id'],owner['id'],'1') in permissions
|
assert ('2',h['customer_id'],owner['id'],'1') in permissions
|
items={materials[x['item_id']]['code']:x for x in lines[h['id']]}
|
assert len(items)==len(q['items'])==len(s['items'])
|
for x,sx in zip(q['items'],s['items']):
|
d=items[x['item_code']]
|
assert d['item_unit_id']==str(x['item_unit_id'])
|
assert units[d['item_unit_id']]['name'].strip()==sx['unit']
|
assert d['remark']==x['remark'] and d['tax_percent']==d['tax_price']==''
|
for stored,field in [('item_price','original_price'),('quotation_price','price'),('count','count'),('total_price','total')]:
|
assert Decimal(d[stored])==Decimal(sx[field]),(q['no'],x['line'],stored)
|
total+=Decimal(h['total_price'])
|
assert total==Decimal('15392805.71')
|
for product in plan['new_products']:
|
matches=[m for m in materials.values() if m['code']==product['code']]
|
assert len(matches)==1 and matches[0]['specification']==product['specification'] and matches[0]['item_type']=='2'
|
assert source['excluded_sheets']==['报价单汇总总表','编制说明']
|
result=f'PASS {target}: 345 quotations, 1702 lines, 344 customers, total {total}; exact source prices/quantities/amounts, dates, units, owner zhuli, draft status and scoped ownership permissions verified. Summary excluded.\n'
|
(P/f'{target}-verification-result.txt').write_text(result,encoding='utf-8');print(result)
|
