From 899e243a3a5307e119e87ea9ef078e8d671237f4 Mon Sep 17 00:00:00 2001 From: Jing Ling Date: Wed, 16 Sep 2020 00:23:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- modules/datasets/robtex.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/modules/datasets/robtex.py b/modules/datasets/robtex.py index 9e21bd2..5a93cd3 100644 --- a/modules/datasets/robtex.py +++ b/modules/datasets/robtex.py @@ -10,7 +10,6 @@ class Robtex(Query): self.domain = domain self.module = 'Dataset' self.source = "RobtexQuery" - self.addr = 'https://freeapi.robtex.com/pdns/' def query(self): """ @@ -18,17 +17,18 @@ class Robtex(Query): """ self.header = self.get_header() self.proxy = self.get_proxy(self.source) - url = self.addr + 'forward/' + self.domain + base_addr = 'https://freeapi.robtex.com/pdns/' + url = f'{base_addr}/forward/{self.domain}' resp = self.get(url) if not resp: return text_list = resp.text.splitlines() - text_json = list(map(lambda x: json.loads(x), text_list)) - for record in text_json: + for item in text_list: + record = json.loads(item) if record.get('rrtype') in ['A', 'AAAA']: time.sleep(self.delay) # Robtex有查询频率限制 ip = record.get('rrdata') - url = self.addr + 'reverse/' + ip + url = f'{base_addr}/reverse/{ip}' resp = self.get(url) self.subdomains = self.collect_subdomains(resp) @@ -55,4 +55,4 @@ def run(domain): if __name__ == '__main__': - run('example.com') + run('google.com')