只能在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模塊替換文件。