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
