Gecko [ www.jdcpatala.org ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	2.77 KB	-rw-r--r--
access_restriction.py	23.34 KB	-rw-r--r--
config_mgr.py	6.28 KB	-rw-r--r--
default_site.py	4.46 KB	-rw-r--r--
dir_tool.py	9.89 KB	-rw-r--r--
dns_api.py	62.05 KB	-rw-r--r--
domain_tool.py	12.46 KB	-rw-r--r--
ip_restrict.py	11.43 KB	-rw-r--r--
limit_net.py	10.47 KB	-rw-r--r--
logmanager.py	34.68 KB	-rw-r--r--
proxy.py	24.07 KB	-rw-r--r--
redirect.py	28.54 KB	-rw-r--r--
referer.py	14.65 KB	-rw-r--r--
ssl.py	51.7 KB	-rw-r--r--
util.py	5.19 KB	-rw-r--r--

Code Editor : config_mgr.py

import os
import time
from hashlib import md5
from typing import Optional
from .util import service_reload, check_server_config, write_file, read_file

# 支持读取配置文件
# 保存并重启配置文件
# 历史文件记录
class ConfigMgr:
    _vhost_path = "/www/server/panel/vhost"

def __init__(self, site_name: str, config_prefix: str = ""):
        self.site_name = site_name
        self.config_prefix = config_prefix

def _read_config(self, web_server: str) -> Optional[str]:
        config_file = "{}/{}/{}{}.conf".format(self._vhost_path, web_server, self.config_prefix, self.site_name)
        res = read_file(config_file)
        if isinstance(res, str):
            return res
        return None

def nginx_config(self) -> Optional[str]:
        return self._read_config("nginx")

def apache_config(self) -> Optional[str]:
        return self._read_config("apache")

def save_config(self, conf_data: str, web_server: str):
        config_file = "{}/{}/{}{}.conf".format(self._vhost_path, web_server, self.config_prefix, self.site_name)
        old_config = self._read_config(web_server)
        write_file(config_file, conf_data)
        errmsg = check_server_config()
        if errmsg:
            write_file(config_file, old_config)
            return errmsg
        self._save_history(web_server)
        service_reload()

def save_nginx_config(self, conf_data: str) -> Optional[str]:
        return self.save_config(conf_data, "nginx")

def save_apache_config(self, conf_data: str) -> Optional[str]:
        return self.save_config(conf_data, "apache")

def history_list(self):
        his_path = '/www/backup/file_history'
        nginx_config_file = "{}/nginx/{}{}.conf".format(self._vhost_path, self.config_prefix, self.site_name)
        ng_save_path = "{}{}".format(his_path, nginx_config_file)
        apache_config_file = "{}/apache/{}{}.conf".format(self._vhost_path, self.config_prefix, self.site_name)
        ap_save_path = "{}{}".format(his_path, apache_config_file)
        return {
            "nginx": [] if not os.path.isdir(ng_save_path) else sorted(os.listdir(ng_save_path), reverse=True),
            "apache": [] if not os.path.isdir(ap_save_path) else sorted(os.listdir(ap_save_path), reverse=True)
        }

def history_conf(self, history_id: str) -> Optional[str]:
        his_path = '/www/backup/file_history'
        nginx_config_file = "{}/nginx/{}{}.conf".format(self._vhost_path, self.config_prefix, self.site_name)
        ng_save_path = "{}{}".format(his_path, nginx_config_file)
        if os.path.isdir(ng_save_path):
            for i in os.listdir(ng_save_path):
                if i == history_id:
                    return read_file(os.path.join(ng_save_path, i))

apache_config_file = "{}/apache/{}{}.conf".format(self._vhost_path, self.config_prefix, self.site_name)
        ap_save_path = "{}{}".format(his_path, apache_config_file)
        if os.path.isdir(ap_save_path):
            for i in os.listdir(ap_save_path):
                if i == history_id:
                    return read_file(os.path.join(ap_save_path, i))
        return None

def remove_history_file(self, history_id: str) -> None:
        his_path = '/www/backup/file_history'
        nginx_config_file = "{}/nginx/{}{}.conf".format(self._vhost_path, self.config_prefix, self.site_name)
        ng_save_path = "{}{}".format(his_path, nginx_config_file)
        if os.path.isdir(ng_save_path):
            for i in os.listdir(ng_save_path):
                if i == history_id:
                    os.remove(os.path.join(ng_save_path, i))

def clear_history_file(self) -> None:
        """
        清空所有的历史文件
        """
        his_path = '/www/backup/file_history'
        nginx_config_file = "{}/nginx/{}{}.conf".format(self._vhost_path, self.config_prefix, self.site_name)
        ng_save_path = "{}{}".format(his_path, nginx_config_file)
        if os.path.isdir(ng_save_path):
            for i in os.listdir(ng_save_path):
                os.remove(os.path.join(ng_save_path, i))

@staticmethod
    def _file_md5(filename):
        if not os.path.isfile(filename):
            return False
        md5_obj = md5()
        with open(filename, mode="rb") as f:
            while True:
                b = f.read(8096)
                if not b:
                    break
                md5_obj.update(b)

return md5_obj.hexdigest()

def _save_history(self, web_server: str):
        if os.path.exists('/www/server/panel/data/not_file_history.pl'):
            return True

his_path = '/www/backup/file_history'
        filename = "{}/{}/{}{}.conf".format(self._vhost_path, web_server, self.config_prefix, self.site_name)
        save_path = "{}{}".format(his_path, filename)
        if not os.path.isdir(save_path):
            os.makedirs(save_path, 384)

his_list = sorted(os.listdir(save_path), reverse=True)  # 倒序排列已有的历史文件
        try:
            num = int(read_file('data/history_num.pl'))
        except (ValueError, TypeError):
            num = 100

is_write = True
        if len(his_list) > 0:
            new_file_md5 = self._file_md5(filename)
            last_file_md5 = self._file_md5(os.path.join(save_path, his_list[0]))
            is_write = new_file_md5 != last_file_md5

if is_write:
            new_name = str(int(time.time()))
            write_file(os.path.join(save_path, new_name), read_file(filename, 'rb'), "wb")
            his_list.insert(0, new_name)

# 删除多余的副本
        for i in his_list[num:]:
            rm_file = save_path + '/' + i
            if os.path.exists(rm_file):
                os.remove(rm_file)

${this.title}

Code Editor : config_mgr.py