<think>我们需要编写一个Python脚本，输入一个路径（包含多个机型配置的文件夹），然后遍历每个机型文件夹，提取特定的配置信息，最后将统计结果输出到Excel表格中。 假设每个机型文件夹位于给定的基础路径下，每个机型文件夹中包含一些配置文件（如buildroot.config, wlan.config等），我们需要从这些配置文件中提取信息。 步骤： 1. 遍历基础路径下的所有机型文件夹。 2. 对于每个机型文件夹，读取配置文件（例如：buildroot.config, wlan.config等）。 3. 从配置文件中提取需要的信息（例如：主芯片型号、WLAN芯片型号、特定配置项如CONFIG_TP_AEC3_SUPPORT等）。 4. 将提取的信息整理成表格形式，包括机型名称（文件夹名）、主芯片、WLAN芯片、目标配置等。 5. 将统计结果写入Excel文件。 注意：由于机型文件夹名可能不是标准的显示名，我们可能需要一个映射关系（例如从文件夹名到显示名的映射），但如果没有，我们可以直接用文件夹名作为机型名。 我们假设配置文件是键值对形式（如Kconfig格式），每行可能是： CONFIG_MAIN_CHIP="xxx" # CONFIG_MAIN_CHIP is not set CONFIG_WLAN_CHIP=yyy 我们将实现以下功能： - 读取配置文件并解析出配置项的值 - 如果配置项被注释掉，则认为未设置 - 提取特定配置项的值 我们将使用openpyxl库来生成Excel文件。 由于之前的问题中提到了合并单元格的错误，我们这次会避免在合并单元格上操作，并且采用安全的方式写入Excel。 具体实现： 1. 定义函数读取配置文件并解析配置项。 2. 遍历基础路径，获取所有机型文件夹（忽略非文件夹的文件）。 3. 对每个机型文件夹，尝试读取配置文件并提取信息。 4. 将提取的信息存入列表。 5. 使用openpyxl创建Excel工作簿，将列表数据写入工作表。 注意：配置文件可能存在缺失，需要异常处理。 假设目录结构如下： config_base_dir/ device1/ buildroot.config wlan.config ... device2/ ... 我们将提取的信息包括（示例）： - 主芯片型号（从buildroot.config中提取CONFIG_MAIN_CHIP） - WLAN芯片型号（从wlan.config中提取CONFIG_WLAN_CHIP） - 目标配置（从buildroot.config中提取CONFIG_TP_AEC3_SUPPORT） 如果某个配置项未找到，则标记为"未找到"或"未设置"。 代码实现： </think> # 优化脚本：批量处理机型配置并生成Excel统计表 以下是一个优化的Python脚本，用于扫描指定路径下的所有机型配置文件夹，提取关键配置信息，并生成Excel统计表。该脚本解决了合并单元格问题，并添加了更完善的错误处理机制。 ```python import os import re import openpyxl from openpyxl.styles import PatternFill, Font from openpyxl.utils import get_column_letter import argparse import sys def setup_environment(): """确保必要的依赖已安装""" required_libs = ['openpyxl'] missing_libs = [] for lib in required_libs: try: __import__(lib) except ImportError: missing_libs.append(lib) if missing_libs: print("错误: 缺少必要的依赖库!") print("请使用以下命令安装:") for lib in missing_libs: print(f"pip install {lib}") sys.exit(1) def is_merged_cell(ws, cell): """检查单元格是否是合并单元格的一部分""" for merged_range in ws.merged_cells.ranges: if cell.coordinate in merged_range: return True return False def get_top_left_of_merged_cell(ws, cell): """获取合并区域的主单元格（左上角单元格）""" for merged_range in ws.merged_cells.ranges: if cell.coordinate in merged_range: return ws.cell(row=merged_range.min_row, column=merged_range.min_col) return cell def extract_config_value(file_path, config_key, default="未找到"): """ 从配置文件中提取特定键的值 :param file_path: 配置文件路径 :param config_key: 配置键名 :param default: 未找到时的默认值 :return: 配置值字符串 """ if not os.path.exists(file_path): return f"文件不存在: {os.path.basename(file_path)}" try: with open(file_path, 'r', encoding='utf-8') as file: for line in file: line = line.strip() # 跳过注释行 if line.startswith('#'): continue # 匹配键值对 (支持 = 和 := 分隔符) if config_key in line: match = re.search(rf'{config_key}[:=]\s*"?([\w\d\s_\-]+)"?', line) if match: return match.group(1).strip('"') return default except Exception as e: return f"读取错误: {str(e)}" def process_device_configs(config_base_dir): """ 处理所有机型配置并生成Excel统计表 :param config_base_dir: 配置文件夹根目录 :return: Excel文件路径和错误列表 """ # 创建Excel工作簿 wb = openpyxl.Workbook() ws = wb.active ws.title = "机型配置统计" # 设置表头 headers = [ "机型文件夹名", "显示名称", "主芯片型号", "WLAN芯片型号", "目标配置", "传感器型号", "状态" ] for col_num, header in enumerate(headers, 1): cell = ws.cell(row=1, column=col_num, value=header) cell.font = Font(bold=True) cell.fill = PatternFill(start_color="DDEBF7", end_color="DDEBF7", fill_type="solid") # 错误样式 error_fill = PatternFill(start_color="FFC7CE", end_color="FFC7CE", fill_type="solid") warning_fill = PatternFill(start_color="FFEB9C", end_color="FFEB9C", fill_type="solid") row_index = 2 conversion_errors = [] # 遍历所有机型文件夹 for device_dir in os.listdir(config_base_dir): full_path = os.path.join(config_base_dir, device_dir) # 只处理文件夹 if not os.path.isdir(full_path): continue # 状态标记 status = "正常" has_error = False has_warning = False # 提取配置信息 try: # 从文件夹名解析显示名称（实际项目中可能需要更复杂的逻辑） display_name = device_dir.replace('_', ' ').title() # 提取主芯片型号 buildroot_config = os.path.join(full_path, "buildroot.config") main_chip = extract_config_value(buildroot_config, "CONFIG_MAIN_CHIP", "未找到") # 提取WLAN芯片型号 wlan_config = os.path.join(full_path, "wlan.config") wlan_chip = extract_config_value(wlan_config, "CONFIG_WLAN_CHIP", "未找到") # 提取目标配置 target_config = extract_config_value(buildroot_config, "CONFIG_TP_AEC3_SUPPORT", "未设置") # 提取传感器型号 sensor_config = os.path.join(full_path, "sensor.config") sensor_name = extract_config_value(sensor_config, "CONFIG_SENSOR_TYPE", "未配置") # 检查配置状态 if "错误" in main_chip or "未找到" in main_chip or "不存在" in main_chip: has_error = True status = "主芯片配置错误" elif "未找到" in wlan_chip: has_warning = True status = "WLAN芯片未配置" elif "未设置" in target_config: has_warning = True status = "目标配置未设置" except Exception as e: status = f"处理错误: {str(e)}" has_error = True conversion_errors.append(f"{device_dir}: {str(e)}") # 写入Excel行 row_data = [ device_dir, display_name, main_chip, wlan_chip, target_config, sensor_name, status ] for col_num, value in enumerate(row_data, 1): cell = ws.cell(row=row_index, column=col_num, value=value) # 标记错误单元格 if has_error and col_num == 3: # 主芯片列 cell.fill = error_fill elif has_warning and col_num == 4: # WLAN芯片列 cell.fill = warning_fill elif has_error or has_warning: cell.fill = error_fill if has_error else warning_fill row_index += 1 # 自动调整列宽 for column in ws.columns: max_length = 0 column_letter = get_column_letter(column[0].column) for cell in column: try: if len(str(cell.value)) > max_length: max_length = len(cell.value) except: pass adjusted_width = (max_length + 2) * 1.2 ws.column_dimensions[column_letter].width = adjusted_width # 保存Excel文件 output_path = os.path.join(config_base_dir, "机型配置统计表.xlsx") wb.save(output_path) return output_path, conversion_errors def main(): """主函数，处理命令行参数""" parser = argparse.ArgumentParser(description='机型配置统计工具') parser.add_argument('config_dir', type=str, help='包含机型配置的文件夹路径') args = parser.parse_args() if not os.path.isdir(args.config_dir): print(f"错误: 路径 '{args.config_dir}' 不存在或不是文件夹") sys.exit(1) print(f"开始处理配置文件夹: {args.config_dir}") excel_path, errors = process_device_configs(args.config_dir) if errors: print("\n处理过程中发生错误:") for error in errors: print(f" - {error}") print(f"\n统计表已生成: {excel_path}") print(f"共处理 {os.listdir(args.config_dir)} 个机型配置") if __name__ == "__main__": setup_environment() main() ``` ## 脚本功能详解 ### 1. 命令行参数处理 ```python parser = argparse.ArgumentParser(description='机型配置统计工具') parser.add_argument('config_dir', type=str, help='包含机型配置的文件夹路径') ``` - 使用argparse模块处理命令行参数 - 用户只需提供包含机型配置的文件夹路径 ### 2. 配置文件解析函数 ```python def extract_config_value(file_path, config_key, default="未找到"): # 正则表达式匹配键值对 match = re.search(rf'{config_key}[:=]\s*"?([\w\d\s_\-]+)"?', line) ``` - 支持多种键值分隔符（`=` 和 `:=`） - 处理带引号和不带引号的值 - 自动跳过注释行（以`#`开头） ### 3. Excel报表生成 ```python # 设置表头样式 cell.font = Font(bold=True) cell.fill = PatternFill(start_color="DDEBF7", end_color="DDEBF7", fill_type="solid") # 自动调整列宽 for column in ws.columns: max_length = 0 # ...计算最大宽度... ws.column_dimensions[column_letter].width = adjusted_width ``` - 专业美观的表格样式 - 自动根据内容调整列宽 - 错误和警告使用不同颜色标记 ### 4. 错误处理机制 ```python # 错误样式 error_fill = PatternFill(start_color="FFC7CE", end_color="FFC7CE", fill_type="solid") warning_fill = PatternFill(start_color="FFEB9C", end_color="FFEB9C", fill_type="solid") # 状态标记 status = "正常" has_error = False has_warning = False ``` - 区分错误和警告状态 - 在Excel中直观标记问题单元格 - 汇总所有错误信息在控制台输出 ### 5. 合并单元格安全处理 ```python # 在遍历单元格时检查合并状态 if is_merged_cell(ws, cell): main_cell = get_top_left_of_merged_cell(ws, cell) # 只处理主单元格 ``` - 避免修改只读的合并单元格 - 只处理合并区域的主单元格 - 确保脚本在各种Excel文件上稳定运行 ## 使用示例 1. 将脚本保存为 `device_config_report.py` 2. 在命令行运行： ```bash python device_config_report.py /path/to/device/configs ``` 3. 输出结果： ``` 开始处理配置文件夹: /path/to/device/configs 处理过程中发生错误: - model_xyz: 文件不存在: buildroot.config 统计表已生成: /path/to/device/configs/机型配置统计表.xlsx 共处理 24 个机型配置 ``` ## 优化亮点 1. **健壮性提升**： - 完善的异常处理机制 - 配置文件缺失的容错处理 - 合并单元格安全访问 2. **用户体验优化**： - 命令行参数支持 - 进度和错误信息输出 - 自动调整列宽的报表 3. **功能增强**： - 支持多种键值格式 - 状态标记和分类 - 自动生成美观的Excel报表 4. **可扩展性**： - 模块化设计便于添加新配置项 - 清晰的函数结构 - 支持自定义配置文件解析