195 lines
7.9 KiB
Python
195 lines
7.9 KiB
Python
import hashlib
|
||
import hmac
|
||
import json
|
||
import time
|
||
from datetime import datetime
|
||
|
||
import public
|
||
import requests
|
||
from sslModel.base import sslBase
|
||
|
||
|
||
class main(sslBase):
|
||
dns_provider_name = "tencentcloud"
|
||
_type = 0
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.endpoint = "dnspod.tencentcloudapi.com"
|
||
self.host = "https://dnspod.tencentcloudapi.com"
|
||
self.version = "2021-03-23"
|
||
self.algorithm = "TC3-HMAC-SHA256"
|
||
|
||
def __init_data(self, data):
|
||
self.secret_id = data["secret_id"]
|
||
self.secret_key = data["secret_key"]
|
||
|
||
def get_headers(self, dns_id, action, payload, region="", token=""):
|
||
self.__init_data(self.get_dns_data(None)[dns_id])
|
||
timestamp = int(time.time())
|
||
# timestamp = 1551113065
|
||
date = datetime.utcfromtimestamp(timestamp).strftime("%Y-%m-%d")
|
||
# ************* 步骤 1：拼接规范请求串 *************
|
||
http_request_method = "POST"
|
||
canonical_uri = "/"
|
||
canonical_querystring = ""
|
||
ct = "application/json; charset=utf-8"
|
||
canonical_headers = "content-type:%s\nhost:%s\nx-tc-action:%s\n" % (ct, self.endpoint, action.lower())
|
||
signed_headers = "content-type;host;x-tc-action"
|
||
hashed_request_payload = hashlib.sha256(payload.encode("utf-8")).hexdigest()
|
||
canonical_request = (http_request_method + "\n" +
|
||
canonical_uri + "\n" +
|
||
canonical_querystring + "\n" +
|
||
canonical_headers + "\n" +
|
||
signed_headers + "\n" +
|
||
hashed_request_payload)
|
||
# ************* 步骤 2：拼接待签名字符串 *************
|
||
credential_scope = date + "/" + "dnspod" + "/" + "tc3_request"
|
||
hashed_canonical_request = hashlib.sha256(canonical_request.encode("utf-8")).hexdigest()
|
||
string_to_sign = (self.algorithm + "\n" +
|
||
str(timestamp) + "\n" +
|
||
credential_scope + "\n" +
|
||
hashed_canonical_request)
|
||
# ************* 步骤 3：计算签名 *************
|
||
# 计算签名摘要函数
|
||
def sign(key, msg):
|
||
return hmac.new(key, msg.encode("utf-8"), hashlib.sha256).digest()
|
||
secret_date = sign(("TC3" + self.secret_key).encode("utf-8"), date)
|
||
secret_service = sign(secret_date, "dnspod")
|
||
secret_signing = sign(secret_service, "tc3_request")
|
||
signature = hmac.new(secret_signing, string_to_sign.encode("utf-8"), hashlib.sha256).hexdigest()
|
||
# ************* 步骤 4：拼接 Authorization *************
|
||
authorization = (self.algorithm + " " +
|
||
"Credential=" + self.secret_id + "/" + credential_scope + ", " +
|
||
"SignedHeaders=" + signed_headers + ", " +
|
||
"Signature=" + signature)
|
||
headers = {
|
||
"Authorization": authorization,
|
||
"Content-Type": "application/json; charset=utf-8",
|
||
"Host": self.endpoint,
|
||
"X-TC-Action": action,
|
||
"X-TC-Timestamp": str(timestamp),
|
||
"X-TC-Version": self.version
|
||
}
|
||
if region:
|
||
headers["X-TC-Region"] = region
|
||
if token:
|
||
headers["X-TC-Token"] = token
|
||
return headers
|
||
|
||
def create_dns_record(self, get):
|
||
domain_name = get.domain_name
|
||
domain_dns_value = get.domain_dns_value
|
||
record_type = 'TXT'
|
||
if 'record_type' in get:
|
||
record_type = get.record_type
|
||
record_line = '默认'
|
||
if 'record_line' in get:
|
||
record_line = get.record_line
|
||
|
||
domain_name, sub_domain, _ = self.extract_zone(domain_name)
|
||
|
||
try:
|
||
params = json.dumps({
|
||
"Domain": domain_name,
|
||
"SubDomain": sub_domain,
|
||
"RecordType": record_type,
|
||
"RecordLine": record_line,
|
||
"Value": domain_dns_value,
|
||
})
|
||
headers = self.get_headers(get.dns_id, "CreateRecord", params)
|
||
res = requests.post(self.host, headers=headers, data=params).json()
|
||
if "Error" in res['Response']:
|
||
return public.returnMsg(False, res['Response']["Error"]["Message"])
|
||
return public.returnMsg(True, '添加成功')
|
||
except Exception as e:
|
||
return public.returnMsg(False, '添加失败，msg：{}'.format(e))
|
||
|
||
def delete_dns_record(self, get):
|
||
try:
|
||
domain_name = get.domain_name
|
||
RecordId = get.RecordId
|
||
root_domain, _, sub_domain = self.extract_zone(domain_name)
|
||
|
||
params = json.dumps({
|
||
"Domain": root_domain,
|
||
"RecordId": int(RecordId)
|
||
})
|
||
headers = self.get_headers(get.dns_id, "DeleteRecord", params)
|
||
res = requests.post(self.host, headers=headers, data=params).json()
|
||
if "Error" in res['Response']:
|
||
return public.returnMsg(False, res['Response']["Error"]["Message"])
|
||
return public.returnMsg(True, '删除成功!')
|
||
except Exception as e:
|
||
return public.returnMsg(False, e)
|
||
|
||
def get_dns_record(self, get):
|
||
|
||
domain_name, _, sub_domain = self.extract_zone(get.domain_name)
|
||
params = json.dumps({
|
||
"Domain": domain_name,
|
||
})
|
||
|
||
data = {}
|
||
try:
|
||
headers = self.get_headers(get.dns_id, "DescribeRecordList", params)
|
||
json_data = requests.post(self.host, headers=headers, data=params).json()
|
||
json_data = json_data['Response']
|
||
if "Error" in json_data:
|
||
if json_data["Error"]["Code"] == "ResourceNotFound.NoDataOfRecord":
|
||
data = {"info": {'record_total': 0}, "list": []}
|
||
else:
|
||
data = json_data
|
||
# if 'RecordList' in json_data:
|
||
# data['list'] = json_data['RecordList']
|
||
if 'RecordCountInfo' in json_data:
|
||
data['info'] = {
|
||
"record_total": json_data['RecordCountInfo']['SubdomainCount']
|
||
}
|
||
data["list"] = [
|
||
{
|
||
"RecordId": i["RecordId"],
|
||
"name": i["Name"] + "." + domain_name if i["Name"] != '@' else domain_name,
|
||
"value": i["Value"],
|
||
"line": i["Line"],
|
||
"ttl": i["TTL"],
|
||
"type": i["Type"],
|
||
"status": "启用" if i["Status"] == "ENABLE" else "暂停" if i["Status"] == "DISABLE" else i["Status"],
|
||
"mx": i.get("MX"),
|
||
"updated_on": i["UpdatedOn"],
|
||
"remark": i.get("Remark") or "",
|
||
}
|
||
for i in json_data["RecordList"]
|
||
]
|
||
|
||
except Exception as e:
|
||
pass
|
||
self.set_record_data({domain_name: data})
|
||
return data
|
||
|
||
def update_dns_record(self, get):
|
||
RecordId = get.RecordId
|
||
RecordLine = get.RecordLine
|
||
domain_name = get.domain_name
|
||
domain_dns_value = get.domain_dns_value
|
||
record_type = get.record_type
|
||
|
||
domain_name, sub_domain, _ = self.extract_zone(domain_name)
|
||
|
||
try:
|
||
params = json.dumps({
|
||
"Domain": domain_name,
|
||
"SubDomain": sub_domain,
|
||
"RecordType": record_type,
|
||
"RecordLine": RecordLine,
|
||
"Value": domain_dns_value,
|
||
"RecordId": int(RecordId)
|
||
})
|
||
headers = self.get_headers(get.dns_id, "ModifyRecord", params)
|
||
res = requests.post(self.host, headers=headers, data=params).json()
|
||
if "Error" in res['Response']:
|
||
return public.returnMsg(False, res['Response']["Error"]["Message"])
|
||
return public.returnMsg(True, '修改成功!')
|
||
except Exception as e:
|
||
return public.returnMsg(False, e)
|