v1d0m.py 文件源码

def main (argv):
    parser = argparse.ArgumentParser(description='This script obtains subdomains throught VirusTotal', formatter_class=RawTextHelpFormatter)
    parser.add_argument('-e','--export', help="Export the results to a json file (Y/N)\n Format available:\n\t1.json\n\t2.xlsx", required=False)
    parser.add_argument('-d','--domain', help="The domain to search subdomains",required=True)
    args = parser.parse_args()
    banner()
    help()
    target = args.domain
    output=args.export
    export = ""
    if output is None:
        export='N'
    if ((output == 'y') or (output == 'Y')):
        print "Select the output format:"
        print "\n\t(js).json"
        print "\n\t(xl).xlsx"
        export = raw_input()
        if ((export != "js") and (export != "xl")):
            print "Incorrect output format selected."
            exit(1)
    SendRequest(target,export)