python - 仅在 Python 中的特定文件中修改

Question

我必须在文件开头添加一些行到目录中的特定文件，如（*.c 或 *.h）。我写了如下代码。

f = open('filename') 
text = f.read() 
f.close() 
f = open('filename~', 'w') 
f.write("This is the new first line\n") 
f.write(text) 
f.close() 
os.rename('filename~', 'filename') 

它获取目录中的所有文件并更新它。但我需要它应该更新一些特定的文件。怎么做

Answer 1

您可以使用glob模块来获取与 UNIX shell 模式匹配的所有文件名（例如*.c）。此外，您也许可以使用fileinput模块对文件进行就地替换。