iup-nft 0.1__py3-none-any.whl

iup/__init__.py

@@ -0,0 +1,5 @@
+import importlib_metadata
+
+metadata = importlib_metadata.metadata("iup")
+
+__version__ = metadata['version']

iup/config.py

@@ -0,0 +1,98 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import configparser
+import sys
+
+SECTION_COMMON = 'common'
+SECTIONS = [SECTION_COMMON]
+
+class Config:
+
+    def __init__(self, config_path=None):
+        self.host = '127.0.0.1'
+        self.port = 22
+        self.username = 'root'
+        self.password = '123456'
+        self.refresh = True
+
+        if config_path:
+            self.__update(config_path)
+
+    def update(self, host=None, port=None, username=None, password=None, refresh=None):
+        self.host = host
+        self.port = port
+        self.username = username
+        self.password = password
+        self.refresh = refresh
+
+    @property
+    def parser(self):
+        parser = configparser.ConfigParser()
+        variables = vars(self)
+        for section in SECTIONS:
+            parser[section] = {}
+        for k, v in variables.items():
+            if k.startswith('cm'):
+                pass
+            else:
+                parser[SECTION_COMMON].update({k: str(v)})
+        return parser
+
+    def save_to(self, path):
+        with open(path, 'w', encoding='utf-8') as f:
+            self.parser.write(f)
+
+    def __update(self, path):
+        variables = vars(self)
+        parser = configparser.ConfigParser()
+        parser.read(path, 'utf-8')
+        for section in parser.sections():
+            if section in SECTIONS:
+                pass
+            else:
+                # 不要parse 无关的section
+                break
+            for option in parser.options(section):
+                value = parser.get(section, option)
+                key = option
+                if not self.check_key_valid(key):
+                    break
+                original_value = variables.get(key)
+                type_expr = None
+                if original_value:
+                    type_of_original_value = type(original_value)
+                    if type_of_original_value == str:
+                        type_expr = 'str'
+                if type_expr:
+                    expr = "self.{} = '{}'".format(key, value)
+                else:
+                    expr = 'self.{} = {}'.format(key, value)
+                # print(expr)
+                exec(expr)
+
+    def check_key_valid(self, key):
+        variables = vars(self)
+        for k, _ in variables.items():
+            if k == key:
+                return True
+        return False
+
+    def __str__(self) -> str:
+        variables = vars(self)
+        info = ''
+        for k, v in variables.items():
+            temp_str = f"{k} = {v}\n"
+            type_of_v = type(v)
+            if type_of_v == str:
+                temp_str = f"{k} = '{v}'\n"
+            info += temp_str
+        return info[:-1]
+
+if __name__ == '__main__':
+    path = None
+    if len(sys.argv) > 1:
+        path = sys.argv[1]
+    print('config path:', path)
+    test_config = Config(path)
+    print(test_config)

iup/main.py

@@ -0,0 +1,170 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+import argparse
+from os import path
+import re
+import paramiko
+import iup
+from iup.config import Config
+import ipaddress
+
+
+HOSTNAME_PATTERN = r'^([a-zA-Z0-9]([a-zA-Z0-9\-]{0,61}[a-zA-Z0-9])?\.)+[a-zA-Z]{2,}$'
+CMD_TEMPLATE = 'iup {}'
+CMD_DNSMASQ = '/etc/init.d/dnsmasq restart'
+CMD_CHNDNS = '/etc/init.d/chinadns-ng restart'
+MAGIC_STR = "update rule to"
+MAGIC_STR2 = "update blk"
+
+def is_domain_name(string: str):
+    if re.match(HOSTNAME_PATTERN, string):
+        return True
+    else:
+        return False
+
+def is_ip_address(ip_str: str):
+    try:
+        ip = ipaddress.ip_address(ip_str)
+        return ip is not None
+    except ValueError:
+        return False
+
+def parse_host_name(hostname_str: str):
+    if is_domain_name(hostname_str) or is_ip_address(hostname_str):
+        return hostname_str
+    return None
+
+def parse_port(port: str):
+    if port.isdigit():
+        port_num = int(port)
+        if port_num > 1 and port_num < 65535:
+            return port
+    return None
+
+
+def parse_yes_or_no(msg: str):
+    msg = msg.upper()
+    if msg in ['Y', 'YES', 'T', 'TRUE']:
+        return True
+    if msg in ['N', 'NO', 'F', 'FALSE']:
+        return False
+    return None
+
+
+def read_hostname_from_file(path: str):
+    hostnames = []
+    with open(path, 'r') as f:
+        for line in f:
+            host_name_str = line.strip()
+            if is_domain_name(host_name_str):
+                hostnames.append(host_name_str)
+    hostnames = list(dict.fromkeys(hostnames))
+    return hostnames
+
+def load_and_overlay_config() -> Config:
+    parser = argparse.ArgumentParser(description='参数说明')
+    parser.add_argument("-nr", "--no_refresh", action='store_true', default=False, help="是否刷新")
+    parser.add_argument("sources", nargs='+', help='域名或者域名文件路径')
+    parser.add_argument(
+        "--version",
+        action="version",
+        version=f"%(prog)s version {iup.__version__} installed in {path.dirname(__file__)}",
+    )
+    args = parser.parse_args()
+    config_path = path.join(path.dirname(__file__), 'config.ini')
+    if not path.exists(config_path):
+        gen_config()
+    print('当前配置文件路径为:', config_path)
+    config = Config(config_path)
+    if args.no_refresh:
+        config.refresh = False
+    print('是否重启相关服务:', config.refresh)
+    return args.sources, config
+
+def update_rule_by_ssh(hostnames: list, config: Config):
+    need_restart_dns = False
+    need_restart_cdns = False
+    ssh = paramiko.SSHClient()
+    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+    ssh.connect(config.host, port=config.port, username=config.username, password=config.password)
+    for hostname in hostnames:
+        cmd = CMD_TEMPLATE.format(hostname)
+        _, stdout, stderr = ssh.exec_command(cmd)
+        msg = stdout.readlines()
+        err = stderr.readlines()
+        if MAGIC_STR in str(msg):
+            need_restart_dns = True
+        if MAGIC_STR2 in str(msg):
+            need_restart_cdns = True
+        print('msg', msg) if msg else print()
+        print('err', err) if err else print()
+    if config.refresh:
+        if need_restart_dns:
+            print('reboot dnsmasq')
+            _, stdout, stderr = ssh.exec_command(CMD_DNSMASQ)
+            msg = stdout.readlines()
+            err = stderr.readlines()
+            print('msg', msg) if msg else print()
+            print('err', err) if err else print()
+        if need_restart_cdns:
+            print('reboot chinadns')
+            _, stdout, stderr = ssh.exec_command(CMD_CHNDNS)
+            msg = stdout.readlines()
+            err = stderr.readlines()
+            print('msg', msg) if msg else print()
+            print('err', err) if err else print()
+    ssh.close()
+
+def gen_hostnames_by_sources(sources):
+    hostnames = []
+    for source in sources:
+        if not path.exists(source):
+            if is_domain_name(source):
+                hostnames.append(source)
+        elif path.isfile(source):
+            hostnames += read_hostname_from_file(source)
+    return list(dict.fromkeys(hostnames))
+
+def main():
+    sources, config = load_and_overlay_config()
+    print('sources:', sources)
+    hostnames = gen_hostnames_by_sources(sources)
+    if hostnames:
+        update_rule_by_ssh(hostnames, config)
+    else:
+        print('没有读取到有效域名，请检查！！！')
+
+def read_info_from_input(prompt, parser=None):
+    while True:
+        info = input(prompt)
+        if parser:
+            info = parser(info)
+            if not info is None:
+                break
+            else:
+                print('输入有误，请重新输入！')
+                continue
+        break
+    return info
+
+def gen_config():
+    print('配置向导，请按提示输入信息...')
+    host = read_info_from_input('请输入有效主机名:\n', parse_host_name)
+    port = int(read_info_from_input('请输入有效端口:\n', parse_port))
+    username = read_info_from_input('请输入openwrt用户名:\n')
+    password = read_info_from_input('请输入openwrt密码:\n')
+    refresh = read_info_from_input('是否重启相关服务(dnsmasq、chinadns),Y(es)/N(o)/T(rue)/F(alse):\n', parse_yes_or_no)
+    config = Config()
+    config.update(host, port, username, password, refresh)
+    config_path = path.join(path.dirname(__file__), 'config.ini')
+    config.save_to(config_path)
+    print('已生成配置文件写入到{}'.format(config_path))
+
+
+if __name__ == '__main__':
+    gen_config()
+
+
+
+
+

iup_nft-0.1.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.1
+Name: iup-nft
+Version: 0.1
+Summary: ipset updater for openwrt
+Author: nobitaqaq
+Author-email: xiaoleigs@gmail.com
+Keywords: ipset,openwrt
+Requires-Python: >=3.8
+Requires-Dist: paramiko
+

iup_nft-0.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+iup/__init__.py,sha256=WXPOI9BwESIVzJ-XhG7bIFhsRg98u_v5gp16nY0bcbI,108
+iup/config.py,sha256=Ecg9m83D_Cp_inYrltt--oPcupVvCw9oo8m5ukKyWLA,2862
+iup/main.py,sha256=j7-6cFrI6_RFkWR-Y4_LChs3fBVb1JGKQ8aW_w_9kMk,5508
+iup_nft-0.1.dist-info/METADATA,sha256=SiGliRq46XFLR0k3q-mV62-0tnU8QI-p9vke0ftFqVw,208
+iup_nft-0.1.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
+iup_nft-0.1.dist-info/entry_points.txt,sha256=ER1eUBmB6bXwudHBT51NoOtDSlnUgBi4T6J9_kA4pwc,68
+iup_nft-0.1.dist-info/top_level.txt,sha256=abec3YOzJRti4IxzOLIn-uDOPTtYUE-Uv65cAv1v8JY,4
+iup_nft-0.1.dist-info/RECORD,,

iup_nft-0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (73.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

iup_nft-0.1.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+iup = iup.main:main
+iup-cfg = iup.main:gen_config

iup_nft-0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+iup