# coding: utf-8 # ------------------------------------------------------------------- # aaPanel # ------------------------------------------------------------------- # Copyright (c) 2015-2099 aaPanel(www.aapanel.com) All rights reserved. # ------------------------------------------------------------------- # Author: wzz # ------------------------------------------------------------------- # ------------------------------ # Docker模型 # ------------------------------ import json import os import time import gettext _ = gettext.gettext import public from btdockerModelV2 import containerModel as dc from btdockerModelV2 import dk_public as dp from btdockerModelV2.dockerBase import dockerBase from public.validate import Param class main(dockerBase): def get_docker_compose_version(self): try: import subprocess result = subprocess.run(["docker-compose", "version", "--short"], capture_output=True, text=True) version_str = result.stdout.strip() major, minor, patch = map(int, version_str.split('.')) return major, minor, patch except Exception as e: print("Error:", e) return None # 验证配置文件 def check_conf(self, path): # 2024/3/21 下午 5:58 检测path是否存在中文，如果有就false return if public.check_chinese(path): return public.returnMsg(False, "The file path cannot contain Chinese characters!") # 2024/3/20 下午 4:33 获取docker-compose的版本，如果大于v2.24.7则不需要检测，比如是v2.25，解决高版本的docker-compose version = self.get_docker_compose_version() if version and version > (2, 24, 7): tmpfile = public.md5(path) public.ExecShell(r"\cp -r {} /tmp/{}.yml".format(path, public.md5(path))) public.ExecShell("sed -i '/version:/d' /tmp/{}.yml".format(tmpfile)) path = "/tmp/{}.yml".format(tmpfile) shell = "/usr/bin/docker-compose -f {} config".format(path) a, e = public.ExecShell(shell) if e and "setlocale: LC_ALL: cannot change locale (en_US.UTF-8)" not in e: return public.returnMsg(False, "Detection failed: {}".format(e)) return public.returnMsg(True, "Detection passes!") # 用引导方式创建模板 def add_template_gui(self, get): """ 用引导方式创建模板 :param name 模板名 :param description 模板描述 :param data 模板内容 {"version":3,"services":{...}...} :param get: 模板文件参数： version 2/3version 2: 仅支持单机 3：支持单机和多机模式 services: 多个容器的集合 下一层执行服务名 如web1,服务名下面指定服务的变量 web1: build: . 基于dockerfile构建一个镜像 image: nginx 服务所使用的镜像为nginx container_name: "web" 容器名 depends_on: 该服务在db服务启动后再启动 - db ports: - "6061:80" 将容器的80端口映射到主机的6061端口 networks: - frontend 该容器所在的网络 deploy: 指定与部署和运行服务相关的配置(在使用 swarm时才会生效) replicas: 6 6个副本 update_config: parallelism: 2 delay: 10s restart_policy: condition: on-failure 其他详细描述可以参考 https://docs.docker.com/compose/compose-file/compose-file-v3 :return: """ import yaml path = "{}/template".format(self.compose_path) file = "{}/{}.yaml".format(path, get.name) if not os.path.exists(path): os.makedirs(path) data = json.loads(get.data) yaml.dump(data, file) def get_template_kw(self, get): data = { "version": "", "services": { "server_name_str": { # 用户输入 "build": { "context": "str", "dockerfile": "str", "get": [], "cache_from": [], "labels": [], "network": "str", "shm_size": "str", "target": "str" }, "cap_add": "", "cap_drop": "", "cgroup_parent": "str", "command": "str", "configs": { "my_config_str": [] }, "container_name": "str", "credential_spec": { "file": "str", "registry": "str" }, "depends_on": [], "deploy": { "endpoint_mode": "str", "labels": { "key": "value" }, "mode": "str", "placement": [{"key": "value"}], "max_replicas_per_node": "int", "replicas": "int", "resources": { "limits": { "cpus": "str", "memory": "str", }, "reservations": { "cpus": "str", "memory": "str", }, "restart_policy": { "condition": "str", "delay": "str", "max_attempts": "int", "window": "str" } } } } } } # 创建项目配置文件 def add_template(self, get): """ 添加一个模板文件 :param name 模板名 :param remark 模板描述 :param data 模板内容 :param get: :return: """ import re name = get.name if not re.search(r"^[\w\.\-]+$", name): return public.return_message(-1, 0, "Template names cannot contain special characters; only letters, numbers, underscores, dots, and underscores are supported") template_list = self._template_list(get) for template in template_list: if name == template['name']: return public.return_message(-1, 0, _("This template name already exists!")) path = "{}/{}/template".format(self.compose_path, name) file = "{}/{}.yaml".format(path, name) if not os.path.exists(path): os.makedirs(path) public.writeFile(file, get.data) check_res = self.check_conf(file) if not check_res['status']: if os.path.exists(file): os.remove(file) return public.return_message(-1, 0, check_res['msg']) pdata = { "name": name, "remark": public.xsssec(get.remark), "path": file } dp.sql("templates").insert(pdata) dp.write_log("Added template [{}] successfully!".format(name)) public.set_module_logs('docker', 'add_template', 1) return public.return_message(0, 0,_("Template added successfully!")) def edit_template(self, get): """ :param id 模板id :param data 模板内容 :param remark 模板描述 :param get: :return: """ template_info = dp.sql("templates").where("id=?", (get.id,)).find() if not template_info: return public.return_message(-1, 0, _("Did not change the template!")) if "data" not in get: return public.return_message(-1, 0, "Template content format error, please enter a valid docker-compose template!") if "version" not in get.data: return public.return_message(-1, 0, "Template content format error, please enter a valid docker-compose template!") public.writeFile(template_info['path'], get.data) check_res = self.check_conf(template_info['path']) if not check_res['status']: return public.return_message(-1, 0,check_res['msg']) pdata = { "name": get.name, "remark": public.xsssec(get.remark), "path": template_info['path'] } dp.sql("templates").where("id=?", (get.id,)).update(pdata) dp.write_log("Edit template [{}] successful!".format(template_info['name'])) return public.return_message(0, 0, _("Modified template successfully!")) def get_template(self, get): """ id 模板ID 获取模板内容 :return: """ # 校验参数 try: get.validate([ Param('template_id').Require().Integer(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) template_info = dp.sql("templates").where("id=?", (get.template_id,)).find() if not template_info: return public.return_message(-1, 0, _("This template was not found!")) return public.return_message(0, 0, public.readFile(template_info['path'])) def template_list(self, get): """ 获取所有模板 :param get: :return: """ template = dp.sql("templates").select()[::-1] if not isinstance(template, list): template = [] return public.return_message(0, 0, template) # 内部调用 不改响应格式 def _template_list(self, get): """ 获取所有模板 :param get: :return: """ template = dp.sql("templates").select()[::-1] if not isinstance(template, list): template = [] return template def remove_template(self, get): """ 删除模板 :param template_id :param get: :return: """ # 校验参数 try: get.validate([ Param('template_id').Require().Integer(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) data = dp.sql("templates").where("id=?", (get.template_id,)).find() if not data: return public.return_message(-1, 0, _("This template was not found!")) if os.path.exists(data['path']): os.remove(data['path']) dp.sql("templates").delete(id=get.template_id) dp.write_log("Delete template [{}] successfully!".format(data['name'])) return public.return_message(0, 0, _("successfully delete!")) def edit_project_remark(self, get): """ 编辑项目 :param project_id 项目 :param remark备注 :param get: :return: """ stacks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not stacks_info: return public.returnMsg(False, "The item was not found!") pdata = { "remark": public.xsssec(get.remark) } dp.write_log("Comment for project [{}] changed successfully [{}] --> [{}]!".format(stacks_info['name'], stacks_info['remark'], public.xsssec(get.remark))) dp.sql("stacks").where("id=?", (get.project_id,)).update(pdata) def edit_template_remark(self, get): """ 编辑项目 :param templates_id 项目 :param remark备注 :param get: :return: """ stacks_info = dp.sql("templates").where("id=?", (get.templates_id,)).find() if not stacks_info: return public.returnMsg(False, "The template was not found!") pdata = { "remark": public.xsssec(get.remark) } dp.write_log( "Modify template [{}] Remark successful [{}] --> [{}]!".format(stacks_info['name'], stacks_info['remark'], public.xsssec(get.remark))) dp.sql("templates").where("id=?", (get.templates_id,)).update(pdata) def create_project_in_path(self, name, path): shell = "cd {} && /usr/bin/docker-compose -p {} up -d &> {}".format("/".join(path.split("/")[:-1]), name, self._log_path) public.ExecShell(shell) def create_project_in_file(self, project_name, file): project_path = "{}/{}".format(self.compose_path, project_name) project_file = "{}/docker-compose.yaml".format(project_path) if not os.path.exists(project_path): os.makedirs(project_path) template_content = public.readFile(file) public.writeFile(project_file, template_content) shell = "/usr/bin/docker-compose -p {} -f {} up -d &> {}".format(project_name, project_file, self._log_path) public.ExecShell(shell) def check_project_container_name(self, template_data, get): """ 检测模板文件中的容器名是否已经存在 :return: """ import re data = [] template_container_name = re.findall("container_name\\s*:\\s*[\"\']+(.*)[\'\"]", template_data) # 调用容器列表接口 选择不改统一返回的 container_list = dc.main()._get_list(get) container_list = container_list['container_list'] for container in container_list: if container['name'] in template_container_name: data.append(container['name']) if data: return public.returnMsg(False, "The container name already exists!:
[{}]".format(", ".join(data))) # 获取模板所使用的端口 rep = r"(\d+):\d+" port_list = re.findall(rep, template_data) for port in port_list: if dp.check_socket(port): return public.returnMsg(False, "This port [{}] is already used by other templates".format(port)) # 创建项目 def create(self, get): """ :param project_name 项目名 :param remark 描述 :param template_id 模板ID :param rags: :return: """ # {"template_id": "13", "project_name": "dedf2f", "remark": "从本地添加asd"} # 校验参数 try: get.validate([ Param('template_id').Require().Integer(), Param('project_name').Require().String().Xss(), Param('remark').Require().String(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) try: project_name = public.md5(get.project_name) # if "template_id" not in get: # return public.returnMsg(False, "Parameter error, please pass in template_id!") template_id = get.template_id template_info = dp.sql("templates").where("id=?", template_id).find() if len(template_info) < 1: return public.return_message(-1, 0, _("This template was not found, or file is corrupt!")) if not os.path.exists(template_info['path']): return public.return_message(-1, 0, _("Template file does not exist")) template_exist = dp.sql("stacks").where("template_id=?", (template_id,)).find() if template_exist: return public.return_message(-1, 0, "Template [{}] has been deployed by project: [{}], please change a template and try again!".format( template_info['name'], template_exist['name'])) name_exist = self.check_project_container_name(public.readFile(template_info['path']), get) if name_exist: return public.return_message(-1, 0, name_exist['msg']) stacks_info = dp.sql("stacks").where("name=?", (project_name)).find() if not stacks_info: pdata = { "name": public.xsssec(get.project_name), "status": "1", "path": template_info['path'], "template_id": template_id, "time": time.time(), "remark": public.xsssec(get.remark) } dp.sql("stacks").insert(pdata) else: return public.return_message(-1, 0, _("The project name already exists!")) if template_info['add_in_path'] == 1: self.create_project_in_path( project_name, template_info['path'] ) else: self.create_project_in_file( project_name, template_info['path'] ) dp.write_log("Project [{}] deployed successfully!".format(project_name)) public.set_module_logs('docker', 'add_project', 1) return public.return_message(0, 0, _("Successful deployment!")) except Exception as ex: public.print_log(traceback.format_exc()) return public.return_message(-1, 0, str(ex)) def compose_project_list(self, get): """ 获取所有已部署的项目列表 @param get: """ compose_project = dp.sql("stacks").select() # public.print_log("部署项目 {}".format(compose_project)) try: cmd_result = public.ExecShell("/usr/bin/docker-compose ls -a --format json")[0] if "Segmentation fault" in cmd_result: return public.returnMsg(False, "docker-compose is too low, please upgrade to the latest version!") result = json.loads(cmd_result) except: result = [] for i in compose_project: for j in result: if public.md5(i['name']) in j['Name']: i['run_status'] = j['Status'].split("(")[0].lower() break else: i['run_status'] = "exited" return public.return_message(0, 0, compose_project) def project_container_count(self, get): """ 获取项目容器数量 @param get: @return: """ from btdockerModelV2.dockerSock import container sk_container = container.dockerContainer() sk_container_list = sk_container.get_container() stacks_info = dp.sql("stacks").select() net_info = [] for i in stacks_info: count = 0 for c in sk_container_list: if public.md5(i['name']) in c["Names"][0].replace("/", ""): count += 1 continue if 'com.docker.compose.project' in c.keys(): if public.md5(i['name']) in c['com.docker.compose.project.config_files']: count += 1 continue if public.md5(i['name']) in public.md5(c['com.docker.compose.project.config_files']): count += 1 continue if 'com.docker.compose.project' in c['Labels'].keys(): if public.md5(i['name']) in c['Labels']['com.docker.compose.project.config_files']: count += 1 continue if public.md5(i['name']) in public.md5(c['Labels']['com.docker.compose.project.config_files']): count += 1 continue net_info.append(count) return public.return_message(0, 0, net_info) def get_compose_container(self, get): """ 目前仅支持本地 url: unix:///var/run/docker.sock """ # 校验参数 try: get.validate([ Param('name').Require().String(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) from btdockerModelV2.dockerSock import container sk_container = container.dockerContainer() sk_container_list = sk_container.get_container() project_container_list = [] for c in sk_container_list: if public.md5(get.name) in dp.rename(c["Names"][0].replace("/", "")): project_container_list.append(dc.main().struct_container_list(c)) continue if 'com.docker.compose.project' in c.keys(): if public.md5(get.name) in c['com.docker.compose.project.config_files']: project_container_list.append(dc.main().struct_container_list(c)) if public.md5(get.name) in public.md5(c['com.docker.compose.project.config_files']): project_container_list.append(dc.main().struct_container_list(c)) if 'com.docker.compose.project' in c['Labels'].keys(): if public.md5(get.name) in c['Labels']['com.docker.compose.project.config_files']: project_container_list.append(dc.main().struct_container_list(c)) if public.md5(get.name) in public.md5(c['Labels']['com.docker.compose.project.config_files']): project_container_list.append(dc.main().struct_container_list(c)) return public.return_message(0, 0, project_container_list) # 删除项目 def remove(self, get): """ project_id 数据库记录的项目ID :param get: :return: """ # 校验参数 try: get.validate([ Param('project_id').Require().Integer(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.return_message(-1, 0, _("The project name was not found!")) container_name = public.ExecShell("docker ps --format \"{{.Names}}\"") if statcks_info['name'] in container_name[0]: shell = f"/usr/bin/docker-compose -p {statcks_info['name']} -f {statcks_info['path']} down &> {self._log_path}" else: shell = f"/usr/bin/docker-compose -p {public.md5(statcks_info['name'])} -f" \ f" {statcks_info['path']} down &> {self._log_path}" public.ExecShell(shell) dp.sql("stacks").delete(id=get.project_id) dp.write_log("Delete project [{}] success!".format(statcks_info['name'])) return public.return_message(0, 0, _("successfully delete!")) def prune(self, get): """ 删除所有没有容器的项目 @param get: @return: """ stacks_info = dp.sql("stacks").select() container_name = public.ExecShell("docker ps --format \"{{.Names}}\"")[0] container_name = container_name.split("\n") for i in stacks_info: # 2024/3/21 下午 6:26 如果i['name']在container_name[0]中，说明容器还在运行，不删除 is_run = False docker_name = public.ExecShell("grep 'container_name' {}".format(i["path"]))[0] for j in container_name: if j == "": continue if public.md5(i['name']) in j or j in docker_name: is_run = True break if is_run: continue shell = "/usr/bin/docker-compose -f {} down &> {}".format(i['path'], self._log_path) public.ExecShell(shell) dp.sql("stacks").delete(id=i['id']) dp.write_log("Cleanup project [{}] successful!".format(i['name'])) return public.return_message(0, 0, _("Clean up successfully!")) def set_compose_status(self, get): """ 设置项目状态 @param get: @return: """ try: get.validate([ Param('status').Require().String('in', ['start', 'stop','restart','pause','unpause','kill']), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) if get.status == 'start': data = self.start(get) elif get.status == 'stop': data = self.stop(get) elif get.status == 'restart': data = self.restart(get) elif get.status == 'pause': data = self.pause(get) elif get.status == 'unpause': data = self.unpause(get) else: data = self.kill(get) if data["status"]: return public.return_message(0, 0, data['msg']) else: return public.return_message(-1, 0, data['msg']) def kill(self, get): """ 强制停止项目 @param get: @return: """ statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.returnMsg(False, "Project configuration not found!") shell = "/usr/bin/docker-compose -f {} kill &> {}".format( "{}/data/compose/{}/docker-compose.yaml".format(public.get_panel_path(), public.md5(statcks_info['name'])), self._log_path ) a, e = public.ExecShell(shell) if e: return public.returnMsg(False, "Stopping project failed: {}".format(e)) dp.write_log("Stopping project [{}] succeeded".format(statcks_info['name'])) return public.returnMsg(True, "Setup successful!") def stop(self, get): """ 停止项目 project_id 数据库记录的项目ID kill 强制停止项目 0/1 :param get: :return: """ statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.returnMsg(False, "Project configuration not found!") shell = "/usr/bin/docker-compose -f {} stop &> {}".format( "{}/data/compose/{}/docker-compose.yaml".format(public.get_panel_path(), public.md5(statcks_info['name'])), self._log_path ) a, e = public.ExecShell(shell) if e: return public.returnMsg(False, "Stopping project failed: {}".format(e)) dp.write_log("Stopping project [{}] succeeded!".format(statcks_info['name'])) return public.returnMsg(True, "Setup successful!") def start(self, get): """ 启动项目 project_id 数据库记录的项目ID :param get: :return: """ statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.returnMsg(False, "Project configuration not found!") shell = "/usr/bin/docker-compose -f {} start &> {}".format( "{}/data/compose/{}/docker-compose.yaml".format(public.get_panel_path(), public.md5(statcks_info['name'])), self._log_path ) a, e = public.ExecShell(shell) if e: return public.returnMsg(False, "Failed to start project: {}".format(e)) dp.write_log("Start project [{}] successful!".format(statcks_info['name'])) return public.returnMsg(True, "Setup successful!") def restart(self, get): """ 拉取项目内需要的镜像 project_id 数据库记录的项目ID :param get: :return: """ statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.returnMsg(False, "Project configuration not found!") shell = "/usr/bin/docker-compose -f {} restart &> {}".format( "{}/data/compose/{}/docker-compose.yaml".format(public.get_panel_path(), public.md5(statcks_info['name'])), self._log_path ) a, e = public.ExecShell(shell) if e: return public.returnMsg(False, "Failed to restart project: {}".format(e)) dp.write_log("Restart project [{}] successfully!".format(statcks_info['name'])) return public.returnMsg(True, "Successfully set!") def pull(self, get): """ 拉取模板内需要的镜像 template_id 数据库记录的项目ID :param get: :return: """ # 校验参数 try: get.validate([ Param('template_id').Require().Integer(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) statcks_info = dp.sql("templates").where("id=?", (get.template_id,)).find() if not statcks_info: return public.return_message(0, 0, _("The template was not found!")) os.system( "nohup /usr/bin/docker-compose -f {} pull >> {} 2>&1 " "&& echo 'bt_successful' >> {} " "|| echo 'bt_failed' >> {} &".format( statcks_info['path'], self._log_path, self._log_path, self._log_path, )) dp.write_log("The image inside the template [{}] was pulled successfully !".format(statcks_info['name'])) return public.return_message(0, 0, _("Pull successfully!")) def pause(self, get): """ 暂停项目 project_id 数据库记录的项目ID :param get: :return: """ statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.returnMsg(False, "Project configuration not found!") shell = "/usr/bin/docker-compose -f {} pause &> {}".format( "{}/data/compose/{}/docker-compose.yaml".format(public.get_panel_path(), public.md5(statcks_info['name'])), self._log_path ) a, e = public.ExecShell(shell) if e: return public.returnMsg(False, "Failed to suspend project: {}".format(e)) dp.write_log("Pause [{}] success!".format(statcks_info['name'])) return public.returnMsg(True, "Successfully set!") def unpause(self, get): """ 取消暂停项目 project_id 数据库记录的项目ID :param get: :return: """ statcks_info = dp.sql("stacks").where("id=?", (get.project_id,)).find() if not statcks_info: return public.returnMsg(False, "Project configuration not found!") shell = "/usr/bin/docker-compose -f {} unpause &> {}".format( "{}/data/compose/{}/docker-compose.yaml".format(public.get_panel_path(), public.md5(statcks_info['name'])), self._log_path ) a, e = public.ExecShell(shell) if e: return public.returnMsg(False, "Failed to unpause project: {}".format(e)) dp.write_log("Unpause [{}] success!".format(statcks_info['name'])) return public.returnMsg(True, "Successfully set!") def scan_compose_file(self, path, data): """ 递归扫描目录下的compose文件 :param path 需要扫描的目录 :param data 需要返回的数据 一个字典 :param get: :return: """ file_list = os.listdir(path) for file in file_list: current_path = os.path.join(path, file) # 判断是否是文件夹 if os.path.isdir(current_path): self.scan_compose_file(current_path, data) else: if file == "docker-compose.yaml" or file == "docker-compose.yam" or file == "docker-compose.yml": if "/www/server/panel/data/compose" in current_path: continue data.append(current_path) if ".yaml" in file or ".yam" in file or ".yml" in file: if "/www/server/panel/data/compose" in current_path: continue data.append(current_path) return data def get_compose_project(self, get): """ :param path 需要获取的路径 是一个目录 :param sub_dir 扫描子目录 :param get: :return: """ # 校验参数 try: get.validate([ Param('path').Require().SafePath(), Param('sub_dir').Require().Integer(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) data = list() suffix = ["yaml", "yam", "yml"] if get.path == "/": return public.return_message(-1, 0, _("Unable to scan the root directory")) if get.path[-1] == "/": get.path = get.path[:-1] if str(get.sub_dir) == "1": res = self.scan_compose_file(get.path, data) if not res: res = [] else: tmp = list() p_name_tmp = list() for i in res: if i.split(".")[1] not in suffix: continue project_name = i.split("/")[-1].split(".")[0] if project_name in p_name_tmp: project_name = "{}_{}".format(project_name, i.split("/")[-2]) tmp_data = { "project_name": project_name, "conf_file": "/".join(i.split("/")), "remark": "Add locally" } tmp.append(tmp_data) p_name_tmp.append(tmp_data['project_name']) res = tmp p_name_tmp.clear() else: yaml = "{}/docker-compose.yaml".format(get.path) yam = "{}/docker-compose.yam".format(get.path) yml = "{}/docker-compose.yml".format(get.path) if os.path.exists(yaml): res = [{ "project_name": get.path.split("/")[-1], "conf_file": yaml, "remark": "Add locally" }] elif os.path.exists(yam): res = [{ "project_name": get.path.split("/")[-1], "conf_file": yam, "remark": "Add locally" }] elif os.path.exists(yml): res = [{ "project_name": get.path.split("/")[-1], "conf_file": yml, "remark": "Add locally" }] else: res = list() if not os.path.isdir(get.path): return public.return_message(0, 0, res) dir_list = os.listdir(get.path) for i in dir_list: if i.rsplit(".")[-1] in suffix: res.append({ "project_name": i.rsplit(".")[0], "conf_file": "/".join(get.path.split("/") + [i]), "remark": "Add locally" }) return public.return_message(0, 0, res) # 从现有目录中添加模板 def add_template_in_path(self, get): """ :param template_list list [{"project_name":"pathtest_template","conf_file":"/www/dockerce/mysecent-project/docker-compose.yaml","remark":"描述描述"}] :param get: :return: """ create_failed = dict() create_successfully = dict() for template in get.template_list: path = template['conf_file'] name = template['project_name'] remark = template['remark'] exists = self._template_list(get) for i in exists: if name == i['name']: create_failed[name] = "Template already exists!" continue if not os.path.exists(path): create_failed[name] = "This template was not found!" continue check_res = self.check_conf(path) if not check_res['status']: create_failed[name] = "Template validation failed, possibly malformed!" continue pdata = { "name": name, "remark": remark, "path": path, "add_in_path": 1 } dp.sql("templates").insert(pdata) create_successfully[name] = "Template added successfully!" for i in create_failed: if i in create_successfully: del (create_successfully[i]) else: dp.write_log("Template added successfully from path [{}]!".format(i)) if not create_failed and create_successfully: # return {'status': True, 'msg': 'Template added successfully: [{}]'.format(','.join(create_successfully))} return public.return_message(0, 0, 'Template added successfully: [{}]'.format(','.join(create_successfully))) elif not create_successfully and create_failed: # return {'status': False, # 'msg': 'Failed to add template: template name already exists or is incorrectly formatted [{}],Use docker-compose -f [specify compose.yml file] config to check' # .format(','.join(create_failed))} return public.return_message(-1, 0, 'Failed to add template: template name already exists or is incorrectly formatted [{}],Use docker-compose -f [specify compose.yml file] config to check' .format(','.join(create_failed))) # return {'status': False, 'msg': 'These templates succeed: [{}]
These templates fail: the template name already exists or is incorrectly formatted [{}]'.format( # ','.join(create_successfully), ','.join(create_failed))} return public.return_message(-1, 0, 'These templates succeed: [{}]
These templates fail: the template name already exists or is incorrectly formatted [{}]'.format( ','.join(create_successfully), ','.join(create_failed))) def get_pull_log(self, get): """ 获取镜像拉取日志，websocket @param get: @return: """ get.wsLogTitle = "Start to pull the template image, please wait..." get._log_path = self._log_path return self.get_ws_log(get) # 编辑项目 todo 根据删除适配 def edit(self, get): """ :param project_id: 要编辑的项目的ID :param project_name: 新的项目名 :param remark: 新的描述 :param template_id: 新的模板ID :return: """ # {"project_id": 1, "template_id": 2, "project_name": "福达坊", "remark": ""} # 校验参数 try: get.validate([ Param('project_id').Require().Integer(), Param('project_name').Require().String(), Param('remark').String(), Param('template_id').Require().Integer(), ], [ public.validate.trim_filter(), ]) except Exception as ex: public.print_log("error info: {}".format(ex)) return public.return_message(-1, 0, str(ex)) # 删除旧的项目 remove_result = self.remove(get) # if not remove_result['status']: # return public.return_message(-1, 0, _("Fail to modify!")) # 创建新的项目 self.create(get) return public.return_message(0, 0, _("Modify successfully!"))