#!/usr/bin/env python3
"""
验证Excel文件格式是否符合MAC与批次导入要求
"""
import sys
import pandas as pd
import warnings

warnings.filterwarnings('ignore', category=UserWarning, module='openpyxl')

def validate_excel(file_path):
    """
    验证Excel文件格式
    返回: (is_valid, error_message)
    """
    try:
        df = pd.read_excel(file_path)
        
        if df.empty:
            return False, "文件为空，没有数据"
        
        columns = df.columns.tolist()
        
        # 检查是否有批次号列
        if '批次号' not in columns:
            return False, "缺少必需的列：批次号"
        
        # 检查是否有MAC或SN_MAC列
        has_mac = 'MAC' in columns
        has_sn_mac = 'SN_MAC' in columns
        
        if not has_mac and not has_sn_mac:
            return False, "缺少必需的列：MAC 或 SN_MAC"
        
        # 检查列数（应该只有2列）
        if len(columns) != 2:
            return False, f"文件应该只包含2列数据，当前有{len(columns)}列：{', '.join(columns)}"
        
        # 验证通过
        mac_col = 'MAC' if has_mac else 'SN_MAC'
        return True, f"文件格式正确，包含列：{mac_col} 和 批次号，共{len(df)}行数据"
        
    except Exception as e:
        return False, f"读取文件失败：{str(e)}"

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("用法: python validate_excel.py <excel文件路径>")
        sys.exit(1)
    
    file_path = sys.argv[1]
    is_valid, message = validate_excel(file_path)
    
    if is_valid:
        print(f"✓ {message}")
        sys.exit(0)
    else:
        print(f"✗ {message}")
        sys.exit(1)