#coding: utf-8 #------------------------------------------------------------------- # YakPanel #------------------------------------------------------------------- # Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved. #------------------------------------------------------------------- # Author: zouhw #------------------------------------------------------------------- #------------------------------ # Docker模型 #------------------------------ import public import docker.errors import projectModel.bt_docker.dk_public as dp class main: def __init__(self): self.alter_table() def alter_table(self): if not dp.sql('sqlite_master').where('type=? AND name=? AND sql LIKE ?', ('table', 'container', '%sid%')).count(): dp.sql('container').execute("alter TABLE container add container_name VARCHAR DEFAULT ''", ()) def docker_client(self,url): return dp.docker_client(url) # 添加容器 def run(self,args): """ :param name:容器名 :param image: 镜像 :param publish_all_ports 暴露所有端口 1/0 :param ports 暴露某些端口 {'1111/tcp': ('127.0.0.1', 1111)} :param command 命令 :param entrypoint 配置容器启动后执行的命令 :param environment 环境变量 xxx=xxx 一行一条 :param auto_remove 当容器进程退出时，在守护进程端启用自动移除容器。 0/1 :param args: :return: """ if not hasattr(args,'ports'): args.ports = False if not hasattr(args,'volumes'): args.volumes = False #检测端口是否已经在使用 if args.ports: for i in args.ports: if dp.check_socket(args.ports[i]): return public.returnMsg(False,"The server port [{}] has been used, please replace it with another port!".format(args.ports[i])) if not args.image: return public.returnMsg(False, "If there is no image selected, please go to the image tab to pull the image you need!") if args.restart_policy['Name'] == "always": args.restart_policy = {"Name":"always"} # return args.restart_policy # if args.cpu_quota = float(args.cpuset_cpus) * 100000 # if not args.volumes: # args.volumes = {"/sys/fs/cgroup":{"bind":"/sys/fs/cgroup","mode":"rw"}} # else: # if not "/sys/fs/cgroup" in args.volumes: # args.volumes['/sys/fs/cgroup'] = {"bind":"/sys/fs/cgroup","mode":"rw"} try: if not args.name: args.name = "{}-{}".format(args.image,public.GetRandomString(8)) if int(args.cpu_quota) / 100000 > dp.get_cpu_count(): return public.returnMsg(False,"The CPU quota has exceeded the number of cores available!") mem_limit_byte = dp.byte_conversion(args.mem_limit) if mem_limit_byte > dp.get_mem_info(): return public.returnMsg(False, "The memory quota has exceeded the available number!") res = self.docker_client(args.url).containers.run( name=args.name, image=args.image, detach=True, publish_all_ports=True if args.publish_all_ports == "1" else False, ports=args.ports if args.ports else None, command=args.command, auto_remove=True if str(args.auto_remove) == "1" else False, environment=dp.set_kv(args.environment), #"HOME=/value\nHOME11=value1" volumes=args.volumes, #一个字典对象 {'服务器路径/home/user1/': {'bind': '容器路径/mnt/vol2', 'mode': 'rw'},'/var/www': {'bind': '/mnt/vol1', 'mode': 'ro'}} # cpuset_cpus=args.cpuset_cpus ,#指定容器使用的cpu个数 cpu_quota=int(args.cpu_quota), mem_limit=args.mem_limit, #b,k,m,g restart_policy=args.restart_policy, labels=dp.set_kv(args.labels), #"key=value\nkey1=value1" tty=True, stdin_open=True, privileged=True ) if res: pdata = { "cpu_limit": str(args.cpu_quota), "container_name": args.name } dp.sql('container').insert(pdata) public.set_module_logs('docker', 'run_container', 1) dp.write_log("Create container [{}] successful!".format(args.name)) return public.returnMsg(True,"The container was created successfully!") return public.returnMsg(False, 'Create failed!') except docker.errors.APIError as e: if "container to be able to reuse that name." in str(e): return public.returnMsg(False, "The container name already exists!") if "Invalid container name" in str(e): return public.returnMsg(False, "The container name is illegal, please do not use Chinese container name!") if "bind: address already in use" in str(e): port = "" for i in args.ports: if ":{}:".format(args.ports[i]) in str(e): port = args.ports[i] args.id = args.name self.del_container(args) return public.returnMsg(False, "Server port {} is in use! Please change other ports".format(port)) return public.returnMsg(False, 'Create failed! {}'.format(public.get_error_info())) # 保存为镜像 def commit(self,args): """ :param repository 推送到的仓库 :param tag 镜像标签 jose:v1 :param message 提交的信息 :param author 镜像作者 :param changes :param conf dict :param path 导出路径 :param name 导出文件名 :param args: :return: """ if not hasattr(args,'conf') or not args.conf: args.conf = None if args.repository == "docker.io": args.repository = "" container = self.docker_client(args.url).containers.get(args.id) container.commit( repository=args.repository if args.repository else None, tag=args.tag if args.tag else None, message=args.message if args.message else None, author=args.author if args.author else None, # changes=args.changes if args.changes else None, conf=args.conf ) if hasattr(args,"path") and args.path: args.id = "{}:{}".format(args.name,args.tag) import projectModel.bt_docker.dk_image as dk return dk.main().save(args) dp.write_log("Submitting container [{}] as image [{}] succeeded!".format(container.attrs['Name'],args.tag)) return public.returnMsg(True,"提交成功！") # 容器执行命令 def docker_shell(self, args): """ :param container_id :param args: :return: """ try: self.docker_client(args.url).containers.get(args.container_id) cmd = 'docker container exec -it {} /bin/bash'.format(args.container_id) return public.returnMsg(True, cmd) except docker.errors.APIError as ex: return public.returnMsg(False, 'Failed to get container') # 导出容器为tar 没有导入方法，目前弃用 def export(self,args): """ :param path 保存路径 :param name 包名 :param args: :return: """ from os import path as ospath from os import makedirs as makedirs try: if "tar" in args.name: file_name = '{}/{}'.format(args.path,args.name) else: file_name = '{}/{}.tar'.format(args.path, args.name) if not ospath.exists(args.path): makedirs(args.path) public.writeFile(file_name,'') f = open(file_name, 'wb') container = self.docker_client(args.url).containers.get(args.id) data = container.export() for i in data: f.write(i) f.close() return public.returnMsg(True, "Successfully exported to: {}".format(file_name)) except: return public.returnMsg(False, 'Operation failed:' + str(public.get_error_info())) # 删除容器 def del_container(self,args): """ :return: """ import projectModel.bt_docker.dk_public as dp container = self.docker_client(args.url).containers.get(args.id) container.remove(force=True) dp.sql("cpu_stats").where("container_id=?", (args.id,)).delete() dp.sql("io_stats").where("container_id=?", (args.id,)).delete() dp.sql("mem_stats").where("container_id=?", (args.id,)).delete() dp.sql("net_stats").where("container_id=?", (args.id,)).delete() dp.sql("container").where("container_nam=?", (container.attrs['Name'])).delete() dp.write_log("Delete container [{}] succeeded!".format(container.attrs['Name'])) return public.returnMsg(True,"Successfully deleted!") # 设置容器状态 def set_container_status(self,args): import time container = self.docker_client(args.url).containers.get(args.id) if args.act == "start": container.start() elif args.act == "stop": container.stop() elif args.act == "pause": container.pause() elif args.act == "unpause": container.unpause() elif args.act == "reload": container.reload() else: container.restart() time.sleep(1) tmp = self.docker_client(args.url).containers.get(args.id) return {"name":container.attrs['Name'].replace('/',''),"status":tmp.attrs['State']['Status']} #返回设置后的状态 # 停止容器 def stop(self,args): """ :param url :param id :param args: :return: """ try: args.act = "stop" data = self.set_container_status(args) if data['status'] != "exited": return public.returnMsg(False, "Stop failing!") dp.write_log("Stop container [{}] succeeded!".format(data['name'])) return public.returnMsg(True, "Stop success!") except docker.errors.APIError as e: if "is already paused" in str(e): return public.returnMsg(False,"The container has been suspended!") if "No such container" in str(e): return public.returnMsg(True, "The container has been stopped and deleted because the container has the option to automatically delete after stopping!") return public.returnMsg(False,"Stop failing!{}".format(e)) def start(self,args): """ :param url :param id :param args: :return: """ try: args.act = "start" data = self.set_container_status(args) if data['status'] != "running": return public.returnMsg(False, "Startup failed!") dp.write_log("Start the container [{}] successfully!".format(data['name'])) return public.returnMsg(True, "Started successfully!") except docker.errors.APIError as e: if "cannot start a paused container, try unpause instead" in str(e): return self.unpause(args) def pause(self,args): """ Pauses all processes within this container. :param url :param id :param args: :return: """ try: args.act = "pause" data = self.set_container_status(args) if data['status'] != "paused": return public.returnMsg(False, "Container pause failed!") dp.write_log("Suspended container [{}] succeeded!".format(data['name'])) return public.returnMsg(True, "Container paused successfully!") except docker.errors.APIError as e: if "is already paused" in str(e): return public.returnMsg(False,"The container has been suspended!") if "is not running" in str(e): return public.returnMsg(False, "The container is not started and cannot be paused!") if "is not paused" in str(e): return public.returnMsg(False, "The container has not been suspended!") return str(e) def unpause(self,args): """ unPauses all processes within this container. :param url :param id :param args: :return: """ try: args.act = "unpause" data = self.set_container_status(args) if data['status'] != "running": return public.returnMsg(False, "Startup failed!") dp.write_log("Unpausing the container [{}] succeeded!".format(data['name'])) return public.returnMsg(True, "Container unpause succeeded!") except docker.errors.APIError as e: if "is already paused" in str(e): return public.returnMsg(False,"The container has been suspended!") if "is not running" in str(e): return public.returnMsg(False, "The container is not started and cannot be paused!") if "is not paused" in str(e): return public.returnMsg(False, "The container has not been suspended!") return str(e) def reload(self,args): """ Load this object from the server again and update attrs with the new data. :param url :param id :param args: :return: """ args.act = "reload" data = self.set_container_status(args) if data['status'] != "running": return public.returnMsg(False, "Startup failed!") dp.write_log("Reloading container [{}] succeeded!".format(data['name'])) return public.returnMsg(True, "Container reload succeeded!") def restart(self,args): """ Restart this container. Similar to the docker restart command. :param url :param id :param args: :return: """ args.act = "restart" data = self.set_container_status(args) if data['status'] != "running": return public.returnMsg(False, "Startup failed!") dp.write_log("Restarting the container [{}] succeeded!".format(data['name'])) return public.returnMsg(True, "The container restarted successfully!") def get_container_ip(self,container_networks): data = list() for network in container_networks: data.append(container_networks[network]['IPAddress']) return data def get_container_path(self,detail): import os if not "GraphDriver" in detail: return False if "Data" not in detail["GraphDriver"]: return False if "MergedDir" not in detail["GraphDriver"]["Data"]: return False path = detail["GraphDriver"]["Data"]["MergedDir"] if not os.path.exists(path): return "" return path # 获取容器列表所需的外部数据 def get_other_data_for_container_list(self,args): import projectModel.bt_docker.dk_image as di import projectModel.bt_docker.dk_volume as dv import projectModel.bt_docker.dk_compose as dc import projectModel.bt_docker.dk_setup as ds # 获取镜像列表 images = di.main().image_list(args) if images['status']: images = images['msg']['images_list'] else: images = list() # 获取卷列表 volumes = dv.main().get_volume_list(args) if volumes['status']: volumes = volumes['msg']['volume'] else: volumes = list() # 获取模板列表 template = dc.main().template_list(args) if template['status']: template = template['msg']['template'] else: template = list() online_cpus = dp.get_cpu_count() mem_total = dp.get_mem_info() docker_setup = ds.main() return { "images":images, "volumes":volumes, "template":template, "online_cpus":online_cpus, "mem_total":mem_total, "installed":docker_setup.check_docker_program(), "service_status":docker_setup.get_service_status() } # 获取容器列表 def get_list(self,args): """ :param url :return: """ # 判断docker是否安装 import projectModel.bt_docker.dk_setup as ds data = self.get_other_data_for_container_list(args) if not ds.main().check_docker_program(): data['container_list'] = list() return public.returnMsg(True,data) client = self.docker_client(args.url) if not client: return public.returnMsg(True,data) containers = client.containers attr_list = self.get_container_attr(containers) # data = self.get_other_data_for_container_list(args) container_detail = list() for attr in attr_list: cpu_usage = dp.sql("cpu_stats").where("container_id=?",(attr["Id"],)).select() if cpu_usage and isinstance(cpu_usage,list): cpu_usage = cpu_usage[-1]['cpu_usage'] else: cpu_usage = "0.0" tmp = { "id": attr["Id"], "name": attr['Name'].replace("/",""), "status": attr["State"]["Status"], "image": attr["Config"]["Image"], "time": attr["Created"], "merged": self.get_container_path(attr), "ip": self.get_container_ip(attr["NetworkSettings"]['Networks']), "ports": attr["NetworkSettings"]["Ports"], "detail": attr, "cpu_usage":cpu_usage if attr["State"]["Status"] == "running" else "" } container_detail.append(tmp) data['container_list'] = container_detail return public.returnMsg(True,data) # 获取容器的attr def get_container_attr(self,containers): c_list = containers.list(all=True) return [container_info.attrs for container_info in c_list] # 获取容器日志 def get_logs(self,args): """ :param url :param id :param args: :return: """ try: container = self.docker_client(args.url).containers.get(args.id) res = container.logs().decode() return public.returnMsg(True,res) except docker.errors.APIError as e: if "configured logging driver does not support reading" in str(e): return public.returnMsg(False,"The container has no log files!") # 登录容器 # 获取容器配置文件