1 import os
  2 import glob
  3 import xml.etree.ElementTree as ET
  4 import argparse
  5 
  6 def indent(elem, level=0):
  7     """ Helper function to indent the XML for pretty printing. """
  8     i = "\n" + level * "    "
  9     if len(elem):
 10         if not elem.text or not elem.text.strip():
 11             elem.text = i + "    "
 12         if not elem.tail or not elem.tail.strip():
 13             elem.tail = i
 14         for elem in elem:
 15             indent(elem, level + 1)
 16         if not elem.tail or not elem.tail.strip():
 17             elem.tail = i
 18     else:
 19         if level and (not elem.tail or not elem.tail.strip()):
 20             elem.tail = i
 21         if not elem.tail:
 22             elem.tail = "\n"
 23 
 24 def add_files_to_group(uvprojx_file_path, mode,folder_path, group_name_target):
 25     # 改变文件扩展名从 .uvprojx 到 .xml
 26     base, ext = os.path.splitext(uvprojx_file_path)
 27     if ext != '.uvprojx':
 28         print("工程文件扩展名不正确")
 29         return
 30     
 31     xml_path = base + '.xml'
 32     os.rename(uvprojx_file_path, xml_path)
 33 
 34     try:
 35         #解析XML文件
 36         tree = ET.parse(xml_path)
 37         #获取根节点
 38         root = tree.getroot()
 39 
 40         if mode == 0 or mode == 1:
 41             # 找到指定GroupName的Group节点
 42             target_group = None
 43             for group in root.findall('.//Group'):
 44                 group_name = group.find('GroupName')
 45                 if group_name is not None and group_name.text == group_name_target:
 46                     target_group = group
 47                     break
 48 
 49             if target_group is None:
 50                 print(f"未发现 '{group_name_target}' 分组，请先创建分组后再尝试")
 51                 # 将文件扩展名改回 .uvprojx
 52                 os.rename(xml_path, uvprojx_file_path)
 53                 return
 54 
 55             # 找到目标 Group 节点下的 Files 节点，如果不存在则创建一个
 56             files_node = target_group.find('Files')
 57             if files_node is None:
 58                 files_node = ET.SubElement(target_group, 'Files')
 59                                 
 60         #寻找头文件分组
 61         if mode == 0 or mode == 2:
 62             print("寻找头文件分组......")
 63             target_header = None
 64             heard_inc = None
 65             target_header = root.find('.//Cads')
 66             if target_header == None:
 67                 print("未发现头文件分组Cads")
 68                 return
 69             else:
 70                 heard_inc = target_header.find('VariousControls')
 71                 if heard_inc == None:
 72                     print("未发现头文件分组VariousControls")
 73                     return
 74                 else:
 75                     heard_inc = heard_inc.find('IncludePath')
 76                     if heard_inc == None:
 77                         print("未发现头文件分组IncludePath")
 78                         return
 79                     else:
 80                         print("找到头文件分组")
 81                        
 82         
 83         
 84         #下面没有节点
 85         if mode == 0 or mode == 1:
 86             creat_dot = 0 #是否需要创建节点标志,如果有重复则跳过 
 87             init_creat = 0
 88             file_init = files_node.find('File')
 89             if file_init == None:
 90                 creat_dot = 1
 91                 init_creat = 1
 92                 print("初始节点为空需要创建节点")
 93   
 94         # 遍历指定文件夹，查找所有 .c 文件  
 95         if mode == 0 or mode == 2:
 96             #print(heard_inc.text)  
 97             heard_data = heard_inc.text   
 98             
 99         for subdir, _, files in os.walk(folder_path):
100             #.h路径
101             if mode == 0 or mode == 2:
102                 dir_path = os.path.relpath(subdir, start=os.path.dirname(xml_path))
103                 if dir_path in heard_inc.text:
104                     print("需要添加的头文件路径已存在本次跳过")
105                 else:
106                     heard_data = heard_data + dir_path + ";"
107                     
108             #.c添加到分组
109             if mode == 0 or mode == 1:
110                 for file in files:     
111                     if file.endswith('.c'):
112                         # 计算相对路径
113                         file_path = os.path.relpath(os.path.join(subdir, file), start=os.path.dirname(xml_path))
114                         #print("路径",file_path)
115                         file_check = files_node.findall('File')
116                         if init_creat == 0:
117                             #print("长度",len(file_check))
118                             #遍历当前分组下的节点,检测是否已经包含了该路径,如果有直接跳过
119                             for i in range(len(file_check)):
120                                 if file_path in file_check[i].find("FilePath").text:
121                                     print("节点已存在本次跳过")
122                                     creat_dot = 0
123                                     break
124                                 else: 
125                                     if i == len(file_check) - 1:
126                                         creat_dot = 1
127                                         print("节点不存在,创建节点")
128                                     else:
129                                        creat_dot = 0
130                                     continue    
131                         if creat_dot == 1:
132                             # 创建 File 节点并添加到 Files 节点下
133                             file_node = ET.SubElement(files_node, 'File')
134                             file_name_node = ET.SubElement(file_node, 'FileName')
135                             file_name_node.text = file
136                             file_type_node = ET.SubElement(file_node, 'FileType')
137                             file_type_node.text = '1'  # .c 文件类型都为 1
138 
139                             file_path_node = ET.SubElement(file_node, 'FilePath')
140                             file_path_node.text = file_path
141                             creat_dot = 0
142                             init_creat = 0
143                             
144         if mode == 0 or mode == 2:
145             heard_data = heard_data.rstrip(";") #移除最后一个多加的;
146             heard_inc.text = heard_data
147             #print(heard_inc.text)                    
148                   
149         # 格式化 XML
150         indent(root)
151 
152         # 保存修改后的 XML 文件
153         tree.write(xml_path, encoding='utf-8', xml_declaration=True)
154         print("已完成")
155 
156     except ET.ParseError as e:
157         print(f"ParseError: {e}")
158         with open(xml_path, 'r', encoding='utf-8') as file:
159             lines = file.readlines()
160             start = max(0, e.position[0] - 5)
161             end = min(len(lines), e.position[0] + 5)
162             print("Context around the error:")
163             for i in range(start, end):
164                 print(f"{i+1}: {lines[i].strip()}")
165 
166     finally:
167         # 将文件扩展名改回 .uvprojx
168         os.rename(xml_path, uvprojx_file_path)
169 
170 #寻找工程文件
171 def find_uvprojx_file():
172     uvprojx_files = glob.glob("*.uvprojx")
173     if not uvprojx_files:
174         print("未找到工程文件,请把此文件放在keil工程目录下")
175         return None
176     elif len(uvprojx_files) > 1:
177         print("在当前目录中找到多个.uvprojx文件：")
178         for i, file in enumerate(uvprojx_files, start=1):
179             print(f"{i}. {file}")
180         print("请确保目录中只有一个.uvprojx文件")
181         return None
182     else:
183         return uvprojx_files[0]
184 
185 if __name__ == "__main__":
186     print("keil一键添加文件和头文件路径脚本\n\
187     需放在keil工程同级目录下\n\
188     参数格式,参数用空格隔开\n\
189     默认模式:0\n\
190     默认路径:\"../../../external/lvgl\"\n\
191     默认分组:\"lvgl\"\n\
192     1.添加模式 0:全部添加(.c文件全添加到分组.h文件夹加入include路径里) 1:只添加.c文件到分组 2:只添加.h文件夹到include里\n\
193     2.要添加的文件夹路径,请使用相对路径\n\
194     3.要添加的分组名称,如果没有分组需要先去keil手动添加分组\n")
195     
196     param = input("请输入参数:")
197     
198     if param:
199         #print(param)
200         args = param.split()
201         args[0] = int(args[0])
202         print(args)
203     else:
204         args = [0,"../../../external/lvgl","lvgl"]
205         print("使用默认参数:",args)
206     uvprojx_file_path = find_uvprojx_file()
207     if uvprojx_file_path:
208         add_files_to_group(uvprojx_file_path, args[0],args[1],args[2])