错误处理与详细输出

2024-06-06 12:13:33 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在我的脚本中实现--verbose选项。这个想法是为了调试打开额外的错误打印,但是由于某些原因它似乎不起作用。我尝试了if verbose语句的一些变体,但没有找到乐趣。我希望有人能给我指出正确的方向?在

CLI示例

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

CLI输出

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

ARG打印

^{pr2}$

编码

#!/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'])

Tags: pyverbosecliifdomainsocketdocoptarguments
2条回答

函数subcheck中的以下行

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

应该是

^{pr2}$

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

我好像打错了电话。在

代码应该是这样的

工作代码

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

相关问题 更多 >