Hugging Face's logo

Spaces:
GGSheng
/
action
Running

App Files Community
action
File size: 9,823 Bytes
020c337
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
# coding: utf-8
# -------------------------------------------------------------------
# 宝塔Linux面板
# -------------------------------------------------------------------
# Copyright (c) 2015-2099 宝塔软件(http://bt.cn) All rights reserved.
# -------------------------------------------------------------------
# Author: zouhw <zhw@bt.cn>
# -------------------------------------------------------------------

# ------------------------------
# Docker模型
# ------------------------------

import os
import json
import public
import projectModel.bt_docker.dk_public as dp


class main:
    def get_config(self, args):
        import projectModel.bt_docker.dk_public as dp
        # 获取加速配置
        registry_mirrors = self.get_registry_mirrors(args)
        if not registry_mirrors["status"]:
            return registry_mirrors
        else:
            if not isinstance(registry_mirrors['msg'], list):
                registry_mirrors['msg'] = [registry_mirrors['msg']]
        service_status = self.get_service_status()
        return public.returnMsg(
            True, {
                "registry_mirrors": registry_mirrors['msg'],
                "service_status": service_status,
                "installed": self.check_docker_program(),
                "monitor_status": self.get_monitor_status(),
                "monitor_save_date": dp.docker_conf()['SAVE']
            })

    def set_monitor_save_date(self, args):
        """
        :param save_date: int 例如30 表示 30天
        :param args:
        :return:
        """
        import re
        conf_path = "{}/data/docker.conf".format(public.get_panel_path())
        docker_conf = public.readFile(conf_path)
        try:
            save_date = int(args.save_date)
        except:
            return public.returnMsg(False, "监控保存时间需要为正整数！")
        if save_date > 999:
            return public.returnMsg(False, "监控数据不能保留超过999天！")
        if not docker_conf:
            docker_conf = "SAVE={}".format(save_date)
            public.writeFile(conf_path, docker_conf)
            return public.returnMsg(True, "设置成功！")
        docker_conf = re.sub("SAVE\s*=\s*\d+", "SAVE={}".format(save_date),
                             docker_conf)
        public.writeFile(conf_path, docker_conf)
        dp.write_log("设置监控时间为[{}]天！".format(save_date))
        return public.returnMsg(True, "设置成功！")

    def get_service_status(self):
        import projectModel.bt_docker.dk_public as dp
        sock = '/var/run/docker.pid'
        if os.path.exists(sock):
            try:
                client = dp.docker_client()
                if client:
                    return True
                else:
                    return False
            except:
                return False
        else:
            return False

    # docker服务状态设置
    def docker_service(self, args):
        """
        :param act start/stop/restart
        :param args:
        :return:
        """
        import public
        act_dict = {'start': 'start', 'stop': 'stop', 'restart': 'restart'}
        if args.act not in act_dict:
            return public.returnMsg(False, '没有办法做到这一点！')
        exec_str = 'systemctl {} docker'.format(args.act)
        if args.act == "stop":
            exec_str += "&& systemctl {} docker.socket".format(args.act)
        public.ExecShell(exec_str)
        dp.write_log("将 Docker 服务状态设置为 [{}] 成功".format(act_dict[args.act]))
        return public.returnMsg(True,
                                "将状态设置为 [{}] 成功".format(act_dict[args.act]))

    # 获取加速配置
    def get_registry_mirrors(self, args):
        try:
            if not os.path.exists('/etc/docker/daemon.json'):
                return public.returnMsg(True, [])
            if public.readFile('/etc/docker/daemon.json') == '':
                return public.returnMsg(True, [])
            conf = json.loads(public.readFile('/etc/docker/daemon.json'))
            if "registry-mirrors" not in conf:
                return public.returnMsg(True, [])
            return public.returnMsg(True, conf['registry-mirrors'])
        except:
            return public.returnMsg(
                False, '失败！失败原因：{}'.format(public.get_error_info()))

    # 设置加速配置
    def set_registry_mirrors(self, args):
        """
        :param registry_mirrors_address registry.docker-cn.com\nhub-mirror.c.163.com
        :param args:
        :return:
        """
        import re
        try:
            conf = {}
            if os.path.exists('/etc/docker/daemon.json'):
                conf = json.loads(public.readFile('/etc/docker/daemon.json'))
            if not args.registry_mirrors_address.strip():
                # return public.returnMsg(False, '加速地址不能为空！')
                if 'registry-mirrors' not in conf:
                    return public.returnMsg(True, '设置成功')
                del (conf['registry-mirrors'])
            else:
                registry_mirrors = args.registry_mirrors_address.strip().split(
                    '\n')
                for i in registry_mirrors:
                    if not re.search('https?://', i):
                        return public.returnMsg(
                            False,
                            '加速地址[{}]格式错误<br>参考：https://mirror.ccs.tencentyun.com'
                            .format(i))
                tmp_registry = registry_mirrors
                if isinstance(registry_mirrors, list) and registry_mirrors:
                    tmp_registry = registry_mirrors[0]
                conf['registry-mirrors'] = public.xsssec2(tmp_registry)
                if isinstance(conf['registry-mirrors'], str):
                    conf['registry-mirrors'] = [conf['registry-mirrors']]
            public.writeFile('/etc/docker/daemon.json',
                             json.dumps(conf, indent=2))
            dp.write_log("设置Docker加速成功!")
            return public.returnMsg(True, '设置成功')
        except:
            return public.returnMsg(
                False, '设置失败！失败原因:{}'.format(public.get_error_info()))

    def get_monitor_status(self):
        """
        :return:
        """
        # 进程是否存在
        res = public.process_exists(
            "python",
            cmdline=
            "/www/server/panel/class/projectModel/bt_docker/dk_monitor.py")
        if res:
            return res
        res = public.process_exists(
            "python3",
            cmdline=
            "/www/server/panel/class/projectModel/bt_docker/dk_monitor.py")
        if res:
            return res
        return res

    def set_docker_monitor(self, args):
        """
        开启docker监控获取docker相取资源信息
        :param act: start/stop
        :return:
        """
        import time
        import projectModel.bt_docker.dk_public as dp
        python = "/www/server/panel/pyenv/bin/python"
        if not os.path.exists(python):
            python = "/www/server/panel/pyenv/bin/python3"
        cmd_line = "/www/server/panel/class/projectModel/bt_docker/dk_monitor.py"
        if args.act == "start":
            shell = "nohup {} {} &".format(python, cmd_line)
            public.ExecShell(shell)
            time.sleep(1)
            if self.get_monitor_status():
                dp.write_log("Docker监控启动成功！")
                return public.returnMsg(True, "启动监控成功！")
            return public.returnMsg(False, "启动监控失败！")
        else:
            pid = dp.get_process_id(
                "python",
                "/www/server/panel/class/projectModel/bt_docker/dk_monitor.py")
            if not pid:
                pid = dp.get_process_id(
                    "python3",
                    "/www/server/panel/class/projectModel/bt_docker/dk_monitor.py"
                )
            public.ExecShell("kill -9 {}".format(pid))
            dp.write_log("Docker监控成功停止！")
            return public.returnMsg(True, "Docker监控成功停止！")

    def check_docker_program(self):
        """
        检查docker和docker-compose是否已经安装
        :return:
        """
        docker = "/usr/bin/docker"
        docker_compose = "/usr/bin/docker-compose"
        if not os.path.exists(docker_compose):
            dk_compose_list = ["/usr/libexec/docker/cli-plugins/docker-compose", "/usr/local/docker-compose"]
            for i in dk_compose_list:
                if os.path.exists(i):
                    public.ExecShell("ln -sf {} {}".format(i, docker_compose))
        if not os.path.exists(docker) or not os.path.exists(docker_compose):
            return False
        return True

    def install_docker_program(self, args):
        """
        安装docker和docker-compose
        :param args:
        :return:
        """
        import time
        mmsg = "Install Docker service"
        # Docker_Install_File = "/www/server/panel/install/docker_install.sh"
        # execstr = "wget -O {} http://download.bt.cn/install/0/docker_install.sh -T 5"
        execstr = "/bin/bash /www/server/panel/install/install_soft.sh 0 install docker_install"
        public.M('tasks').add('id,name,type,status,addtime,execstr',
                              (None, mmsg, 'execshell', '0',
                               time.strftime('%Y-%m-%d %H:%M:%S'), execstr))
        public.httpPost(
            public.GetConfigValue('home') + '/api/panel/plugin_total', {
                "pid": "1111111",
                'p_name': "Docker商用模块"
            }, 3)
        return public.returnMsg(True, "安装任务已添加到队列中！")