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

ddns/util/config.py

@@ -1,317 +0,0 @@
-# -*- coding:utf-8 -*-
-from argparse import Action, ArgumentParser, Namespace, RawTextHelpFormatter
-from json import load as loadjson, dump as dumpjson
-from os import stat, environ, path
-from logging import critical, error, getLevelName
-from ast import literal_eval
-
-import platform
-import sys
-
-
-__cli_args = Namespace()
-__config = {}  # type: dict
-log_levels = [
-    "CRITICAL",  # 50
-    "ERROR",  # 40
-    "WARNING",  # 30
-    "INFO",  # 20
-    "DEBUG",  # 10
-    "NOTSET",  # 0
-]
-
-# 支持数组的参数列表
-ARRAY_PARAMS = ["index4", "index6", "ipv4", "ipv6", "proxy"]
-# 简单数组，支持’,’, ‘;’ 分隔的参数列表
-SIMPLE_ARRAY_PARAMS = ["ipv4", "ipv6", "proxy"]
-
-
-def str2bool(v):
-    """
-    parse string to boolean
-    """
-    if isinstance(v, bool):
-        return v
-    if v.lower() in ("yes", "true", "t", "y", "1"):
-        return True
-    elif v.lower() in ("no", "false", "f", "n", "0"):
-        return False
-    else:
-        return v
-
-
-def log_level(value):
-    """
-    parse string to log level
-    or getattr(logging, value.upper())
-    """
-    return getLevelName(value.upper())
-
-
-def parse_array_string(value, enable_simple_split):
-    """
-    解析数组字符串
-    仅当 trim 之后以 '[' 开头以 ']' 结尾时，才尝试使用 ast.literal_eval 解析
-    默认返回原始字符串
-    """
-    if not isinstance(value, str):
-        return value
-
-    trimmed = value.strip()
-    if trimmed.startswith("[") and trimmed.endswith("]"):
-        try:
-            # 尝试使用 ast.literal_eval 解析数组
-            parsed_value = literal_eval(trimmed)
-            # 确保解析结果是列表或元组
-            if isinstance(parsed_value, (list, tuple)):
-                return list(parsed_value)
-        except (ValueError, SyntaxError) as e:
-            # 解析失败时返回原始字符串
-            error("Failed to parse array string: %s. Exception: %s", value, e)
-    elif enable_simple_split:
-        # 尝试使用逗号或分号分隔符解析
-        sep = None
-        if ',' in trimmed:
-            sep = ','
-        elif ';' in trimmed:
-            sep = ';'
-        if sep:
-            return [item.strip() for item in trimmed.split(sep) if item.strip()]
-    return value
-
-
-def get_system_info_str():
-    system = platform.system()
-    release = platform.release()
-    machine = platform.machine()
-    arch = platform.architecture()[0]  # '64bit' or '32bit'
-
-    return "{}-{} {} ({})".format(system, release, machine, arch)
-
-
-def get_python_info_str():
-    version = platform.python_version()
-    branch, py_build_date = platform.python_build()
-    return "Python-{} {} ({})".format(version, branch, py_build_date)
-
-
-def init_config(description, doc, version, date):
-    """
-    配置
-    """
-    global __cli_args
-    parser = ArgumentParser(
-        description=description, epilog=doc, formatter_class=RawTextHelpFormatter
-    )
-    sysinfo = get_system_info_str()
-    pyinfo = get_python_info_str()
-    version_str = "v{} ({})\n{}\n{}".format(version, date, pyinfo, sysinfo)
-    parser.add_argument("-v", "--version", action="version", version=version_str)
-    parser.add_argument(
-        "-c", "--config", metavar="FILE", help="load config file [配置文件路径]"
-    )
-    parser.add_argument(
-        "--debug",
-        action="store_true",
-        help="debug mode [调试模式等效 --log.level=DEBUG]",
-    )
-
-    # 参数定义
-    parser.add_argument(
-        "--dns",
-        help="DNS provider [DNS服务提供商]",
-        choices=[
-            "alidns",
-            "cloudflare",
-            "dnscom",
-            "dnspod",
-            "dnspod_com",
-            "he",
-            "huaweidns",
-            "callback",
-        ],
-    )
-    parser.add_argument("--id", help="API ID or email [对应账号ID或邮箱]")
-    parser.add_argument("--token", help="API token or key [授权凭证或密钥]")
-    parser.add_argument(
-        "--index4",
-        nargs="*",
-        action=ExtendAction,
-        metavar="RULE",
-        help="IPv4 rules [获取IPv4方式, 多次可配置多规则]",
-    )
-    parser.add_argument(
-        "--index6",
-        nargs="*",
-        action=ExtendAction,
-        metavar="RULE",
-        help="IPv6 rules [获取IPv6方式, 多次可配置多规则]",
-    )
-    parser.add_argument(
-        "--ipv4",
-        nargs="*",
-        action=ExtendAction,
-        metavar="DOMAIN",
-        help="IPv4 domains [IPv4域名列表, 可配置多个域名]",
-    )
-    parser.add_argument(
-        "--ipv6",
-        nargs="*",
-        action=ExtendAction,
-        metavar="DOMAIN",
-        help="IPv6 domains [IPv6域名列表, 可配置多个域名]",
-    )
-    parser.add_argument("--ttl", type=int, help="DNS TTL(s) [设置域名解析过期时间]")
-    parser.add_argument(
-        "--proxy",
-        nargs="*",
-        action=ExtendAction,
-        help="HTTP proxy [设置http代理，可配多个代理连接]",
-    )
-    parser.add_argument(
-        "--cache",
-        type=str2bool,
-        nargs="?",
-        const=True,
-        help="set cache [启用缓存开关，或传入保存路径]",
-    )
-    parser.add_argument(
-        "--no-cache",
-        dest="cache",
-        action="store_const",
-        const=False,
-        help="disable cache [关闭缓存等效 --cache=false]",
-    )
-    parser.add_argument(
-        "--log.file", metavar="FILE", help="log file [日志文件，默认标准输出]"
-    )
-    parser.add_argument("--log.level", type=log_level, metavar="|".join(log_levels))
-    parser.add_argument(
-        "--log.format", metavar="FORMAT", help="log format [设置日志打印格式]"
-    )
-    parser.add_argument(
-        "--log.datefmt", metavar="FORMAT", help="date format [日志时间打印格式]"
-    )
-
-    __cli_args = parser.parse_args()
-    if __cli_args.debug:
-        # 如果启用调试模式，则设置日志级别为 DEBUG
-        setattr(__cli_args, "log.level", log_level("DEBUG"))
-
-    is_configfile_required = not get_config("token") and not get_config("id")
-    config_file = get_config("config")
-    if not config_file:
-        # 未指定配置文件且需要读取文件时，依次查找
-        cfgs = [
-            path.abspath("config.json"),
-            path.expanduser("~/.ddns/config.json"),
-            "/etc/ddns/config.json",
-        ]
-        config_file = next((cfg for cfg in cfgs if path.isfile(cfg)), cfgs[0])
-
-    if path.isfile(config_file):
-        __load_config(config_file)
-        __cli_args.config = config_file
-    elif is_configfile_required:
-        error("Config file is required, but not found: %s", config_file)
-        # 如果需要配置文件但没有指定，则自动生成
-        if generate_config(config_file):
-            sys.stdout.write("Default configure file %s is generated.\n" % config_file)
-            sys.exit(1)
-        else:
-            sys.exit("fail to load config from file: %s\n" % config_file)
-
-
-def __load_config(config_path):
-    """
-    加载配置
-    """
-    global __config
-    try:
-        with open(config_path, "r") as configfile:
-            __config = loadjson(configfile)
-            __config["config_modified_time"] = stat(config_path).st_mtime
-            if "log" in __config:
-                if "level" in __config["log"] and __config["log"]["level"] is not None:
-                    __config["log.level"] = log_level(__config["log"]["level"])
-                if "file" in __config["log"]:
-                    __config["log.file"] = __config["log"]["file"]
-                if "format" in __config["log"]:
-                    __config["log.format"] = __config["log"]["format"]
-                if "datefmt" in __config["log"]:
-                    __config["log.datefmt"] = __config["log"]["datefmt"]
-            elif "log.level" in __config:
-                __config["log.level"] = log_level(__config["log.level"])
-    except Exception as e:
-        critical("Failed to load config file `%s`: %s", config_path, e)
-        raise
-        # 重新抛出异常
-
-
-def get_config(key, default=None):
-    """
-    读取配置
-    1. 命令行参数
-    2. 配置文件
-    3. 环境变量
-    """
-    if hasattr(__cli_args, key) and getattr(__cli_args, key) is not None:
-        return getattr(__cli_args, key)
-    if key in __config:
-        return __config.get(key)
-    # 检查环境变量
-    env_name = "DDNS_" + key.replace(".", "_")  # type:str
-    variations = [env_name, env_name.upper(), env_name.lower()]
-    value = next((environ.get(v) for v in variations if v in environ), None)
-
-    # 如果找到环境变量值且参数支持数组，尝试解析为数组
-    if value is not None and key in ARRAY_PARAMS:
-        return parse_array_string(value, key in SIMPLE_ARRAY_PARAMS)
-
-    return value if value is not None else default
-
-
-class ExtendAction(Action):
-    """
-    兼容 Python <3.8 的 extend action
-    """
-
-    def __call__(self, parser, namespace, values, option_string=None):
-        items = getattr(namespace, self.dest, None)
-        if items is None:
-            items = []
-        # values 可能是单个值或列表
-        if isinstance(values, list):
-            items.extend(values)
-        else:
-            items.append(values)
-        setattr(namespace, self.dest, items)
-
-
-def generate_config(config_path):
-    """
-    生成配置文件
-    """
-    configure = {
-        "$schema": "https://ddns.newfuture.cc/schema/v4.0.json",
-        "id": "YOUR ID or EMAIL for DNS Provider",
-        "token": "YOUR TOKEN or KEY for DNS Provider",
-        "dns": "dnspod",
-        "ipv4": ["newfuture.cc", "ddns.newfuture.cc"],
-        "ipv6": ["newfuture.cc", "ipv6.ddns.newfuture.cc"],
-        "index4": "default",
-        "index6": "default",
-        "ttl": None,
-        "proxy": None,
-        "log": {"level": "INFO"},
-    }
-    try:
-        with open(config_path, "w") as f:
-            dumpjson(configure, f, indent=2, sort_keys=True)
-            return True
-    except IOError:
-        critical("Cannot open config file to write: `%s`!", config_path)
-        return False
-    except Exception as e:
-        critical("Failed to write config file `%s`: %s", config_path, e)
-        return False

ddns/util/ip.py

@@ -1,96 +0,0 @@
-#!/usr/bin/env python
-# -*- coding:utf-8 -*-
-from re import compile
-from os import name as os_name, popen
-from socket import socket, getaddrinfo, gethostname, AF_INET, AF_INET6, SOCK_DGRAM
-from logging import debug, error
-try:  # python3
-    from urllib.request import urlopen, Request
-except ImportError:  # python2
-    from urllib2 import urlopen, Request
-
-# IPV4正则
-IPV4_REG = r'((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])'
-# IPV6正则
-# https://community.helpsystems.com/forums/intermapper/miscellaneous-topics/5acc4fcf-fa83-e511-80cf-0050568460e4
-IPV6_REG = r'((([0-9A-Fa-f]{1,4}:){7}([0-9A-Fa-f]{1,4}|:))|(([0-9A-Fa-f]{1,4}:){6}(:[0-9A-Fa-f]{1,4}|((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){5}(((:[0-9A-Fa-f]{1,4}){1,2})|:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){4}(((:[0-9A-Fa-f]{1,4}){1,3})|((:[0-9A-Fa-f]{1,4})?:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){3}(((:[0-9A-Fa-f]{1,4}){1,4})|((:[0-9A-Fa-f]{1,4}){0,2}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){2}(((:[0-9A-Fa-f]{1,4}){1,5})|((:[0-9A-Fa-f]{1,4}){0,3}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){1}(((:[0-9A-Fa-f]{1,4}){1,6})|((:[0-9A-Fa-f]{1,4}){0,4}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(:(((:[0-9A-Fa-f]{1,4}){1,7})|((:[0-9A-Fa-f]{1,4}){0,5}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:)))'  # noqa: E501
-
-
-def default_v4():  # 默认连接外网的ipv4
-    s = socket(AF_INET, SOCK_DGRAM)
-    s.connect(("1.1.1.1", 53))
-    ip = s.getsockname()[0]
-    s.close()
-    return ip
-
-
-def default_v6():  # 默认连接外网的ipv6
-    s = socket(AF_INET6, SOCK_DGRAM)
-    s.connect(('1:1:1:1:1:1:1:1', 8))
-    ip = s.getsockname()[0]
-    s.close()
-    return ip
-
-
-def local_v6(i=0):  # 本地ipv6地址
-    info = getaddrinfo(gethostname(), 0, AF_INET6)
-    debug(info)
-    return info[int(i)][4][0]
-
-
-def local_v4(i=0):  # 本地ipv4地址
-    info = getaddrinfo(gethostname(), 0, AF_INET)
-    debug(info)
-    return info[int(i)][-1][0]
-
-
-def _open(url, reg):
-    try:
-        debug("open: %s", url)
-        res = urlopen(
-            Request(url, headers={'User-Agent': 'Mozilla/5.0 ddns'}),  timeout=60
-        ).read().decode('utf8', 'ignore')
-        debug("response: %s",  res)
-        return compile(reg).search(res).group()
-    except Exception as e:
-        error(e)
-
-
-def public_v4(url="https://api-ipv4.ip.sb/ip", reg=IPV4_REG):  # 公网IPV4地址
-    return _open(url, reg)
-
-
-def public_v6(url="https://api-ipv6.ip.sb/ip", reg=IPV6_REG):  # 公网IPV6地址
-    return _open(url, reg)
-
-
-def _ip_regex_match(parrent_regex, match_regex):
-
-    ip_pattern = compile(parrent_regex)
-    matcher = compile(match_regex)
-
-    if os_name == 'nt':  # windows:
-        cmd = 'ipconfig'
-    else:
-        cmd = 'ip address || ifconfig 2>/dev/null'
-
-    for s in popen(cmd).readlines():
-        addr = ip_pattern.search(s)
-        if addr and matcher.match(addr.group(1)):
-            return addr.group(1)
-
-
-def regex_v4(reg):  # ipv4 正则提取
-    if os_name == 'nt':  # Windows: IPv4 xxx: 192.168.1.2
-        regex_str = r'IPv4 .*: ((?:\d{1,3}\.){3}\d{1,3})\W'
-    else:
-        regex_str = r'inet (?:addr\:)?((?:\d{1,3}\.){3}\d{1,3})[\s/]'
-    return _ip_regex_match(regex_str, reg)
-
-
-def regex_v6(reg):  # ipv6 正则提取
-    if os_name == 'nt':  # Windows: IPv4 xxx: ::1
-        regex_str = r'IPv6 .*: ([\:\dabcdef]*)?\W'
-    else:
-        regex_str = r'inet6 (?:addr\:\s*)?([\:\dabcdef]*)?[\s/%]'
-    return _ip_regex_match(regex_str, reg)