195 lines
7.7 KiB
Python
195 lines
7.7 KiB
Python
from sslModel.base import sslBase
|
||
import requests
|
||
from urllib.parse import urlparse
|
||
from urllib.parse import urljoin
|
||
import public
|
||
|
||
|
||
class main(sslBase):
|
||
dns_provider_name = "cloudflare"
|
||
_type = 0 # 0:lest 1：锐成
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
def __init_data(self, data):
|
||
self.CLOUDFLARE_EMAIL = data['E-Mail']
|
||
self.CLOUDFLARE_API_KEY = data['API Key']
|
||
self.CLOUDFLARE_API_BASE_URL = 'https://api.cloudflare.com/client/v4/'
|
||
self.HTTP_TIMEOUT = 65 # seconds
|
||
|
||
self.headers = {
|
||
"X-Auth-Email": self.CLOUDFLARE_EMAIL,
|
||
"X-Auth-Key": self.CLOUDFLARE_API_KEY
|
||
}
|
||
|
||
def get_dns_record(self, get):
|
||
domain_name = get.domain_name
|
||
dns_id = get.dns_id
|
||
self.__init_data(self.get_dns_data(None)[dns_id])
|
||
|
||
root_domain, _, sub_domain = self.extract_zone(domain_name)
|
||
data = {}
|
||
try:
|
||
zone_dic = self.get_zoneid_dic(get)
|
||
zone_id = zone_dic[root_domain]
|
||
url = urljoin(self.CLOUDFLARE_API_BASE_URL, "zones/{}/dns_records".format(zone_id))
|
||
response = requests.get(url, headers=self.headers, timeout=self.HTTP_TIMEOUT).json()
|
||
data = {
|
||
"info": {
|
||
'record_total': response['result_info']['total_count']
|
||
},
|
||
"list": [
|
||
{
|
||
"RecordId": i["id"],
|
||
"name": i["name"],
|
||
"domain_name": i["name"],
|
||
"value": i["content"],
|
||
"line": "默认",
|
||
"ttl": i["ttl"],
|
||
"type": i["type"],
|
||
"status": "启用",
|
||
"mx": i.get("priority") or 0,
|
||
"updated_on": i["modified_on"],
|
||
"remark": i.get("comment") or "",
|
||
}
|
||
for i in response["result"]
|
||
]
|
||
}
|
||
|
||
except Exception as e:
|
||
pass
|
||
self.set_record_data({domain_name: data})
|
||
return data
|
||
|
||
def create_dns_record(self, get):
|
||
domain_name = get.domain_name
|
||
dns_id = get.dns_id
|
||
record_type = get.record_type
|
||
domain_dns_value = get.domain_dns_value
|
||
|
||
self.__init_data(self.get_dns_data(None)[dns_id])
|
||
|
||
root_domain, sub_domain, _ = self.extract_zone(domain_name)
|
||
body = {
|
||
"content": get.domain_dns_value,
|
||
"name": sub_domain or '@',
|
||
"type": get.record_type if 'record_type' in get else 'TXT'
|
||
}
|
||
# CAA记录特殊处理
|
||
if record_type == 'CAA':
|
||
values = domain_dns_value.split(' ')
|
||
if len(values) != 3 or values[1] not in ("issue", "issuewild", "iodef"):
|
||
return public.returnMsg(False, '解析记录格式错误，请检查后重试')
|
||
body["data"] = {
|
||
"flags": values[0],
|
||
"tag": values[1],
|
||
"value": values[2].replace('"', ''),
|
||
}
|
||
|
||
try:
|
||
zone_dic = self.get_zoneid_dic(get)
|
||
zone_id = zone_dic.get(root_domain)
|
||
if not zone_id:
|
||
return public.returnMsg(False, '此域名配置的dns账号不正确')
|
||
url = urljoin(self.CLOUDFLARE_API_BASE_URL, "zones/{}/dns_records".format(zone_id))
|
||
response = requests.post(url, headers=self.headers, json=body, timeout=self.HTTP_TIMEOUT)
|
||
if response.status_code == 200:
|
||
return public.returnMsg(True, '添加成功')
|
||
else:
|
||
return public.returnMsg(False, '添加失败，{}'.format(self.get_error(response.text)))
|
||
except Exception as e:
|
||
return public.returnMsg(False, '添加失败，msg：{}'.format(e))
|
||
|
||
def delete_dns_record(self, get):
|
||
domain_name = get.domain_name
|
||
dns_id = get.dns_id
|
||
RecordId = get.RecordId
|
||
|
||
self.__init_data(self.get_dns_data(None)[dns_id])
|
||
|
||
root_domain, sub_domain, _ = self.extract_zone(domain_name)
|
||
|
||
try:
|
||
zone_dic = self.get_zoneid_dic(get)
|
||
zone_id = zone_dic.get(root_domain)
|
||
if not zone_id:
|
||
return public.returnMsg(False, '此域名配置的dns账号不正确')
|
||
url = urljoin(self.CLOUDFLARE_API_BASE_URL, "zones/{}/dns_records/{}".format(zone_id, RecordId))
|
||
response = requests.delete(url, headers=self.headers, timeout=self.HTTP_TIMEOUT)
|
||
if response.status_code == 200:
|
||
return public.returnMsg(True, '删除成功')
|
||
else:
|
||
return public.returnMsg(False, self.get_error(response.text))
|
||
except Exception as e:
|
||
return public.returnMsg(False, '删除失败，msg：{}'.format(e))
|
||
|
||
def get_zoneid_dic(self, get):
|
||
dns_id = get.dns_id
|
||
self.__init_data(self.get_dns_data(None)[dns_id])
|
||
|
||
url = urljoin(self.CLOUDFLARE_API_BASE_URL, "zones?status=active&per_page=1000")
|
||
try:
|
||
response = requests.get(url, headers=self.headers, timeout=self.HTTP_TIMEOUT)
|
||
data = response.json()
|
||
return {i["name"]: i["id"] for i in data["result"]}
|
||
except:
|
||
return {}
|
||
|
||
def update_dns_record(self, get):
|
||
domain_name = get.domain_name
|
||
RecordId = get.RecordId
|
||
domain_dns_value = get.domain_dns_value
|
||
record_type = get.record_type
|
||
dns_id = get.dns_id
|
||
|
||
self.__init_data(self.get_dns_data(None)[dns_id])
|
||
|
||
root_domain, sub_domain, _ = self.extract_zone(domain_name)
|
||
|
||
body = {
|
||
"content": get.domain_dns_value,
|
||
"name": sub_domain or '@',
|
||
"type": get.record_type,
|
||
}
|
||
|
||
# CAA记录特殊处理
|
||
if record_type == 'CAA':
|
||
values = domain_dns_value.split(' ')
|
||
if len(values) != 3:
|
||
return public.returnMsg(False, '解析记录格式错误，请检查后重试')
|
||
body["data"] = {
|
||
"flags": values[0],
|
||
"tag": values[1],
|
||
"value": values[2].replace('"', ''),
|
||
}
|
||
try:
|
||
zone_dic = self.get_zoneid_dic(get)
|
||
zone_id = zone_dic.get(root_domain)
|
||
if not zone_id:
|
||
return public.returnMsg(False, '此域名配置的dns账号不正确')
|
||
url = urljoin(self.CLOUDFLARE_API_BASE_URL, "zones/{}/dns_records/{}".format(zone_id, RecordId))
|
||
response = requests.patch(url, headers=self.headers, json=body, timeout=self.HTTP_TIMEOUT)
|
||
if response.status_code == 200:
|
||
return public.returnMsg(True, '修改成功')
|
||
else:
|
||
return public.returnMsg(False, '修改失败，{}'.format(self.get_error(response.text)))
|
||
except Exception as e:
|
||
return public.returnMsg(False, '修改失败，msg：{}'.format(e))
|
||
|
||
def get_error(self, error):
|
||
if "Record does not exist" in error:
|
||
return "解析记录不存在"
|
||
elif "Content for A record must be a valid IPv4 address" in error:
|
||
return "【A】记录的解析值必须为IPv4地址"
|
||
elif "Content for CNAME record is invalid" in error:
|
||
return "请正确填写【CNAME】类型的解析值"
|
||
elif "DNS record type is invalid" in error:
|
||
return "解析记录类型无效"
|
||
elif "A record with the same settings already exists" in error:
|
||
return "已存在相同的解析记录"
|
||
elif "Error unable to get DNS zone for domain_name" in error:
|
||
return "这个cloudflare账户下面不存在这个域名，请检查dns接口配置后重试"
|
||
else:
|
||
return error
|