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:
JrD
2020-07-08 12:08:43 +08:00
parent 6de0776c0b
commit 85f9770c97
1 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/datasets/cloudflare_api.py
+27 -17
View File
@@ -23,32 +23,36 @@ class CloudFlareAPI(Query):
query from source
"""
account_id_resp = self.get(self.addr + 'accounts')
if not account_id_resp:
if account_id_resp.status_code != 200:
return
account_id = account_id_resp.json()['result'][0]['id']
# query domain zone, if it not exist, create
zones_resp = self.get(self.addr + 'zones',
params={'name': self.domain}, check=False)
if not zones_resp:
return
if zones_resp.json()['success'] and not zones_resp.json()['result']:
zone_id = self.create_zone(account_id)
if zone_id:
self.list_dns(zone_id)
if zones_resp.status_code == 200:
if zones_resp.json()['success'] and not zones_resp.json()['result']:
zone_id = self.create_zone(account_id)
if zone_id:
self.list_dns(zone_id)
return
else:
return
elif zones_resp.json()['success']:
zone_id = zones_resp.json()['result'][0]['id']
delete_zone_resp = self.delete(self.addr + f'zones/{zone_id}', check=False)
zone_id = self.create_zone(account_id)
if zone_id:
self.list_dns(zone_id)
return
else:
return
elif zones_resp.json()['success']:
zone_id = zones_resp.json()['result'][0]['id']
delete_zone_resp = self.delete(self.addr + f'zones/{zone_id}', check=False)
zone_id = self.create_zone(account_id)
if zone_id:
self.list_dns(zone_id)
return
elif not zones_resp.json()['success']:
elif zones_resp.status_code == 403:
logger.log('DEBUG',
f'{self.domain} is banned or not a registered domain, so cannot be added to Cloudflare.')
return
else:
logger.log('DEBUG',
f'{zones_resp.status_code} {zones_resp.text}')
return
def create_zone(self, account_id):
data = {"name": self.domain, "account": {"id": account_id}, "jump_start": True,
@@ -56,6 +60,8 @@ class CloudFlareAPI(Query):
create_zone_resp = self.post(
self.addr + 'zones', json=data, check=False)
if not create_zone_resp:
logger.log('DEBUG',
f'{create_zone_resp.status_code} {create_zone_resp.text}')
return
if create_zone_resp.json()['success']:
return create_zone_resp.json()['result']['id']
@@ -67,6 +73,8 @@ class CloudFlareAPI(Query):
page = 1
list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records', params={'page': page, 'per_page': 10})
if not list_dns_resp:
logger.log('DEBUG',
f'{list_dns_resp.status_code} {list_dns_resp.text}')
return
subdomains = self.match_subdomains(self.domain, list_dns_resp.text)
self.subdomains = self.subdomains.union(subdomains)
@@ -79,6 +87,8 @@ class CloudFlareAPI(Query):
list_dns_resp = self.get(self.addr + f'zones/{zone_id}/dns_records',
params={'page': page, 'per_page': 10})
if not list_dns_resp:
logger.log('DEBUG',
f'{list_dns_resp.status_code} {list_dns_resp.text}')
return
total_pages = list_dns_resp.json()['result_info']['total_pages']
subdomains = (self.match_subdomains(self.domain, list_dns_resp.text))