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模块对文件进行就地替换。