修改为api调用

oneforall/modules/intelligence/threatbook_api.py

@@ -1,5 +1,4 @@
 # coding=utf-8
-import queue
 
 import config
 from common.query import Query
@@ -28,34 +27,29 @@ class ThreatBookAPI(Query):
         subdomain_find = self.match(self.domain, str(resp.json()))
         self.subdomains = self.subdomains.union(subdomain_find)
 
-    def run(self, rx_queue):
+    def run(self):
         """
         类执行入口
         """
-        if not self.key:
+        if not self.check(self.key):
-            logger.log('ERROR', f'{self.source}模块API配置错误')
-            logger.log('ALERT', f'不执行{self.source}模块')
             return
         self.begin()
         self.query(self.domain)
         self.save_json()
         self.gen_result()
         self.save_db()
-        rx_queue.put(self.results)
         self.finish()
 
 
-def do(domain, rx_queue):  # 统一入口名字 方便多线程调用
+def do(domain):  # 统一入口名字 方便多线程调用
     """
     类统一调用入口
 
     :param str domain: 域名
-    :param rx_queue: 结果集队列
     """
     query = ThreatBookAPI(domain)
-    query.run(rx_queue)
+    query.run()
 
 
 if __name__ == '__main__':
-    result_queue = queue.Queue()
+    do('example.com')
-    do('example.com', result_queue)