From ef3a1f5245b6167384243967ff997fcba2ea6981 Mon Sep 17 00:00:00 2001
From: shmilylty <admin@hackfun.org>
Date: Tue, 13 Aug 2019 19:13:56 +0800
Subject: [PATCH] pep8

---
 oneforall/aiobrute.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/oneforall/aiobrute.py b/oneforall/aiobrute.py
index c5bf69b..0b9a4c2 100644
--- a/oneforall/aiobrute.py
+++ b/oneforall/aiobrute.py
@@ -14,7 +14,7 @@ import secrets
 import signal
 import time
 
-import aiomultiprocess
+import aiomultiprocess as aiomp
 import exrex
 import fire
 import tqdm
@@ -242,10 +242,9 @@ class AIOBrute(Module):
         logger.log('INFOR', f'正在爆破{domain}的域名')
         for task in tqdm.tqdm(tasks, desc='Progress',
                               smoothing=1.0, ncols=True):
-            async with aiomultiprocess.Pool(processes=self.processes,
-                                            initializer=init_worker,
-                                            childconcurrency=self.coroutine)\
-                    as pool:
+            async with aiomp.Pool(processes=self.processes,
+                                  initializer=init_worker,
+                                  childconcurrency=self.coroutine) as pool:
                 try:
                     results = await pool.map(resolve.aiodns_query_a, task)
                 except KeyboardInterrupt: