# coding: utf-8 # ------------------------------------------------------------------- # 宝塔Linux面板 # ------------------------------------------------------------------- # Copyright (c) 2014-2099 宝塔软件(http://bt.cn) All rights reserved. # ------------------------------------------------------------------- # Author: wzz # ------------------------------------------------------------------- # ------------------------------ # Docker模型 # ------------------------------ import public import os import time import json import re from btdockerModel import dk_public as dp from btdockerModel import setupModel as ds from btdockerModel import volumeModel as dv from btdockerModel.dockerBase import dockerBase class main(dockerBase): compose_path = "{}/data/compose".format(public.get_panel_path()) project_path = "/www/dk_project" templates_path = "{}/templates".format(project_path) config_path = "{}/config".format(public.get_panel_path()) info_path = "{}/docker_project_info.json".format(config_path) __first_pl = "{}/first.pl".format(project_path) def __init__(self): self.log_file = "/tmp/dk_project_run.log" self.docker_setup = ds.main() if not os.path.exists(self.templates_path): os.system("mkdir -p {}".format(self.templates_path)) self.compose_cmd = "/usr/bin/docker-compose" if self.docker_setup.check_docker_compose_service()[0] \ else "/usr/local/bin/docker-compose" def __check_conf(self, filename): ''' 验证配置文件是否可执行 @param filename: docker-compose.yml文件路劲 @return: ''' return public.ExecShell("{} -f {} config".format(self.compose_cmd, filename)) def sync_item(self, get): ''' 同步官方可以一键部署的项目 @param get: 空对象 @return: ''' os.remove(self.info_path) project_info = self.get_project_list(get) failed_list = [] successes_list = [] for info in project_info: if info["server_name"]: down_project_yml = self.__download_project_yml(info["server_name"]) if not down_project_yml["status"]: failed_list.append(info["server_name"]) continue successes_list.append(info["server_name"]) data = [{"successes": len(successes_list), "server_name": successes_list}, {"failed": len(failed_list), "server_name": failed_list}] return data def __first_sync_item(self, project_info): ''' 同步官方可以一键部署的项目 @param get: 空对象 @return: ''' failed_list = [] successes_list = [] for info in project_info: if info["server_name"]: down_project_yml = self.__download_project_yml(info["server_name"]) if not down_project_yml["status"]: failed_list.append(info["server_name"]) continue successes_list.append(info["server_name"]) data = [{"successes": len(successes_list), "server_name": successes_list}, {"failed": len(failed_list), "server_name": failed_list}] return data def get_project_list(self, get): ''' 获取支持一键部署的项目列表 @param get: @return: ''' project_info = [] try: if not os.path.exists(self.info_path): down_info = self.__download_info(self.info_path) if not down_info["status"]: return project_info project_info = json.loads(public.readFile(self.info_path)) project_info.sort(key=lambda x: x["sort"]) if not os.path.exists(self.__first_pl): sync_result = self.__first_sync_item(project_info) for result in sync_result: if result.get("successes") and result["successes"] <= 0: return project_info public.ExecShell("echo \"first\" > {}".format(self.__first_pl)) except Exception as e: project_info = [] return project_info def __get_docker_status(self, args): ''' 获取docker安装和启动状态 @param args: @return: ''' return { "installed": self.docker_setup.check_docker_compose_service(), "service_status": self.docker_setup.get_service_status() } def __download_info(self, info_path): ''' 下载版本信息: info.json @param info_path: string info.json文件的路劲 @return: ''' url = "{}/install/lib/docker_project/docker_project_info.json".format(public.get_url()) dp.download_file(url, info_path) if os.path.exists(info_path): return public.returnMsg(True, "info.json下载完成!") return public.returnMsg(False, "info.json下载失败!") def __download_project_yml(self, server_name): ''' 下载指定项目压缩包 @param server_name: string 模板名称,如nextcloud @return: ''' try: path = "{}/{}".format(self.templates_path, server_name) filename = "{}/{}.tar.gz".format(self.templates_path, server_name) compose_file = "{}/docker-compose.yml".format(path) url = "{}/install/lib/docker_project/templates/{}.tar.gz".format(public.get_url(), server_name) dp.download_file(url, filename) if not os.path.exists(filename): return public.returnMsg(False, "{}下载失败,请重新同步!".format(server_name)) if os.path.getsize(filename) == 0: os.remove(filename) return public.returnMsg(False, "{}下载失败,请重新同步!".format(server_name)) self.__tar_x_yml(server_name, path, filename) if os.path.exists(compose_file): check_conf = self.__check_conf(compose_file) if check_conf[1]: return public.returnMsg(False, "{}yml文件测试失败,{}".format(server_name, check_conf[1])) return public.returnMsg(True, "{}下载完成!".format(server_name)) except: return public.returnMsg(False, "{}下载失败,请重新同步!".format(server_name)) def __tar_x_yml(self, server_name, path=None, filename=None): ''' 解压项目模板方法 @param server_name: 模板名称,如nextcloud @param path: 项目模板路劲,如/www/dk_project/templates/nextcloud @param filename: 项目模板压缩包,如/www/dk_project/templates/nextcloud.tar.gz @return: ''' tar_result = public.ExecShell("tar xvf {} -C {}".format(filename, self.templates_path)) if tar_result[1]: os.remove(path) os.remove(filename) return public.returnMsg(False, "{}解压失败".format(server_name)) return public.returnMsg(True, "{}解压成功".format(server_name)) def create_project_volume(self, server_name, project_name, dir_names, volume_path): ''' 创建指定项目的数据存储卷 @param volume_path: @param project_name: string @param dir_names: list [dir_name,dir_name,...] @return: ''' args = public.dict_obj() args.url = "unix:///var/run/docker.sock" # volumes = dv.main().get_volume_list(args) # {'status': True, 'msg': {'volume': [], 'installed': True, 'service_status': True}} # if volumes['status']: # volumes = volumes['msg']['volume'] # else: # volumes = list() # volume的值,一个list: [] for dir_name in dir_names: # # 如果已经存在就跳过 # for volume in volumes: # if dir_name == volume["Name"]: # continue if volume_path == "": path = "{}/projects/{}/data/{}".format(self.project_path, project_name, dir_name) else: path = "{}/data/{}".format(volume_path, dir_name) is_mkdir = public.ExecShell("mkdir -p {}".format(path)) if is_mkdir[1]: return public.returnMsg(False, "目录创建失败,原因如下: {}".format(is_mkdir[1])) args.name = "{}_{}_{}".format(project_name, server_name, dir_name) args.driver = "local" args.driver_opts = {'type': 'none', 'device': path, 'o': 'bind'} args.labels = {} dv.main().add(args) return public.returnMsg(True, "存储卷已创建完成") def get_project(self, get): ''' 获取指定一键部署项目的配置信息 @param get: get.server_name @return: ''' try: server_name = getattr(get, "server_name") info_path = "{}/{}/conf.json".format(self.templates_path, server_name) project_info = json.loads(public.readFile(info_path)) volume_placeholder = "默认: {}/projects/您的项目名称/data/".format(self.project_path) total_sum = len(project_info) volume_path = {"id": total_sum + 1, "sort": total_sum + 1, "type": "string", "key": "VOLUME_PATH", "value": "", "placeholder": volume_placeholder, "ps": "数据存储目录"} project_info.append(volume_path) except: project_info = [] return project_info def __get_server_ps(self, project_conf, conf_key): ''' 获取对应服务名的标题 @param project_conf: @param conf_key: @return: ''' get = public.dict_obj() for conf in project_conf: if conf["key"] == "SERVER_NAME": get.server_name = conf["value"] server_conf = self.get_project(get) for server in server_conf: if conf_key == server["key"]: return server["ps"] return conf_key def get_project_logs(self, get): """ 获取一键部署日志，websocket @param get: @return: """ get.wsLogTitle = "开始执行命令,请等待..." print(self.log_file) get._log_path = self.log_file return self.get_ws_log(get) def create_project(self, get): ''' 创建一键部署的项目 @param get: @return: ''' project_conf = getattr(get, "project_conf") remark = "" for conf in project_conf: if conf["key"] != "REMARK" and type(conf["value"]) != list: if re.search('\s', conf["value"]): server_ps = self.__get_server_ps(project_conf, conf["key"]) return public.returnMsg(False, "{}不能存在空格".format(server_ps)) if conf["key"] != "VOLUME_PATH" and conf["key"] != "REMARK": if conf["value"] == "": server_ps = self.__get_server_ps(project_conf, conf["key"]) return public.returnMsg(False, "{}不能为空!".format(server_ps)) if conf["key"].upper() == "PROJECT_NAME": project_name = conf["value"].strip() if conf["key"].upper() == "VOLUME_PATH": project_volume = conf["value"].strip() if conf["key"].upper() == "SERVER_NAME": server_name = conf["value"].strip() if conf["key"].upper() == "VOLUMES": # VOLUMES = list volumes = conf["value"] if conf["key"].upper() == "PORT": if dp.check_socket(conf["value"]): return public.returnMsg(False, "服务器端口[{}]已被占用，请更换为其他端口！".format(conf['value'])) project_port = conf["value"] if conf["key"] == "REMARK": remark = conf["value"] config_path = "{}/config/name_map.json".format(public.get_panel_path()) if not os.path.exists(config_path): public.writeFile(config_path, json.dumps({})) if public.readFile(config_path) == '': public.writeFile(config_path, json.dumps({})) name_map = json.loads(public.readFile(config_path)) name_str = 'q18q' + public.GetRandomString(10).lower() name_map[name_str] = project_name project_name = name_str public.writeFile(config_path, json.dumps(name_map)) server_dir = "{}/{}".format(self.templates_path, server_name) project_dir = "{}/projects/{}/{}_{}".format(self.project_path, project_name, project_name, server_name) public.set_module_logs('docker_project', 'create_project', 1) check_result = self.__create_dir(project_dir, project_name, server_name, server_dir) if not check_result["status"]: return check_result self.__write_config(project_dir, project_name, server_name, project_conf) self.create_project_volume(server_name, project_name, volumes, project_volume) run_result = self.__project_run(project_dir, project_name) if run_result["status"]: self.__add_sql(project_dir, project_name, server_name, remark) dp.write_log("一键部署项目 [{}] 成功!".format(server_name)) return self.__return_msg(project_port) return run_result def __project_run(self, project_dir, project_name): ''' 运行项目 @param project_dir: 项目运行目录 @param server_name: 服务名称 @return: ''' filename = "{}/docker-compose.yml".format(project_dir) check_result = self.__check_conf(filename) if check_result[1]: return public.returnMsg(False, "项目启动失败{}".format(check_result[1])) public.ExecShell("echo -n > {}".format(self.log_file)) public.ExecShell("nohup {} -f {}/docker-compose.yml up -d >> {} 2>&1 &&" " echo 'bt_successful' >> {} || echo 'bt_failed' >> {} &" .format( self.compose_cmd, project_dir, self.log_file, self.log_file, self.log_file )) return public.returnMsg(True, "开始创建项目") def __create_dir(self, project_dir, project_name, server_name, server_dir): ''' 创建项目目录 @param project_dir: 项目目录 @param project_name: 项目名称 @param server_dir: 服务源目录 @return: ''' if self.__check_repeat(project_dir, project_name, server_name): return public.returnMsg(False, "{}已存在,请更换项目名称".format(project_name)) mk_result = public.ExecShell("mkdir -p {}".format(project_dir)) if mk_result[1]: return public.returnMsg(False, "用户项目目录创建失败,详情: {}".format(mk_result[1])) cp_result = public.ExecShell("cp -a {}/. {}/".format(server_dir, project_dir)) if cp_result[1]: return public.returnMsg(False, "项目目录复制失败,详情: {}".format(cp_result[1])) return public.returnMsg(True, "") def __add_sql(self, project_dir, project_name, server_name, remark): ''' 添加项目到docker数据库中 @param project_dir: 项目路劲 @param project_name: 项目名称 @return: ''' pdata = { "name": public.xsssec("{}_{}".format(project_name, server_name)), "status": "1", "path": "{}/docker-compose.yml".format(project_dir), "template_id": "", "time": time.time(), "remark": public.xsssec(remark) } dp.sql("stacks").insert(pdata) def __return_msg(self, project_port): ''' 创建成功后返回给用户的数据 @param project_port: @return: ''' server_ip = public.get_server_ip() local_ip = public.GetLocalIp() data = {"protocol": "http", "server_ip": server_ip, "local_ip": local_ip, "port": project_port} return public.returnMsg(True, data) def __check_repeat(self, project_dir, project_name, server_name): ''' 检查是否存在相同项目 @param project_dir: 项目路劲 @return: ''' # if os.path.exists(project_dir): # return True stacks_info = dp.sql("stacks").where("name=?", ("{}_{}".format(project_name, server_name),)).find() if stacks_info: return True return False def __write_config(self, project_dir, project_name, server_name, project_conf): ''' 写配置文件 @param project_dir: 用户项目目录 @param project_name: 项目名称 @param server_name: 服务名称，如nextcloud @param project_conf: 新的配置文件内容 @return: ''' old_env_path = "{}/{}/.env".format(self.templates_path, server_name) new_env_path = "{}/.env".format(project_dir) env_conf = "" if not os.path.exists(old_env_path): public.ExecShell("echo > {}".format(old_env_path)) with open(old_env_path) as env: lines = env.readlines() # 取旧文件转字典 old_dict = {} for line in lines: if "=" in line: temp = line.split("=") old_dict[temp[0]] = temp[1] # 新数据转字典 new_dict = {} for conf in project_conf: if conf["key"] == "VOLUME_PATH": project_volume = conf["value"] if "默认路劲" in project_volume: conf["value"] = "{}/{}/data/".format(self.project_path, project_name) continue if conf["key"] == "VOLUMES": continue new_dict[conf["key"].upper()] = conf["value"] # 旧字典更新新字典的内容 old_dict.update(new_dict) # 拼接成新的环境变量文件 for key, value in old_dict.items(): env_conf += "{}={}\n".format(key, value.strip()) public.writeFile(new_env_path, env_conf) return True def sync_compose_template(self, server_name): ''' 同步模板到项目模板页面 @param server_name: 模板名称 @return: ''' data = dp.sql("templates").where("name=?", (server_name,)).find() # if data: dp.sql("templates").delete(id=data["id"]) if data: return pdata = { "name": server_name, "remark": "仅限堡塔Docker快速部署项目使用的模板[勿删勿单独使用来创建项目]", "path": "{}/{}/docker-compose.yml".format(self.templates_path, server_name) } dp.sql("templates").insert(pdata) dp.write_log("Add template [{}] successful!".format(server_name))