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

Question

I have this antivirus assignment to find a specific virus SigData = Infected . I am lost on how to finish this script to get it to print a list of infected files to an output file.

I tried os.listdir(path) , but it says the path is not defined. I don't know how else to print the list I need from this code:

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

I expect this script to find the virus from it's sigdata, which I have done, but need it to print a list of the infected files to an output file.

Answer 1

I used this:

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

It created an output txt file, but it won't put the infected files to that file? How can I achieve that from this. Sorry guys I'm new to python and suck.