# -*- coding: utf-8 -*-
import openpyxl, glob, os

tmp = r'C:\Users\jcxiong\AppData\Local\Temp\ta_e2e_524cecf1'
docs = r'C:\Users\jcxiong\Documents\Codex\MyProject\trafficAudit\docs'

pairs = [
    ('gen_freight_turnover.xlsx', '生成_货运量周转量.xlsx'),
    ('gen_city_detail.xlsx', '生成_货运量分市州明细.xlsx'),
    ('gen_freight_rank.xlsx', '生成_货运量排名.xlsx'),
    ('gen_turnover_rank.xlsx', '生成_周转量排名.xlsx'),
]
for gen_name, sample_name in pairs:
    gen_path = os.path.join(tmp, gen_name)
    sample_path = glob.glob(os.path.join(docs, sample_name))[0]
    wg = openpyxl.load_workbook(gen_path)
    ws = wg[wg.sheetnames[0]]
    wb2 = openpyxl.load_workbook(sample_path)
    wss = wb2[wb2.sheetnames[0]]
    match = 'OK' if (ws.max_row, ws.max_column) == (wss.max_row, wss.max_column) else 'DIFF'
    print(f'{gen_name}: {ws.max_row}x{ws.max_column} vs {wss.max_row}x{wss.max_column} -> {match}')
    print('   gen title:', str(ws.cell(1,1).value)[:40])
# detail rank row layout
ws = openpyxl.load_workbook(os.path.join(tmp, 'gen_freight_rank.xlsx'))[openpyxl.load_workbook(os.path.join(tmp, 'gen_freight_rank.xlsx')).sheetnames[0]]
for r in [1,2,3,4,5,6,22,23,24,25,26,27,28,44,45,46,47]:
    a = ws.cell(r,1).value
    b = ws.cell(r,2).value
    print('  R%d:' % r, str(a)[:30] if a is not None else '', '|', str(b)[:16] if b is not None else '')