添加请求超时

This commit is contained in:
shmilylty
2019-08-18 02:45:32 +08:00
parent 4c6af1a925
commit de96822c0d
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -25,7 +25,7 @@ class DNSdb(Query):
scraper.interpreter = 'js2py' scraper.interpreter = 'js2py'
scraper.proxies = self.get_proxy(self.source) scraper.proxies = self.get_proxy(self.source)
try: try:
tokens = scraper.get_tokens(self.url) tokens = scraper.get_tokens(self.url, timeout=self.timeout)
except Exception as e: except Exception as e:
logger.log('ERROR', e.args) logger.log('ERROR', e.args)
return None return None
+1 -1
View File
@@ -19,7 +19,7 @@ class ThreatCrowd(Query):
scraper.proxies = self.get_proxy(self.source) scraper.proxies = self.get_proxy(self.source)
url = self.addr + self.domain url = self.addr + self.domain
try: try:
resp = scraper.get(url) resp = scraper.get(url, timeout=self.timeout)
except Exception as e: except Exception as e:
logger.log('ERROR', e.args) logger.log('ERROR', e.args)
return return