從頭開始編寫文件

Question

我正在編寫我的第一個用於查找和刪除惡意軟件js-code的服務器腳本，但是我從一開始就找不到一個簡單的方法來重寫現有文件，而不是最後。

# -*- 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()然後切斷任何額外的文本。

Answer 2

filepath = os.path.join(dirname, filename) 
text = file.read(filepath) 
if re.search('function g\(\).*\n.*\<script src=\"http://linkfooter.org/linkfooter.js\"></script>\'\);}', line) and os.path.join(filename) != "bezr.py": 
    with open(filepath, 'w') as f: 
      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)