如何在文件顶部添加新的文本行？

Question

我正在开发一个使Python脚本可执行的简单程序，我正在添加解释器路径（#）！/usr/bin/python）。我试过这样做，但没有添加新行，而是替换了当前的并删除了下一行的一部分。我做错什么了？

我把源代码上传到Ubuntu Pastebin:http://pastebin.ubuntu.com/1032683/错误的代码在第28行和第31行之间：

wfile = open(file, 'r+')
if wfile.readline() != "#! /usr/bin/python\n":
    wfile.seek(0)
    wfile.write("#! /usr/bin/python\n")

将Python 2.7.2与ipad2（Python for iOS）一起使用，同时在同一个iPad（Cydia端口）中使用2.5.1进行测试。

Answer 1

要在命令行中添加/替换每个文件中的第一行：

#!/usr/bin/env python
import fileinput

shebang = "#! /usr/bin/python\n"
for line in fileinput.input(inplace=1):
    if fileinput.isfirstline() and line != shebang:
       print shebang,
       if not line.startswith("#!"): 
          print line,
    else:
       print line,