基本病毒扫描递归

Question

因此，我正在编写接受pathname、signature和第三个参数的代码，第三个参数是number，它表示应该扫描的子目录的深度。

因此，假设我有一个文件: test

In it is test1 folder,test2 folder,antivirus.py,simple.py
In test1 folder is antivirus1.py
In test2 folder is test3 folder, and antivirus2.py
In test3 folder is antivirus3.py

，所以它应该是这样工作的：

>>>scan('test',rules, 0)
test\antivirus.py, found virus Virus2
test\antivirus.py, found virus Virus1
>>>
>>>scan('test',rules, 2)
test\antivirus.py, found virus Virus2
test\antivirus.py, found virus Virus1
test\test1\antivirus1.py, found virus Virus2
test\test1\antivirus1.py, found virus Virus1
test\test2\antivirus2.py, found virus Virus2
test\test2\antivirus2.py, found virus Virus1

这里是我的当前代码：

def scan(pathname, signatures, depth):
    for item in os.listdir(pathname) and depth > 0:
        n = os.path.join(pathname, item)
        try:
            scan(n, signatures, depth-1)
        except:
            f = open(n, 'r')
            s = f.read()
            for virus in signatures:
                if s.find(signatures[virus]) > 0:
                    print('{}, found virus {}'.format(n,virus))
            f.close()

Answer 1

for循环并不是那样工作的。它们的语法是for <variable> in <iterable>。

相反，我们只需要一个if语句：

if depth <= 0:
    return

for item in os.listdir(pathname):