ddns 4.0.1__py2.py3-none-any.whl → 4.1.0__py2.py3-none-any.whl

ddns/provider/dnscom.py

@@ -1,180 +1,103 @@
 # coding=utf-8
 """
-DNSCOM API
-DNS.COM 接口解析操作库
-http://open.dns.com/
-@author: Bigjin
-@mailto: i@bigjin.com
+DNSCOM/51dns API 接口解析操作库
+www.51dns.com (原dns.com)
+@author: Bigjin<i@bigjin.com>, NewFuture
 """
 
 from hashlib import md5
-from json import loads as jsondecode
-from logging import debug, info, warning
-from time import mktime
-from datetime import datetime
+from time import time
 
-try:  # python 3
-    from http.client import HTTPSConnection
-    from urllib.parse import urlencode
-except ImportError:  # python 2
-    from httplib import HTTPSConnection
-    from urllib import urlencode
+from ._base import TYPE_FORM, BaseProvider, encode_params
 
-__author__ = 'Bigjin'
-# __all__ = ["request", "ID", "TOKEN", "PROXY"]
 
-
-class Config:
-    ID = "id"
-    TOKEN = "TOKEN"
-    PROXY = None  # 代理设置
-    TTL = None
-
-
-class API:
-    # API 配置
-    SITE = "www.dns.com"  # API endpoint
-    METHOD = "POST"  # 请求方法
-
-
-def signature(params):
-    """
-    计算签名,返回签名后的查询参数
-    """
-    params.update({
-        'apiKey': Config.ID,
-        'timestamp': mktime(datetime.now().timetuple()),
-    })
-    query = urlencode(sorted(params.items()))
-    debug(query)
-    sign = query
-    debug("signString: %s", sign)
-
-    sign = md5((sign + Config.TOKEN).encode('utf-8')).hexdigest()
-    params["hash"] = sign
-
-    return params
-
-
-def request(action, param=None, **params):
-    """
-    发送请求数据
-    """
-    if param:
-        params.update(param)
-    params = dict((k, params[k]) for k in params if params[k] is not None)
-    params = signature(params)
-    info("%s/api/%s/ : params:%s", API.SITE, action, params)
-
-    if Config.PROXY:
-        conn = HTTPSConnection(Config.PROXY)
-        conn.set_tunnel(API.SITE, 443)
-    else:
-        conn = HTTPSConnection(API.SITE)
-
-    conn.request(API.METHOD, '/api/' + action + '/', urlencode(params),
-                 {"Content-type": "application/x-www-form-urlencoded"})
-    response = conn.getresponse()
-    result = response.read().decode('utf8')
-    conn.close()
-
-    if response.status < 200 or response.status >= 300:
-        warning('%s : error[%d]:%s', action, response.status, result)
-        raise Exception(result)
-    else:
-        data = jsondecode(result)
-        debug('%s : result:%s', action, data)
-        if data.get('code') != 0:
-            raise Exception("api error:", data.get('message'))
-        data = data.get('data')
-        if data is None:
-            raise Exception('response data is none')
-        return data
-
-
-def get_domain_info(domain):
-    """
-    切割域名获取主域名和对应ID
-    """
-    if len(domain.split('.')) > 2:
-        domains = domain.split('.', 1)
-        sub = domains[0]
-        main = domains[1]
-    else:
-        sub = ''  # 接口有bug 不能传 @ * 作为主机头，但是如果为空，默认为 @
-        main = domain
-
-    res = request("domain/getsingle", domainID=main)
-    domain_id = res.get('domainID')
-    return sub, main, domain_id
-
-
-def get_records(domain, domain_id, **conditions):
-    """
-        获取记录ID
-        返回满足条件的所有记录[]
-        TODO 大于500翻页
+class DnscomProvider(BaseProvider):
     """
-    if not hasattr(get_records, "records"):
-        get_records.records = {}  # "静态变量"存储已查询过的id
-        get_records.keys = ("recordID", "record", "type", "viewID",
-                            "TTL", "state", "value")
-
-    if domain not in get_records.records:
-        get_records.records[domain] = {}
-        data = request("record/list",
-                       domainID=domain_id, pageSize=500)
-        if data.get('data'):
-            for record in data.get('data'):
-                get_records.records[domain][record["recordID"]] = {
-                    k: v for (k, v) in record.items() if k in get_records.keys}
-    records = {}
-    for (rid, record) in get_records.records[domain].items():
-        for (k, value) in conditions.items():
-            if record.get(k) != value:
-                break
-        else:  # for else push
-            records[rid] = record
-    return records
-
-
-def update_record(domain, value, record_type='A'):
-    """
-        更新记录
+    DNSCOM/51dns API Provider
+    https://www.51dns.com/document/api/index.html
     """
-    info(">>>>>%s(%s)", domain, record_type)
-    sub, main, domain_id = get_domain_info(domain)
 
-    records = get_records(main, domain_id, record=sub, type=record_type)
-    result = {}
-
-    if records:
-        for (rid, record) in records.items():
-            if record["value"] != value:
-                debug(sub, record)
-                res = request("record/modify", domainID=domain_id,
-                              recordID=rid, newvalue=value, newTTL=Config.TTL)
-                if res:
-                    # update records
-                    get_records.records[main][rid]["value"] = value
-                    result[rid] = res
-                else:
-                    result[rid] = "update fail!\n" + str(res)
-            else:
-                result[rid] = domain
-    else:
-        res = request("record/create", domainID=domain_id,
-                      value=value, host=sub, type=record_type, TTL=Config.TTL)
-        if res:
-            # update records INFO
-            rid = res.get('recordID')
-            get_records.records[main][rid] = {
-                'value': value,
-                "recordID": rid,
-                "record": sub,
-                "type": record_type
+    endpoint = "https://www.51dns.com"
+    content_type = TYPE_FORM
+
+    def _validate(self):
+        self.logger.warning(
+            "DNS.COM 缺少充分的真实环境测试，请及时在 GitHub Issues 中反馈: %s",
+            "https://github.com/NewFuture/DDNS/issues",
+        )
+        super(DnscomProvider, self)._validate()
+
+    def _signature(self, params):
+        """https://www.51dns.com/document/api/70/72.html"""
+        params = {k: v for k, v in params.items() if v is not None}
+        params.update(
+            {
+                "apiKey": self.id,
+                "timestamp": time(),  # 时间戳
             }
-            result = res
-        else:
-            result = domain + " created fail!"
-    return result
+        )
+        query = encode_params(params)
+        sign = md5((query + self.token).encode("utf-8")).hexdigest()
+        params["hash"] = sign
+        return params
+
+    def _request(self, action, **params):
+        params = self._signature(params)
+        data = self._http("POST", "/api/{}/".format(action), body=params)
+        if data is None or not isinstance(data, dict):
+            raise Exception("response data is none")
+        if data.get("code", 0) != 0:
+            raise Exception("api error: " + str(data.get("message")))
+        return data.get("data")
+
+    def _query_zone_id(self, domain):
+        """https://www.51dns.com/document/api/74/31.html"""
+        res = self._request("domain/getsingle", domainID=domain)
+        self.logger.debug("Queried domain: %s", res)
+        if res:
+            return res.get("domainID")
+        return None
+
+    def _query_record(self, zone_id, subdomain, main_domain, record_type, line, extra):
+        """https://www.51dns.com/document/api/4/47.html"""
+        records = self._request("record/list", domainID=zone_id, host=subdomain, pageSize=500)
+        records = records.get("data", []) if records else []
+        for record in records:
+            if (
+                record.get("record") == subdomain
+                and record.get("type") == record_type
+                and (line is None or record.get("viewID") == line)
+            ):
+                return record
+        return None
+
+    def _create_record(self, zone_id, subdomain, main_domain, value, record_type, ttl, line, extra):
+        """https://www.51dns.com/document/api/4/12.html"""
+        extra["remark"] = extra.get("remark", self.remark)
+        res = self._request(
+            "record/create",
+            domainID=zone_id,
+            value=value,
+            host=subdomain,
+            type=record_type,
+            TTL=ttl,
+            viewID=line,
+            **extra
+        )  # fmt: skip
+        if res and res.get("recordID"):
+            self.logger.info("Record created: %s", res)
+            return True
+        self.logger.error("Failed to create record: %s", res)
+        return False
+
+    def _update_record(self, zone_id, old_record, value, record_type, ttl, line, extra):
+        """https://www.51dns.com/document/api/4/45.html"""
+        extra["remark"] = extra.get("remark", self.remark)
+        res = self._request(
+            "record/modify", domainID=zone_id, recordID=old_record.get("recordID"), newvalue=value, newTTL=ttl
+        )
+        if res:
+            self.logger.info("Record updated: %s", res)
+            return True
+        self.logger.error("Failed to update record: %s", res)
+        return False

ddns/provider/dnspod.py

@@ -1,186 +1,114 @@
 # coding=utf-8
 """
 DNSPOD API
-DNSPOD 接口解析操作库
-http://www.dnspod.cn/docs/domains.html
-@author: New Future
+@doc: https://docs.dnspod.cn/api/
+@author: NewFuture
 """
 
-from json import loads as jsondecode
-from logging import debug, info, warning
-from os import environ
+from ._base import BaseProvider, TYPE_FORM
 
-try:  # python 3
-    from http.client import HTTPSConnection
-    from urllib.parse import urlencode
-except ImportError:  # python 2
-    from httplib import HTTPSConnection
-    from urllib import urlencode
 
-__author__ = 'New Future'
-
-
-class Config:
-    ID = "token id"
-    TOKEN = "token key"
-    PROXY = None  # 代理设置
-    TTL = None
-
-
-class API:
-    # API 配置
-    SITE = "dnsapi.cn"  # API endpoint
-    METHOD = "POST"  # 请求方法
-    TOKEN_PARAM = "login_token"  # token参数
-    DEFAULT = "默认"  # 默认线路名
-    LENGTH = "length"  # 添加参数
-
-
-def request(action, param=None, **params):
+class DnspodProvider(BaseProvider):
     """
-    发送请求数据
+    DNSPOD API
+    DNSPOD 接口解析操作库
     """
-    if param:
-        params.update(param)
-    params = dict((k, params[k]) for k in params if params[k] is not None)
-    params.update({API.TOKEN_PARAM: '***', 'format': 'json'})
-    info("%s/%s : %s", API.SITE, action, params)
-    params[API.TOKEN_PARAM] = "%s,%s" % (Config.ID, Config.TOKEN)
-    params[API.LENGTH] = "3000"  # 添加参数
-    if Config.PROXY:
-        conn = HTTPSConnection(Config.PROXY)
-        conn.set_tunnel(API.SITE, 443)
-    else:
-        conn = HTTPSConnection(API.SITE)
-
-    conn.request(API.METHOD, '/' + action, urlencode(params), {
-        "Content-type": "application/x-www-form-urlencoded",
-        "User-Agent": "DDNS/%s (ddns@newfuture.cc)" % environ.get("DDNS_VERSION", "1.0.0")
-    })
-    response = conn.getresponse()
-    res = response.read().decode('utf8')
-    conn.close()
 
-    if response.status < 200 or response.status >= 300:
-        warning('%s : error[%d]:%s', action, response.status, res)
-        raise Exception(res)
-    else:
-        data = jsondecode(res)
-        debug('%s : result:%s', action, data)
-        if not data:
-            raise Exception("empty response")
-        elif data.get("status", {}).get("code") == "1":
+    endpoint = "https://dnsapi.cn"
+    content_type = TYPE_FORM
+
+    DefaultLine = "默认"
+
+    def _request(self, action, extra=None, **params):
+        # type: (str, dict | None, **(str | int | bytes | bool | None)) -> dict
+        """
+        发送请求数据
+
+        Send request to DNSPod API.
+        Args:
+            action (str): API 动作/Action
+            extra (dict|None): 额外参数/Extra params
+            params (dict): 其它参数/Other params
+        Returns:
+            dict: 响应数据/Response data
+        """
+        # 过滤掉None参数
+        if extra:
+            params.update(extra)
+        params = {k: v for k, v in params.items() if v is not None}
+        params.update({"login_token": "{0},{1}".format(self.id, self.token), "format": "json"})
+        data = self._http("POST", "/" + action, body=params)
+        if data and data.get("status", {}).get("code") == "1":  # 请求成功
+            return data
+        else:  # 请求失败
+            error_msg = "Unknown error"
+            if data and isinstance(data, dict):
+                error_msg = data.get("status", {}).get("message", "Unknown error")
+            self.logger.warning("DNSPod API error: %s", error_msg)
             return data
-        else:
-            raise Exception(data.get('status', {}))
-
-
-def get_domain_info(domain):
-    """
-    切割域名获取主域名和对应ID
-    """
-    domain_split = domain.split('.')
-    sub, did = None, None
-    main = domain_split.pop()
-    while domain_split:  # 通过API判断,最后两个，三个递增
-        main = domain_split.pop() + '.' + main
-        did = get_domain_id(main)
-        if did:
-            sub = ".".join(domain_split) or '@'
-            # root domain根域名https://github.com/NewFuture/DDNS/issues/9
-            break
-    info('domain_id: %s, sub: %s', did, sub)
-    return did, sub
-
-
-def get_domain_id(domain):
-    """
-        获取域名ID
-        http://www.dnspod.cn/docs/domains.html#domain-info
-    """
-    if not hasattr(get_domain_id, "domain_list"):
-        get_domain_id.domain_list = {}  # "静态变量"存储已查询过的id
-
-    if domain in get_domain_id.domain_list:
-        # 如果已经存在直接返回防止再次请求
-        return get_domain_id.domain_list[domain]
-    else:
-        try:
-            d_info = request('Domain.Info', domain=domain)
-        except Exception as e:
-            info("get_domain_id(%s) error: %s", domain, e)
-            return
-        did = d_info.get("domain", {}).get("id")
-        if did:
-            get_domain_id.domain_list[domain] = did
-            return did
-
-
-def get_records(did, **conditions):
-    """
-        获取记录ID
-        返回满足条件的所有记录[]
-        TODO 大于3000翻页
-        http://www.dnspod.cn/docs/records.html#record-list
-    """
-    if not hasattr(get_records, "records"):
-        get_records.records = {}  # "静态变量"存储已查询过的id
-        get_records.keys = ("id", "name", "type", "line",
-                            "line_id", "enabled", "mx", "value")
-
-    if did not in get_records.records:
-        get_records.records[did] = {}
-        data = request('Record.List', domain_id=did)
-        if data:
-            for record in data.get('records'):
-                get_records.records[did][record["id"]] = {
-                    k: v for (k, v) in record.items() if k in get_records.keys}
-
-    records = {}
-    for (did, record) in get_records.records[did].items():
-        for (k, value) in conditions.items():
-            if record.get(k) != value:
-                break
-        else:  # for else push
-            records[did] = record
-    return records
-
-
-def update_record(domain, value, record_type="A"):
-    """
-    更新记录
-    """
-    info(">>>>>%s(%s)", domain, record_type)
-    domainid, sub = get_domain_info(domain)
-    if not domainid:
-        raise Exception("invalid domain: [ %s ] " % domain)
 
-    records = get_records(domainid, name=sub, type=record_type)
-    result = {}
-    if records:  # update
-        # http://www.dnspod.cn/docs/records.html#record-modify
-        for (did, record) in records.items():
-            if record["value"] != value:
-                debug(sub, record)
-                res = request('Record.Modify', record_id=did, record_line=record["line"].replace("Default", "default").encode(
-                    "utf-8"), value=value, sub_domain=sub, domain_id=domainid, record_type=record_type, ttl=Config.TTL)
-                if res:
-                    get_records.records[domainid][did]["value"] = value
-                    result[did] = res.get("record")
-                else:
-                    result[did] = "update fail!\n" + str(res)
-            else:
-                result[did] = domain
-    else:  # create
-        # http://www.dnspod.cn/docs/records.html#record-create
-        res = request("Record.Create", domain_id=domainid, value=value,
-                      sub_domain=sub, record_type=record_type, record_line=API.DEFAULT, ttl=Config.TTL)
-        if res:
-            did = res.get("record")["id"]
-            get_records.records[domainid][did] = res.get("record")
-            get_records.records[domainid][did].update(
-                value=value, sub_domain=sub, record_type=record_type)
-            result = res.get("record")
-        else:
-            result = domain + " created fail!"
-    return result
+    def _create_record(self, zone_id, subdomain, main_domain, value, record_type, ttl, line, extra):
+        # type: (str, str, str, str, str, int | str | None, str | None, dict) -> bool
+        """https://docs.dnspod.cn/api/add-record/"""
+        res = self._request(
+            "Record.Create",
+            extra=extra,
+            domain_id=zone_id,
+            sub_domain=subdomain,
+            value=value,
+            record_type=record_type,
+            record_line=line or self.DefaultLine,
+            ttl=ttl,
+        )
+        record = res and res.get("record")
+        if record:  # 记录创建成功
+            self.logger.info("Record created: %s", record)
+            return True
+        else:  # 记录创建失败
+            self.logger.error("Failed to create record: %s", res)
+        return False
+
+    def _update_record(self, zone_id, old_record, value, record_type, ttl, line, extra):
+        # type: (str, dict, str, str, int | str | None, str | None, dict) -> bool
+        """https://docs.dnspod.cn/api/modify-records/"""
+        record_line = (line or old_record.get("line") or self.DefaultLine).replace("Default", "default")
+        res = self._request(
+            "Record.Modify",
+            domain_id=zone_id,
+            record_id=old_record.get("id"),
+            sub_domain=old_record.get("name"),
+            record_type=record_type,
+            value=value,
+            record_line=record_line,
+            extra=extra,
+        )
+        record = res and res.get("record")
+        if record:  # 记录更新成功
+            self.logger.debug("Record updated: %s", record)
+            return True
+        else:  # 记录更新失败
+            self.logger.error("Failed to update record: %s", res)
+            return False
+
+    def _query_zone_id(self, domain):
+        # type: (str) -> str | None
+        """查询域名信息 https://docs.dnspod.cn/api/domain-info/"""
+        res = self._request("Domain.Info", domain=domain)
+        if res and isinstance(res, dict):
+            return res.get("domain", {}).get("id")
+        return None
+
+    def _query_record(self, zone_id, subdomain, main_domain, record_type, line, extra):
+        # type: (str, str, str, str, str | None, dict) -> dict | None
+        """查询记录 list 然后逐个查找 https://docs.dnspod.cn/api/record-list/"""
+        res = self._request("Record.List", domain_id=zone_id, sub_domain=subdomain, record_type=record_type, line=line)
+        # length="3000"
+        records = res.get("records", [])
+        n = len(records)
+        if not n:
+            self.logger.warning("No record found for [%s] %s<%s>(line: %s)", zone_id, subdomain, record_type, line)
+            return None
+        if n > 1:
+            self.logger.warning("%d records found for %s<%s>(%s):\n %s", n, subdomain, record_type, line, records)
+            return next((r for r in records if r.get("name") == subdomain), None)
+        return records[0]

ddns/provider/dnspod_com.py

@@ -1,15 +1,18 @@
 # coding=utf-8
 """
-DNSPOD API
-DNSPOD 接口解析操作库
+DNSPOD Global (国际版) API
 http://www.dnspod.com/docs/domains.html
-@author: New Future
+@author: NewFuture
 """
 
-from .dnspod import *  # noqa: F403
+from .dnspod import DnspodProvider  # noqa: F401
 
-API.SITE = "api.dnspod.com"  # noqa: F405
-API.DEFAULT = "default"  # noqa: F405
 
-# https://github.com/NewFuture/DDNS/issues/286
-# API.TOKEN_PARAM = "user_token"  # noqa: F405
+class DnspodComProvider(DnspodProvider):
+    """
+    DNSPOD.com Provider (国际版)
+    This class extends the DnspodProvider to use the global DNSPOD API.
+    """
+
+    endpoint = "https://api.dnspod.com"
+    DefaultLine = "default"

ddns/provider/edgeone.py

@@ -0,0 +1,83 @@
+# coding=utf-8
+"""
+Tencent Cloud EdgeOne API
+腾讯云 EdgeOne (边缘安全速平台) API
+API Documentation: https://cloud.tencent.com/document/api/1552/80731
+@author: NewFuture
+"""
+
+from ddns.provider._base import join_domain
+from .tencentcloud import TencentCloudProvider
+
+
+class EdgeOneProvider(TencentCloudProvider):
+    """
+    腾讯云 EdgeOne API 提供商
+    Tencent Cloud EdgeOne API Provider
+    """
+
+    endpoint = "https://teo.tencentcloudapi.com"
+    # 腾讯云 EdgeOne API 配置
+    service = "teo"
+    version_date = "2022-09-01"
+
+    def _query_zone_id(self, domain):
+        # type: (str) -> str | None
+        """查询域名的加速域名信息获取 ZoneId https://cloud.tencent.com/document/api/1552/80713"""
+        # 首先尝试直接查找域名
+        filters = [{"Name": "zone-name", "Values": [domain], "Fuzzy": False}]  # type: Any
+        response = self._request("DescribeZones", Filters=filters)
+
+        if response and "Zones" in response:
+            for zone in response.get("Zones", []):
+                if zone.get("ZoneName") == domain:
+                    zone_id = zone.get("ZoneId")
+                    if zone_id:
+                        self.logger.debug("Found acceleration domain %s with Zone ID: %s", domain, zone_id)
+                        return zone_id
+
+        self.logger.debug("Acceleration domain not found for: %s", domain)
+        return None
+
+    def _query_record(self, zone_id, subdomain, main_domain, record_type, line, extra):
+        # type: (str, str, str, str, str | None, dict) -> dict | None
+        """查询加速域名信息 https://cloud.tencent.com/document/api/1552/86336"""
+        domain = join_domain(subdomain, main_domain)
+        filters = [{"Name": "domain-name", "Values": [domain], "Fuzzy": False}]  # type: Any
+        response = self._request("DescribeAccelerationDomains", ZoneId=zone_id, Filters=filters)
+
+        if response and "AccelerationDomains" in response:
+            for domain_info in response.get("AccelerationDomains", []):
+                if domain_info.get("DomainName") == domain:
+                    self.logger.debug("Found acceleration domain: %s", domain_info)
+                    return domain_info
+
+        self.logger.warning("No acceleration domain found for: %s, response: %s", domain, response)
+        return None
+
+    def _create_record(self, zone_id, subdomain, main_domain, value, record_type, ttl, line, extra):
+        # type: (str, str, str, str, str, int, str | None, dict) -> bool
+        """创建新的加速域名记录 https://cloud.tencent.com/document/api/1552/86338"""
+        domain = join_domain(subdomain, main_domain)
+        origin = {"OriginType": "IP_DOMAIN", "Origin": value}  # type: Any
+        res = self._request("CreateAccelerationDomain", ZoneId=zone_id, DomainName=domain, OriginInfo=origin, **extra)
+        if res:
+            self.logger.info("Acceleration domain created (%s)", res.get("RequestId"))
+            return True
+
+        self.logger.error("Failed to create acceleration domain, response: %s", res)
+        return False
+
+    def _update_record(self, zone_id, old_record, value, record_type, ttl, line, extra):
+        """更新加速域名的源站 IP 地址 https://cloud.tencent.com/document/api/1552/86335"""
+        domain = old_record.get("DomainName")
+        # 构建源站信息
+        backup = old_record.get("OriginDetail", {}).get("BackupOrigin", "")
+        origin = {"OriginType": "IP_DOMAIN", "Origin": value, "BackupOrigin": backup}  # type: Any
+        response = self._request("ModifyAccelerationDomain", ZoneId=zone_id, DomainName=domain, OriginInfo=origin)
+
+        if response:
+            self.logger.info("Acceleration domain updated (%s)", response.get("RequestId"))
+            return True
+        self.logger.error("Failed to update acceleration domain origin, response: %s", response)
+        return False