river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

pep8

Browse Source
This commit is contained in:
shmilylty
2019-08-13 19:13:56 +08:00
parent f223d9cf1b
commit ef3a1f5245
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/aiobrute.py
+4 -5
View File
@@ -14,7 +14,7 @@ import secrets
import signal import signal
import time import time
import aiomultiprocess import aiomultiprocess as aiomp
import exrex import exrex
import fire import fire
import tqdm import tqdm
@@ -242,10 +242,9 @@ class AIOBrute(Module):
logger.log('INFOR', f'正在爆破{domain}的域名') logger.log('INFOR', f'正在爆破{domain}的域名')
for task in tqdm.tqdm(tasks, desc='Progress', for task in tqdm.tqdm(tasks, desc='Progress',
smoothing=1.0, ncols=True): smoothing=1.0, ncols=True):
async with aiomultiprocess.Pool(processes=self.processes, async with aiomp.Pool(processes=self.processes,
initializer=init_worker, initializer=init_worker,
childconcurrency=self.coroutine)\ childconcurrency=self.coroutine) as pool:
as pool:
try: try:
results = await pool.map(resolve.aiodns_query_a, task) results = await pool.map(resolve.aiodns_query_a, task)
except KeyboardInterrupt: except KeyboardInterrupt: