def main(): url = '' t = 30 r = 10 c = '' r = False d = 5 try: opt,argv = getopt.getopt(sys.argv[1:],'hu:t:',['help','url']) if opt == []: print '\t[!] Please input the parameter.' usage() exit() except getopt.GetoptError as e: print e usage() exit(0) for o,v in opt: if o in ('-h','--help'): usage() exit(0) if o in ('-u','--url'): if v == '': print 'url can\'t be null' exit(0) url.strip() if v[0:7] != 'http://': v = 'http://' + v if v[-1] != '/': v += '/' url = v if o in ('-t','--thread_num'): if v.isdigit(): thread_num = int(v) else : print 'Thread_num is NOT a digit!' exit(0) if o in ('-c','--cookie'): c = serializeCookie(v) if o in ('-r','--retry'): if v.isdigit(): retry = int(v) else : print 'retry times is NOT a digit!' exit(0) # if o in ('-R','--recursive'): # r = True # if o in ('-d','--depth') and v.isdigit(): print '[*] Start' sTime = time.time() # print str(sTime) a = Core(url,t,c,r) a.createThread() a.getRes() print '[*] Used ' + str(time.time()-sTime) + ' s'