river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 04:47:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化

Browse Source
This commit is contained in:
Jing Ling
2020-09-16 00:23:34 +08:00
parent add9a864b2
commit 899e243a3a
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/datasets/robtex.py
+6 -6
View File
@@ -10,7 +10,6 @@ class Robtex(Query):
self.domain = domain self.domain = domain
self.module = 'Dataset' self.module = 'Dataset'
self.source = "RobtexQuery" self.source = "RobtexQuery"
self.addr = 'https://freeapi.robtex.com/pdns/'
def query(self): def query(self):
""" """
@@ -18,17 +17,18 @@ class Robtex(Query):
""" """
self.header = self.get_header() self.header = self.get_header()
self.proxy = self.get_proxy(self.source) 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) resp = self.get(url)
if not resp: if not resp:
return return
text_list = resp.text.splitlines() text_list = resp.text.splitlines()
text_json = list(map(lambda x: json.loads(x), text_list)) for item in text_list:
for record in text_json: record = json.loads(item)
if record.get('rrtype') in ['A', 'AAAA']: if record.get('rrtype') in ['A', 'AAAA']:
time.sleep(self.delay) # Robtex有查询频率限制 time.sleep(self.delay) # Robtex有查询频率限制
ip = record.get('rrdata') ip = record.get('rrdata')
url = self.addr + 'reverse/' + ip url = f'{base_addr}/reverse/{ip}'
resp = self.get(url) resp = self.get(url)
self.subdomains = self.collect_subdomains(resp) self.subdomains = self.collect_subdomains(resp)
@@ -55,4 +55,4 @@ def run(domain):
if __name__ == '__main__': if __name__ == '__main__':
run('example.com') run('google.com')