# -*- 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)
    sglob = glob.glob(os.path.join(docs, sample_name))
    if not sglob:
        print(gen_name, '-> sample not found'); continue
    sample_path = sglob[0]
    wg = openpyxl.load_workbook(gen_path)
    ws = wg[wg.sheetnames[0]]
    wsg = openpyxl.load_workbook(sample_path)
    wss = wsg[wsg.sheetnames[0]]
    print('='*70)
    print(f'{gen_name}: dims {ws.max_row}x{ws.max_column} | sample {sample_name}: {wss.max_row}x{wss.max_column}')
    print(' gen R1:', [str(ws.cell(1,c).value)[:14] for c in range(1, min(ws.max_column,6)+1)])
    print(' smp R1:', [str(wss.cell(1,c).value)[:14] for c in range(1, min(wss.max_column,6)+1)])
    print(' gen R2:', [str(ws.cell(2,c).value)[:14] for c in range(1, min(ws.max_column,6)+1)])
    print(' smp R2:', [str(wss.cell(2,c).value)[:14] for c in range(1, min(wss.max_column,6)+1)])
    print(' gen last row:', [str(ws.cell(ws.max_row,c).value)[:14] for c in range(1, min(ws.max_column,5)+1)])