使用详细输出进行错误处理

时间:2014-03-07 11:33:13

标签: python docopt

我试图在我的脚本中实现--verbose选项。这个想法是打开额外的错误打印等进行调试,但由于某些原因它似乎没有用。我尝试了if verbose声明的一些变体,但没有喜悦。我希望有人能指出我正确的方向吗?

CLI示例

./attack2.py -f wordfile.txt -d google.com --verbose 1

CLI OUTPUT

unknown@ubuntu:~$ ./attack2.py -f wordfile.txt -d google.com --verbose 1
173.194.34.149
173.194.34.130
unknown@ubuntu:~$

ARG PRINT

{'--domain': 'google.com',
 '--file': 'wordfile.txt',
 '--help': False,
 '--thread': False,
 '--verbose': True,
 '10': False,
 '<1>': '1'}

CODE

#!/usr/bin/python

"""
Description:

Basic Domain bruteforcer

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

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


def fread(dwords):
        flist = open(dwords).readlines()
        return [s.replace('\n', '.') for s in flist]


def subcheck(subdomain, domain, verbose):

        vdomain = {}
        for sub in subdomain:
                try:
                        check = socket.gethostbyname(sub + domain)
                        vdomain[sub + domain] = check

                except socket.gaierror, e:
                        if verbose == True:
                                print arguments
                                print e, sub + domain
                        else:
                                pass
        return vdomain

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

        fread(arguments['--file'])
        returned_list = fread(arguments['--file'])
        returned_domains = subcheck(returned_list, arguments['--domain'], ['--verbose'])

2 个答案:

答案 0 :(得分:1)

功能subcheck

中的以下行
returned_domains = subcheck(returned_list, arguments['--domain'], ['--verbose'])

应该是

returned_domains = subcheck(returned_list, arguments['--domain'], arguments['--verbose'])

您忘记从参数传递verbose参数,而是传递了list

答案 1 :(得分:0)

我似乎错误地调用了子检查。

代码应该看起来像这样

工作代码

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

        fread(arguments['--file'])
        returned_list = fread(arguments['--file'])
        returned_domains = subcheck(returned_list, arguments['--domain'], arguments['--verbose'])
        print returned_domains