File size: 9,823 Bytes
3a5cf48 | 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, "安装任务已添加到队列中！")
|