python - 从头开始写文件

Question

我正在编写我的第一个服务器脚本来查找和删除恶意软件 js 代码，但我找不到一种简单的方法来从头重写现有文件，而不是在最后重写。

# -*- coding: utf-8 -*-

import os
import re
import codecs

for dirname, dirnames, filenames in os.walk('.'):
    for filename in filenames:
        f = open(os.path.join(dirname, filename), 'r+b')
        text=f.read()
        if re.search('function g\(\).*\n.*\<script src=\"http://linkfooter.org/linkfooter.js\"></script>\'\);}', text) and os.path.join(filename) != "bezr.py":
            print "starting with " + os.path.join(filename)
            match = re.compile('function g\(\).*\n.*\<script src=\"http://linkfooter.org/linkfooter.js\"></script>\'\);}')
            s = match.sub('', text)
            f.write(s)
        f.close()
        #else:
            #print "in " + os.path.join(dirname, filename) + " none"    
            #f.close()

Answer 1

f.seek(0) 在你写之前。 f.truncate() 之后切断任何额外的文本。