if not options.domain: parse.error('必须输入要收集的域名') if not options.gevent: options.gevent = 30 # region 初始化excel excel = Excel(options.domain) title = ['URL', 'IP地址', '状态', '标题'] for i in range(0, len(title)): excel.sheet1.write(excel.getMaxRows(), i, title[i], excel.setStyle('Microsoft YaHei', 200, True)) excel.xls.save(excel.excel_file_name) # endregion # region 载入数据到队列 task_queue = queue.Queue() with open('subdomain-list.txt', 'r') as f: line = f.readline() while line: task_queue.put(line.strip() + '.' + options.domain) line = f.readline() progress_bar = ProgressBar.Bar(task_queue.qsize()) # endregion # gevent jobs = [] for i in range(options.gevent): jobs.append(gevent.spawn(Scan().run)) gevent.joinall(jobs) # endregion