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

添加响应体检查

Browse Source
This commit is contained in:
Jing Ling
2020-07-08 11:13:12 +08:00
parent f42e02daa0
commit 6de0776c0b
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/datasets/cloudflare_api.py
+12 -5
View File
@@ -23,10 +23,14 @@ class CloudFlareAPI(Query):
query from source query from source
""" """
account_id_resp = self.get(self.addr + 'accounts') account_id_resp = self.get(self.addr + 'accounts')
if not account_id_resp:
return
account_id = account_id_resp.json()['result'][0]['id'] account_id = account_id_resp.json()['result'][0]['id']
# query domain zone, if it not exist, create # query domain zone, if it not exist, create
zones_resp = self.get(self.addr + 'zones', zones_resp = self.get(self.addr + 'zones',
params={'name': self.domain}, check=False) params={'name': self.domain}, check=False)
if not zones_resp:
return
if zones_resp.json()['success'] and not zones_resp.json()['result']: if zones_resp.json()['success'] and not zones_resp.json()['result']:
zone_id = self.create_zone(account_id) zone_id = self.create_zone(account_id)
if zone_id: if zone_id:
@@ -51,6 +55,8 @@ class CloudFlareAPI(Query):
"type": "full"} "type": "full"}
create_zone_resp = self.post( create_zone_resp = self.post(
self.addr + 'zones', json=data, check=False) self.addr + 'zones', json=data, check=False)
if not create_zone_resp:
return
if create_zone_resp.json()['success']: if create_zone_resp.json()['success']:
return create_zone_resp.json()['result']['id'] return create_zone_resp.json()['result']['id']
else: else:
@@ -60,6 +66,8 @@ class CloudFlareAPI(Query):
def list_dns(self, zone_id): def list_dns(self, zone_id):
page = 1 page = 1
list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records', params={'page': page, 'per_page': 10}) list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records', params={'page': page, 'per_page': 10})
if not list_dns_resp:
return
subdomains = self.match_subdomains(self.domain, list_dns_resp.text) subdomains = self.match_subdomains(self.domain, list_dns_resp.text)
self.subdomains = self.subdomains.union(subdomains) self.subdomains = self.subdomains.union(subdomains)
if not self.subdomains: if not self.subdomains:
@@ -67,14 +75,13 @@ class CloudFlareAPI(Query):
sleep(5) sleep(5)
self.list_dns(zone_id) self.list_dns(zone_id)
else: else:
total_pages = list_dns_resp.json()['result_info']['total_pages']
while True: while True:
list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records', list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records',
params={'page': page, 'per_page': 10}) params={'page': page, 'per_page': 10})
total_pages = list_dns_resp.json( if not list_dns_resp:
)['result_info']['total_pages'] return
subdomains = (self.match_subdomains( total_pages = list_dns_resp.json()['result_info']['total_pages']
self.domain, list_dns_resp.text)) subdomains = (self.match_subdomains(self.domain, list_dns_resp.text))
self.subdomains = self.subdomains.union(subdomains) self.subdomains = self.subdomains.union(subdomains)
page += 1 page += 1
if page > total_pages: if page > total_pages: