使用DocOpt定义参数值



im在我的第一个python"应用程序"上工作,并在参与者在Stackoverflow上提供了一些建议。香港专业教育学院决定废除我的所拥有的东西,然后从头开始。

似乎很好地解析了用于用法等的论点,但我不确定我是如何将值分配给args的?

我必须创建IF的巢吗?如果是这样,我该如何为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)

您几乎得到了。您的参数变量包含该论点,并且您会像命令中一样查找它们。因此,如果您想用文件参数调用fread函数,那么您的主要内容将看起来像这样:

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'}

您应该从Python标准库中查看ArgParse。

相关内容

  • 没有找到相关文章

最新更新