From de96822c0df0d7ed35d914df9527229d9cd70af9 Mon Sep 17 00:00:00 2001 From: shmilylty Date: Sun, 18 Aug 2019 02:45:32 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=AF=B7=E6=B1=82=E8=B6=85?= =?UTF-8?q?=E6=97=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- oneforall/modules/datasets/dnsdb.py | 2 +- oneforall/modules/datasets/threatcrowd.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/oneforall/modules/datasets/dnsdb.py b/oneforall/modules/datasets/dnsdb.py index be79806..c50f254 100644 --- a/oneforall/modules/datasets/dnsdb.py +++ b/oneforall/modules/datasets/dnsdb.py @@ -25,7 +25,7 @@ class DNSdb(Query): scraper.interpreter = 'js2py' scraper.proxies = self.get_proxy(self.source) try: - tokens = scraper.get_tokens(self.url) + tokens = scraper.get_tokens(self.url, timeout=self.timeout) except Exception as e: logger.log('ERROR', e.args) return None diff --git a/oneforall/modules/datasets/threatcrowd.py b/oneforall/modules/datasets/threatcrowd.py index 63628f1..f8062bc 100644 --- a/oneforall/modules/datasets/threatcrowd.py +++ b/oneforall/modules/datasets/threatcrowd.py @@ -19,7 +19,7 @@ class ThreatCrowd(Query): scraper.proxies = self.get_proxy(self.source) url = self.addr + self.domain try: - resp = scraper.get(url) + resp = scraper.get(url, timeout=self.timeout) except Exception as e: logger.log('ERROR', e.args) return