Question

我正在研究我的第一个python“app”，并在Stackoverflow的参与者提出一些建议之后。我决定废弃我的东西并从头开始。

它似乎正在很好地解析参数的使用等但我不确定我是如何将值分配给args的？

我是否必须创建ifs的嵌套？如果是这样，我如何为docopt中的args做到这一点？

也许是这样的？

if opt in ("-f", "--file"):
    FWORD = arg

代码

#!/usr/bin/python

"""
Basic domain bruteforcer

Usage:
  your_script.py (-f <file>) (-d <domain>) [-t 10] [-v]
  your_script.py -h | --help

Options:
  -h --help     Show this screen.
  -f --file     File to read potential Sub-domains from. (Required argument)
  -p --proxy    Proxy address and port. [default: http://127.0.0.1:8080] (Optional)
  -d --domain   Domain to bruteforce.(Required argument)
  -t --thread   Thread count. (Optional)
  -v --verbose  Turn debug on. (Optional)
"""
from docopt import docopt

def fread(FWORD, *args):
    flist = open(FWORD).readlines()
    return flist


if __name__ == "__main__":
        arguments = docopt(__doc__, version='0.1a')
        print fread(fword)

Answer 1

你几乎得到了它。你的参数变量包含参数，你可以像在dict中那样查找它们。因此，如果您想使用file参数调用fread函数，那么main将如下所示：

if __name__ == "__main__":
    arguments = docopt(__doc__, version='0.1a')
    fread(arguments['<file>'])

如果你这样调用脚本：

> python your_script.py -f myfiles/file.txt -d google.com

然后你的论点将如下所示：

>>> print arguments
{'--domain': True,
 '--file': True,
 '--help': False,
 '--thread': False,
 '--verbose': False,
 '10': False,
 '<domain>': 'google.com',
 '<file>': 'myfiles/file.txt'}

Answer 2

您应该从python标准库中查看argparse。

使用docopt定义参数值

2 个答案: