我不知道如何获取此脚本以将受感染文件的列表打印到输出文件

Question

我已为此防病毒任务找到了特定的病毒SigData = Infected 。 我不知道如何完成此脚本，以使其将受感染文件的列表打印到输出文件。

我尝试了os.listdir(path) ，但是它说未定义路径。 我不知道如何从此代码打印我需要的列表：

import os
import inspect
SigData="Infected"


def search(path): #Searches for files in path.
    filestodisinfect = []
    filelist = os.listdir(path)
    for filename in filelist:
        if os.path.isdir(path+"/"+filename): #Looks for folder
            filestodisinfect.extend(search(path+"/"+filename))
        elif filename[-3:] == ".py": #Searches for python scripts to disinfect
            for line in open(path+"/"+filename):
                if SigData in line:
                    if infected == True:
                        filestodisinfect.append(path+"/"+filename)
    return filestodisinfect

我希望此脚本能够从sigdata中找到病毒，这已经完成了，但是需要它来将感染文件列表打印到输出文件中。

Answer 1

我用这个：

mylist=('')
myfile=open('output.txt','w') #Creates file to save list
for item in mylist:
    print >>myfile, item
myfile.close()

它创建了一个输出txt文件，但是不会将受感染的文件放到该文件中吗？ 我怎样才能做到这一点。 抱歉，我是python和吮吸的新手。