Files
QD4C-firmware/set_eide_env.py

55 lines
2.0 KiB
Python
Raw Normal View History

2023-12-11 21:45:06 +08:00
import fileinput
import os
tempelate_file_path = '.\\template.code-workspace_t'
target_file_path = '.\\violet_firmware_zf.code-workspace'
print("*************************************************************************************")
while True:
print("*************************************************************************************")
mrs_riscv_gcc_path_str = input("请输入 MRS 工具链下 RISCV-GCC 路径 (例如 D:\\xpack-riscv-none-embed-gcc-8.3.0-2.3):\n")
print("获取到输入:%s" %mrs_riscv_gcc_path_str)
if not os.path.exists(mrs_riscv_gcc_path_str):
print("GCC 路径错误,请输入有效路径")
else:
print("输入正确")
break
while True:
print("*************************************************************************************")
mrs_openocd_path_str = input("请输入 MRS 工具链下 OpenOCD 路径 (例如 D:\\OpenOCD\\bin\\openocd.exe):\n")
print("获取到输入:%s" %mrs_openocd_path_str)
if not os.path.exists(mrs_openocd_path_str):
print("OpenOCD 路径错误,请输入有效路径")
else:
print("输入正确")
break
mrs_riscv_gcc_path_str = mrs_riscv_gcc_path_str.replace("\\", "/")
mrs_openocd_path_str = mrs_openocd_path_str.replace("\\", "/")
# 检查模板文件是否存在
if not os.path.isfile(tempelate_file_path):
print(f"模板文件 {tempelate_file_path} 不存在")
exit(1)
# 复制模板文件到工作区文件
if os.path.exists(target_file_path):
print(f"工作区文件 {target_file_path} 已存在,将会被覆盖")
if_continue = input("输入 y 以继续:\n")
if if_continue != 'y':
print("程序退出")
exit(1)
with open(tempelate_file_path, "r") as f:
lines = f.readlines()
with open(target_file_path, "w") as f:
for line in lines:
# 替换字符串 YOUR_WCH_OPENOCD_PATH 和 YOUR_WCH_GCC_PATH
f.write(line.replace("YOUR_WCH_OPENOCD_PATH", mrs_openocd_path_str).replace("YOUR_WCH_RISCV_GCC_PATH", mrs_riscv_gcc_path_str))
print("处理完成!")