#!/usr/bin/env python3
"""
用本地GPS数据(G7日里程 + 东方北斗月里程)替换125辆测试车辆的实际行驶里程。
测试里程 = TBOX - GPS（负数为0）。
"""
import openpyxl, os, shutil, xlrd
from datetime import datetime, date, timedelta
from openpyxl.styles import PatternFill, Font, Alignment, Border, Side
from openpyxl.utils import get_column_letter

os.chdir('/Users/kkfluous/Downloads')

SRC = '租赁任务考核_2026年6月 (1).xlsx'
DST = '租赁任务考核_2026年6月.xlsx'
GPS_DIR = 'gps里程汇总2026年1-6月 3'

# ========== 读取125辆测试车辆清单 ==========
TEST_FILE = '/Users/kkfluous/Library/Containers/com.tencent.xinWeChat/Data/Documents/xwechat_files/wxid_1704407055112_3af8/temp/RWTemp/2026-07/65b41c10aa83cabb7a365b2b8437c1c7/6月测试车辆.xlsx'
wb_test = openpyxl.load_workbook(TEST_FILE, data_only=True)
ws_test = wb_test.active
TEST_PLATES = set()
for row in ws_test.iter_rows(min_row=2, values_only=True):
    for ci in range(len(row)):
        p = str(row[ci]).strip() if row[ci] else ''
        if p and p != 'None' and len(p) >= 7:
            TEST_PLATES.add(p)
wb_test.close()
print(f"测试车辆清单: {len(TEST_PLATES)}辆")

# ========== Step 1: 加载G7 GPS日里程 ==========
print("\n加载G7 GPS日里程...")
g7_wb = openpyxl.load_workbook(f'{GPS_DIR}/G7/202606_G7.xlsx')
g7_ws = g7_wb.active
gps_daily = {}   # plate -> [day1, day2, ..., day30]
for row in g7_ws.iter_rows(min_row=2, values_only=True):
    plate = str(row[0]).strip() if row[0] else ''
    if not plate or plate not in TEST_PLATES:
        continue
    daily = [float(row[c]) if row[c] else 0 for c in range(3, 33)]
    gps_daily[plate] = daily
g7_wb.close()
print(f"G7覆盖: {len(gps_daily)}辆测试车")

# ========== Step 2: 加载东方北斗GPS月里程 ==========
print("加载东方北斗GPS月里程...")
df_wb = openpyxl.load_workbook(f'{GPS_DIR}/东方北斗/202606_东方北斗.xlsx')
df_ws = df_wb.active
gps_monthly = {}  # plate -> monthly_total_km
for row in df_ws.iter_rows(min_row=4, values_only=True):
    plate = str(row[5]).strip() if row[5] else ''
    if not plate or plate not in TEST_PLATES:
        continue
    monthly = float(row[10]) if row[10] else 0
    gps_monthly[plate] = monthly
df_wb.close()
print(f"东方北斗覆盖: {len(gps_monthly)}辆测试车")

# ========== Step 3: 读取源文件，执行GPS替换 ==========
print("\n执行GPS替换...")

# 备份
BAK = '租赁任务考核_2026年6月_备份_0715.xlsx'
if os.path.exists(DST):
    shutil.copy2(DST, BAK)

shutil.copy2(SRC, DST)

wb = openpyxl.load_workbook(DST)
ws = wb['业务考核视图']
h = [c.value for c in next(ws.iter_rows(min_row=1, max_row=1))]
print(f"源文件: {ws.max_row}行, {len(h)}列")

# 读取所有记录
recs = []
for row in ws.iter_rows(min_row=2, values_only=True):
    r = dict(zip(h, row))
    r['row_num'] = len(recs) + 2
    recs.append(r)

test_recs = [r for r in recs if r['车牌号'] in TEST_PLATES]
print(f"测试车辆记录: {len(test_recs)}条 ({len(set(r['车牌号'] for r in test_recs))}辆)")

# 添加列
if '原TBOX里程(km)' not in h:
    tbox_new_col = ws.max_column + 1
    test_new_col = ws.max_column + 2
    ws.cell(row=1, column=tbox_new_col, value='原TBOX里程(km)')
    ws.cell(row=1, column=test_new_col, value='测试里程(km)')
    h_new = [c.value for c in next(ws.iter_rows(min_row=1, max_row=1))]
else:
    h_new = h

actual_idx = h_new.index('实际行驶里程(km)') + 1
tbox_idx = h_new.index('原TBOX里程(km)') + 1
test_idx = h_new.index('测试里程(km)') + 1
rate_idx = h_new.index('完成率(%)') + 1
qualified_idx = h_new.index('是否达标') + 1

yellow_fill = PatternFill(start_color='FFFF00', end_color='FFFF00', fill_type='solid')

replaced_g7 = 0
replaced_df = 0
no_gps = 0
total_test_km = 0
total_gps_km = 0
total_tbox_km = 0

for r in test_recs:
    plate = r['车牌号']
    rn = r['row_num']

    # 获取考核区间
    start_date = r.get('考核开始日期')
    end_date = r.get('考核结束日期')

    gps_total = None
    source = None

    # 优先用G7日数据
    if plate in gps_daily:
        daily = gps_daily[plate]
        try:
            sd = start_date.date() if hasattr(start_date, 'date') else datetime.strptime(str(start_date)[:10], '%Y-%m-%d').date()
            ed = end_date.date() if hasattr(end_date, 'date') else datetime.strptime(str(end_date)[:10], '%Y-%m-%d').date()
        except:
            sd = date(2026, 6, 1)
            ed = date(2026, 6, 30)

        gps_total = 0
        for day_offset in range(30):
            day = date(2026, 6, 1) + timedelta(days=day_offset)
            if sd <= day <= ed:
                gps_total += daily[day_offset]
        gps_total = round(gps_total, 1)
        source = 'G7'
        replaced_g7 += 1

    # 其次用东方北斗月里程
    elif plate in gps_monthly:
        # 东方北斗只有月度合计，考核区间一般都是整月
        gps_total = gps_monthly[plate]
        source = '东方北斗'
        replaced_df += 1

    if gps_total is None:
        no_gps += 1
        continue

    # 原TBOX里程 = 当前实际行驶里程
    tbox_val = float(ws.cell(row=rn, column=actual_idx).value or 0)

    # 测试里程 = TBOX - GPS（负数为0）
    test_km = max(0, round(tbox_val - gps_total, 1))

    # 写入
    ws.cell(row=rn, column=tbox_idx, value=tbox_val)
    ws.cell(row=rn, column=test_idx, value=test_km)
    ws.cell(row=rn, column=actual_idx, value=gps_total)

    # 标记黄色
    for c in range(1, ws.max_column + 1):
        ws.cell(row=rn, column=c).fill = yellow_fill

    # 重算完成率
    target = float(r.get('应考核里程(km)') or 0)
    pct = round(gps_total / target * 100, 2) if target > 0 else 0
    qualified = '达标' if gps_total >= target and target > 0 else '未达标'
    ws.cell(row=rn, column=rate_idx, value=pct)
    ws.cell(row=rn, column=qualified_idx, value=qualified)

    total_test_km += test_km
    total_gps_km += gps_total
    total_tbox_km += tbox_val

print(f"\n替换统计:")
print(f"  G7日数据: {replaced_g7}条")
print(f"  东方北斗月数据: {replaced_df}条")
print(f"  无GPS数据: {no_gps}条")
print(f"  TBOX总里程: {total_tbox_km:.1f}")
print(f"  GPS总里程: {total_gps_km:.1f}")
print(f"  测试总里程: {total_test_km:.1f}")

# ========== Step 4: TBOX回填（非测试车辆） ==========
print("\nTBOX回填（非测试车辆）...")
filled = 0
for row_idx in range(2, ws.max_row + 1):
    tbox_val = ws.cell(row=row_idx, column=tbox_idx).value
    actual_val = ws.cell(row=row_idx, column=actual_idx).value
    test_val = ws.cell(row=row_idx, column=test_idx).value
    if (tbox_val is None or float(tbox_val) == 0):
        if test_val is None or float(test_val) == 0:
            ws.cell(row=row_idx, column=tbox_idx, value=float(actual_val) if actual_val else 0)
            filled += 1
print(f"TBOX回填: {filled}条")

# ========== Step 5: 列序排版 ==========
print("\n列序排版...")
target_order = [
    '车牌号', '考核年份', '考核月份', '部门名称', '销售经理', '客户名称',
    '合同编号', '项目名称', '考核目标ID', '考核目标', '考核开始日期', '考核结束日期',
    '考核天数', '应考核里程(km)', '原TBOX里程(km)', '实际行驶里程(km)', '测试里程(km)',
    '完成率(%)', '是否达标', '考核状态', '交车日期', '预计还车日期', '实际还车日期'
]

h_cur = [c.value for c in next(ws.iter_rows(min_row=1, max_row=1))]
all_rows = []
for row_idx in range(1, ws.max_row + 1):
    row_data = {}
    for c in range(1, ws.max_column + 1):
        row_data[h_cur[c-1]] = ws.cell(row=row_idx, column=c).value
    all_rows.append(row_data)

# Clear
for r in range(1, ws.max_row + 1):
    for c in range(1, ws.max_column + 1):
        ws.cell(row=r, column=c, value=None)

thin_border = Border(left=Side(style='thin'), right=Side(style='thin'),
                     top=Side(style='thin'), bottom=Side(style='thin'))
header_fill = PatternFill(start_color='4472C4', end_color='4472C4', fill_type='solid')
header_font = Font(name='宋体', size=10, bold=True, color='FFFFFF')
data_font = Font(name='宋体', size=10)

for i, col_name in enumerate(target_order):
    c = i + 1
    cell = ws.cell(row=1, column=c, value=col_name)
    cell.font = header_font
    cell.fill = header_fill
    cell.alignment = Alignment(horizontal='center', vertical='center')
    cell.border = thin_border

for row_idx, row_data in enumerate(all_rows[1:], start=2):
    for i, col_name in enumerate(target_order):
        c = i + 1
        val = row_data.get(col_name)
        cell = ws.cell(row=row_idx, column=c, value=val)
        cell.font = data_font
        cell.alignment = Alignment(vertical='center')
        cell.border = thin_border
        if col_name in ['考核开始日期', '考核结束日期', '交车日期', '预计还车日期', '实际还车日期']:
            cell.number_format = 'yyyy-mm-dd h:mm:ss'
        if row_data.get('测试里程(km)') and float(row_data.get('测试里程(km)', 0) or 0) != 0:
            cell.fill = yellow_fill

for i in range(len(target_order)):
    ws.column_dimensions[get_column_letter(i+1)].width = 15

wb.save(DST)
wb.close()

print(f"列序排版: {len(target_order)}列")
print(f"\n{'='*60}")
print(f"GPS替换完成！")
print(f"测试车辆: {len(TEST_PLATES)}辆 -> 源数据{len(test_recs)}条记录")
print(f"GPS来源: G7({replaced_g7}条) + 东方北斗({replaced_df}条)")
print(f"测试总里程: {total_test_km:.1f} km")
print(f"已保存: {DST}")
print(f"{'='*60}")