60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
|
#!/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)
|