调整执行finish函数位置

oneforall/modules/intelligence/virustotal_api.py

@@ -11,13 +11,13 @@ class VirusTotalAPI(Query):
         self.addr = 'https://www.virustotal.com/vtapi/v2/domain/report'
         self.key = config.virustotal_api_key
 
-    def query(self, domain):
+    def query(self):
         """
         向接口查询子域并做子域匹配
         """
         self.header = self.get_header()
         self.proxy = self.get_proxy(self.source)
-        params = {'apikey': self.key, 'domain': domain}
+        params = {'apikey': self.key, 'domain': self.domain}
         resp = self.get(self.addr, params)
         if not resp:
             return
@@ -32,11 +32,11 @@ class VirusTotalAPI(Query):
         if not self.check(self.key):
             return
         self.begin()
-        self.query(self.domain)
+        self.query()
+        self.finish()
         self.save_json()
         self.gen_result()
         self.save_db()
-        self.finish()
 
 
 def do(domain):  # 统一入口名字 方便多线程调用