使用下面的程序可以将当前文件夹下面的.c文件的写入main.h文件,我这里是将自己的基于标准库stm32初始化io文件为例。
import os
base = [
"#ifndef main_H",
"#define main_H\n",
"#endif"
]
includes = set()
for file in os.listdir():
if file.endswith('.c'):
try:
with open(file, 'r', encoding='utf-8') as f:
includes.update(line.strip() for line in f if line.startswith('#include'))
except Exception: pass
if includes:
base.insert(2, '\n'.join(includes) + '\n')
with open('main.h', 'w', encoding='utf-8') as f:
f.write('\n'.join(base))
演示执行