154 lines
6.6 KiB
Python
154 lines
6.6 KiB
Python
############################################################################################################
|
||
# Author: LHYe200
|
||
# Date: 2025-02-03
|
||
# Version: 1.1
|
||
# Usage: python AddFile.py
|
||
# 注意: 请将本文件放在uvproj文件同一目录下运行
|
||
# Note: Please run this file in the same directory as the uvproj file
|
||
# 描述: 添加文件，自动修改uvproj文件，添加文件到指定组
|
||
# Description: Add files, automatically modify the uvproj file, and add files to the specified group
|
||
# 警告: 请注意备份uvproj文件及相关项目文件，以免出现意外，使用风险自负
|
||
# Warning: Please backup the uvproj file and related project files to avoid accidents, USE AT YOUR OWN RISK
|
||
############################################################################################################
|
||
|
||
import os
|
||
import re
|
||
import xml.dom.minidom
|
||
|
||
# 使用前请根据情况修改
|
||
source_file_path = "./Source"
|
||
header_file_path = "./Header"
|
||
source_group_name = "Source"
|
||
header_group_name = "Header"
|
||
auto_include_in_source = 0
|
||
auto_include_in_header = '#include "common.h"'
|
||
include_h_in_c_relative_path = False # 是否使用相对路径包含头文件
|
||
change_to_create_hex_file = True
|
||
|
||
|
||
|
||
def AddFileToAGroup(file_name, path_name, group_name,DOMTree):
|
||
collection = DOMTree.documentElement
|
||
groups = collection.getElementsByTagName("Group")
|
||
group = None
|
||
for group_t in groups:
|
||
# ad = group_t.getElementsByTagName("GroupName")
|
||
if group_t.getElementsByTagName("GroupName")[0].childNodes[0].data == group_name:
|
||
group = group_t
|
||
break
|
||
else:
|
||
group_t = DOMTree.createElement("Group")
|
||
group_t.appendChild(DOMTree.createElement("GroupName"))
|
||
group_t.getElementsByTagName("GroupName")[0].appendChild(DOMTree.createTextNode(group_name))
|
||
group_t.appendChild(DOMTree.createElement("Files"))
|
||
collection.getElementsByTagName("Groups")[0].appendChild(group_t)
|
||
group = group_t
|
||
|
||
ruled_path = ".\\"+os.path.normpath(os.path.join(path_name, file_name))
|
||
for file_path_C in group.getElementsByTagName("FilePath"):
|
||
if file_path_C.childNodes[0].data == ruled_path:
|
||
input("The file has already existed!")
|
||
exit(-1)
|
||
files = group.getElementsByTagName("Files")[0]
|
||
file = DOMTree.createElement("File")
|
||
file.appendChild(DOMTree.createElement("FileName"))
|
||
file.getElementsByTagName("FileName")[0].appendChild(DOMTree.createTextNode(file_name))
|
||
file.appendChild(DOMTree.createElement("FileType"))
|
||
if file_name.endswith(".c"):
|
||
file.getElementsByTagName("FileType")[0].appendChild(DOMTree.createTextNode("1"))
|
||
elif file_name.endswith(".h"):
|
||
file.getElementsByTagName("FileType")[0].appendChild(DOMTree.createTextNode("5"))
|
||
file.appendChild(DOMTree.createElement("FilePath"))
|
||
file.getElementsByTagName("FilePath")[0].appendChild(DOMTree.createTextNode(ruled_path))
|
||
files.appendChild(file)
|
||
print("Add file to group success!")
|
||
|
||
def createHeaderFile(header_file_name):
|
||
if not os.path.exists(header_file_path):
|
||
os.makedirs(header_file_path)
|
||
if not os.path.exists(os.path.join(header_file_path , header_file_name)):
|
||
with open(os.path.join(header_file_path , header_file_name), "w") as f:
|
||
f.write("#ifndef __" + header_file_name[:-2].upper() + "_H__\n")
|
||
f.write("#define __" + header_file_name[:-2].upper() + "_H__\n\n")
|
||
if auto_include_in_header:
|
||
f.write(auto_include_in_header + "\n\n\n")
|
||
f.write("#endif\n")
|
||
print("Create header file success!")
|
||
|
||
def createSourceFile(file_name,has_header):
|
||
if not os.path.exists(source_file_path):
|
||
os.makedirs(source_file_path)
|
||
if not os.path.exists(os.path.join(source_file_path , file_name)):
|
||
with open(os.path.join(source_file_path , file_name), "w") as f:
|
||
if auto_include_in_source:
|
||
f.write(auto_include_in_source + "\n")
|
||
# f.write(auto_include_in_source + "\n")
|
||
if has_header:
|
||
if include_h_in_c_relative_path:
|
||
f.write("#include \"./"+str(os.path.normpath(os.path.relpath(os.path.join(header_file_path, file_name)[:-2] + ".h",os.path.join(source_file_path, file_name)).replace("..",".",1))).replace("\\","/")+ "\"\n")
|
||
else:
|
||
f.write("#include \""+file_name[:-2] + ".h\"\n")
|
||
print("Create source file success!")
|
||
|
||
|
||
|
||
if __name__ == "__main__":
|
||
file_name = input("Please input the file name: ").strip()
|
||
header_file_name = ""
|
||
if file_name.lower().endswith(".c"):
|
||
if_create_header = input("Do you want to create a header file for this file?([y]/n): ").strip()
|
||
if not if_create_header.lower() == "n":
|
||
header_file_name = file_name[:-1] + "h"
|
||
|
||
elif file_name.lower().endswith(".h"):
|
||
header_file_name = file_name
|
||
file_name = ""
|
||
|
||
else:
|
||
input("The file type is not support!")
|
||
exit(-1)
|
||
|
||
uvproj_files = []
|
||
|
||
for file in os.listdir(os.path.abspath(os.path.dirname(__file__))):
|
||
file = file.lower()
|
||
if file.endswith(".uvproj") or file.endswith(".uvprojx"):
|
||
uvproj_files.append(file)
|
||
if len(uvproj_files) == 0:
|
||
input("No UV project file found!")
|
||
# input("Press any key to exit...")
|
||
exit(-1)
|
||
elif len(uvproj_files) > 1:
|
||
input("More than one UV project file found!")
|
||
exit(-1)
|
||
uvproj_file = uvproj_files[0]
|
||
xml_str = ""
|
||
with open(uvproj_file, "r", encoding="utf-8") as f:
|
||
xml_str = f.readlines()
|
||
xml_str = [x for x in xml_str if x.strip()]
|
||
xml_str = "".join(xml_str)
|
||
xml_str = xml_str.replace("\n", "")
|
||
xml_str = xml_str.replace("\t", "")
|
||
|
||
DOMTree = xml.dom.minidom.parseString(xml_str)
|
||
|
||
collection = DOMTree.documentElement
|
||
|
||
if change_to_create_hex_file:
|
||
collection.getElementsByTagName("CreateHexFile")[0].childNodes[0].data = "1"
|
||
|
||
if file_name != "":
|
||
createSourceFile(file_name,header_file_name != "")
|
||
AddFileToAGroup(file_name, source_file_path, source_group_name,DOMTree)
|
||
|
||
if header_file_name != "":
|
||
createHeaderFile(header_file_name)
|
||
AddFileToAGroup(header_file_name, header_file_path, header_group_name,DOMTree)
|
||
|
||
xml_str = DOMTree.toprettyxml().replace("\t"," ")
|
||
xml_str = re.sub(r'<(\w+)([^>]*)/>', r'<\1\2></\1>', xml_str)
|
||
with open(uvproj_file, "w", encoding="utf-8") as f:
|
||
# DOMTree.writexml(f, addindent=" ", newl="\n", )
|
||
f.write(xml_str)
|
||
print("Modify uvproj file success!")
|
||
input("Add file success! Press any key to exit...") |