python - 我不知道如何让这个脚本将受感染文件列表打印到输出文件

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 中找到病毒，我已经这样做了，但需要它将受感染文件的列表打印到输出文件中。

标签： pythonpython-2.7listdirectory

Answer 1

我用这个：

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

它创建了一个输出 txt 文件，但它不会将受感染的文件放入该文件？我怎样才能从中实现这一目标。对不起，伙计们，我是 python 新手，很烂。