目录raw_input为变量

时间:2013-06-17 20:42:36

标签: python malware raw-input

尝试让我的部分脚本正常工作。此部分将让用户定义Web根目录,以便我们可以运行该目录的搜索以查找恶意软件。这是我正在研究的部分: 我需要能够获取用户输入并使其成为可添加到搜索功能的变量。我确定我这一切都错了。

import os
for root in raw_input("Where is the web root? "):
    if os.path.isdir(root):
        print "Using %r" % (root)
    elif os.path.isdir(root):
        print "Directory not found."

我想让脚本检查目录以确保它存在(如果是拼写错误),然后继续并运行搜索某些字符串。谢谢!

1 个答案:

答案 0 :(得分:1)

我猜你需要一个while循环:

import os
root = raw_input("Where is the web root? "
while not os.path.isdir(root):
    print "Directory not found, try again"
    root = raw_input("Where is the web root? ")

print "Using %r" % (root)
#or do something else with root here