Initial YakPanel commit
This commit is contained in:
397
mod/project/docker/comMod.py
Normal file
397
mod/project/docker/comMod.py
Normal file
@@ -0,0 +1,397 @@
|
||||
# coding: utf-8
|
||||
# -------------------------------------------------------------------
|
||||
# YakPanel
|
||||
# -------------------------------------------------------------------
|
||||
# Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved.
|
||||
# -------------------------------------------------------------------
|
||||
# Author: wzz <wzz@yakpanel.com>
|
||||
# -------------------------------------------------------------------
|
||||
# ------------------------------
|
||||
# docker模型
|
||||
# ------------------------------
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
|
||||
if "/www/server/panel/class" not in sys.path:
|
||||
sys.path.insert(0, "/www/server/panel/class")
|
||||
|
||||
os.chdir("/www/server/panel")
|
||||
import public
|
||||
|
||||
from mod.project.docker.app.appManageMod import AppManage
|
||||
# from mod.project.docker.runtime.runtimeManage import RuntimeManage
|
||||
# from mod.project.docker.sites.sitesManage import SitesManage
|
||||
from mod.project.docker.app.sub_app.ollamaMod import OllamaMod
|
||||
from mod.project.docker.apphub.apphubManage import AppHub
|
||||
from btdockerModelV2 import dk_public as dp
|
||||
|
||||
|
||||
class main(AppManage, OllamaMod):
|
||||
|
||||
def __init__(self):
|
||||
super(main, self).__init__()
|
||||
OllamaMod.__init__(self)
|
||||
|
||||
# 2024/6/26 下午5:49 获取所有已部署的项目列表
|
||||
def get_project_list(self, get):
|
||||
'''
|
||||
@name 获取所有已部署的项目列表
|
||||
@author wzz <2024/6/26 下午5:49>
|
||||
@param "data":{"参数名":""} <数据类型> 参数描述
|
||||
@return dict{"status":True/False,"msg":"提示信息"}
|
||||
'''
|
||||
try:
|
||||
if self.def_name is None: self.set_def_name(get.def_name)
|
||||
if hasattr(get, '_ws') and hasattr(get._ws, 'btws_get_project_list'):
|
||||
return
|
||||
|
||||
while True:
|
||||
compose_list = self.ls(get)
|
||||
if len(compose_list) == 0:
|
||||
if hasattr(get, '_ws'):
|
||||
get._ws.send(json.dumps(self.wsResult(
|
||||
True,
|
||||
data=[],
|
||||
)))
|
||||
|
||||
|
||||
stacks_info = dp.sql("stacks").select()
|
||||
|
||||
compose_project = []
|
||||
|
||||
for j in compose_list:
|
||||
t_status = j["Status"].split(",")
|
||||
container_count = 0
|
||||
for ts in t_status:
|
||||
container_count += int(ts.split("(")[1].split(")")[0])
|
||||
|
||||
j_name = j['Name']
|
||||
if "bt_compose_" in j_name:
|
||||
config_path = "{}/config/name_map.json".format(public.get_panel_path())
|
||||
name_map = json.loads(public.readFile(config_path))
|
||||
if j_name in name_map:
|
||||
j_name = name_map[j_name]
|
||||
else:
|
||||
j_name = j_name.replace("bt_compose_", "")
|
||||
|
||||
tmp = {
|
||||
"id": None,
|
||||
"name": j_name,
|
||||
"status": "1",
|
||||
"path": j['ConfigFiles'],
|
||||
"template_id": None,
|
||||
"time": None,
|
||||
"remark": "",
|
||||
"run_status": j['Status'].split("(")[0].lower(),
|
||||
"container_count": container_count,
|
||||
}
|
||||
for i in stacks_info:
|
||||
if public.md5(i['name']) in j['Name']:
|
||||
|
||||
tmp["name"] = i['name']
|
||||
tmp["run_status"] = j['Status'].split("(")[0].lower()
|
||||
tmp["template_id"] = i['template_id']
|
||||
tmp["time"] = i['time']
|
||||
tmp["remark"] = i["remark"]
|
||||
tmp["id"] = i['id']
|
||||
break
|
||||
|
||||
if i['name'] == j['Name']:
|
||||
tmp["run_status"] = j['Status'].split("(")[0].lower()
|
||||
tmp["template_id"] = i['template_id']
|
||||
tmp["time"] = i['time']
|
||||
tmp["remark"] = i["remark"]
|
||||
tmp["id"] = i['id']
|
||||
break
|
||||
|
||||
if tmp["time"] is None:
|
||||
if os.path.exists(j['ConfigFiles']):
|
||||
get.path = j['ConfigFiles']
|
||||
compose_ps = self.ps(get)
|
||||
if len(compose_ps) > 0 and "CreatedAt" in compose_ps[0]:
|
||||
tmp["time"] = dp.convert_timezone_str_to_timestamp(compose_ps[0]['CreatedAt'])
|
||||
|
||||
compose_project.append(tmp)
|
||||
|
||||
if hasattr(get, '_ws'):
|
||||
setattr(get._ws, 'btws_get_project_list', True)
|
||||
get._ws.send(json.dumps(self.wsResult(
|
||||
True,
|
||||
data=sorted(compose_project, key=lambda x: x["time"] if x["time"] is not None else float('-inf'), reverse=True),
|
||||
)))
|
||||
|
||||
time.sleep(2)
|
||||
except Exception as e:
|
||||
return public.return_message(-1, 0, str(e))
|
||||
|
||||
# 2026/2/26 下午8:55 获取指定compose.yml的docker-compose ps 增加异常处理和类型检查,防止WebSocket推送过程中出现错误导致循环崩溃
|
||||
def get_project_ps(self, get):
|
||||
'''
|
||||
@name 获取指定 compose.yml 的 docker-compose ps 实时状态(WebSocket 推送)
|
||||
@author wzz <2026/2/26>
|
||||
@param get.path: compose 项目路径
|
||||
@return dict {"status": True/False, "msg": "..."}
|
||||
'''
|
||||
try:
|
||||
if self.def_name is None:
|
||||
self.set_def_name(get.def_name)
|
||||
|
||||
# 确保 path 存在
|
||||
if not hasattr(get, 'path') or not get.path:
|
||||
return public.return_message(-1, 0, "Missing 'path' parameter")
|
||||
|
||||
ws_flag_attr = f'btws_get_project_ps_{get.path}'
|
||||
|
||||
# 防止重复订阅
|
||||
if hasattr(get, '_ws') and hasattr(get._ws, ws_flag_attr):
|
||||
result = self.wsResult(True, data=[])
|
||||
try:
|
||||
get._ws.send(json.dumps(result))
|
||||
except:
|
||||
pass
|
||||
return result
|
||||
|
||||
from btdockerModelV2.dockerSock import container
|
||||
sk_container = container.dockerContainer()
|
||||
|
||||
while True:
|
||||
try:
|
||||
compose_list = self.ps(get)
|
||||
# 强制确保是 list
|
||||
if not isinstance(compose_list, list):
|
||||
compose_list = []
|
||||
|
||||
# 发送空结果并退出(无容器)
|
||||
if len(compose_list) == 0:
|
||||
if hasattr(get, '_ws'):
|
||||
try:
|
||||
get._ws.send(json.dumps(self.wsResult(True, data=[])))
|
||||
except:
|
||||
pass
|
||||
break
|
||||
|
||||
# 处理每个容器
|
||||
for l in compose_list:
|
||||
if not isinstance(l, dict):
|
||||
continue
|
||||
|
||||
# 补全 Image
|
||||
if "Image" not in l:
|
||||
l["Image"] = ""
|
||||
if l.get("ID"):
|
||||
try:
|
||||
inspect = sk_container.get_container_inspect(l["ID"])
|
||||
l["Image"] = inspect.get("Config", {}).get("Image", "")
|
||||
except:
|
||||
pass # 忽略 inspect 失败
|
||||
|
||||
# 补全 Ports 字符串
|
||||
if "Ports" not in l:
|
||||
l["Ports"] = ""
|
||||
publishers = l.get("Publishers")
|
||||
if publishers and isinstance(publishers, list):
|
||||
for p in publishers:
|
||||
if not isinstance(p, dict):
|
||||
continue
|
||||
url = p.get("URL", "")
|
||||
target = p.get("TargetPort", "")
|
||||
proto = p.get("Protocol", "")
|
||||
pub_port = p.get("PublishedPort", "")
|
||||
if url == "":
|
||||
l["Ports"] += f"{target}/{proto},"
|
||||
else:
|
||||
l["Ports"] += f"{url}:{pub_port}->{target}/{proto},"
|
||||
|
||||
# 构造结构化 ports(兼容 containerModel.struct_container_ports)
|
||||
ports_data = {}
|
||||
publishers = l.get("Publishers")
|
||||
if publishers and isinstance(publishers, list):
|
||||
for port in publishers:
|
||||
if not isinstance(port, dict):
|
||||
continue
|
||||
key = f"{port.get('TargetPort', '')}/{port.get('Protocol', '')}"
|
||||
host_ip = port.get("URL", "")
|
||||
host_port = str(port.get("PublishedPort", ""))
|
||||
entry = {"HostIp": host_ip, "HostPort": host_port}
|
||||
if key not in ports_data:
|
||||
ports_data[key] = [entry] if host_ip else None
|
||||
elif ports_data[key] is not None:
|
||||
ports_data[key].append(entry)
|
||||
l["ports"] = ports_data
|
||||
|
||||
# 推送数据
|
||||
if hasattr(get, '_ws'):
|
||||
setattr(get._ws, ws_flag_attr, True)
|
||||
try:
|
||||
get._ws.send(json.dumps(self.wsResult(True, data=compose_list)))
|
||||
except:
|
||||
# WebSocket 已断开,退出循环
|
||||
break
|
||||
|
||||
time.sleep(2)
|
||||
|
||||
except Exception:
|
||||
# 内部循环异常,安全退出
|
||||
break
|
||||
|
||||
return self.wsResult(True, data=[])
|
||||
|
||||
except Exception as e:
|
||||
return public.return_message(-1, 0, str(e))
|
||||
|
||||
# 2024/11/11 14:34 获取所有正在运行的容器信息和已安装的应用信息
|
||||
def get_some_info(self, get):
|
||||
'''
|
||||
@name 获取所有正在运行的容器信息和已安装的应用信息
|
||||
'''
|
||||
get.type = get.get("type", "container")
|
||||
if not get.type in ("container", "app"):
|
||||
return public.return_message(-1, 0, public.lang("Only container and app types are supported"))
|
||||
|
||||
if get.type == "container":
|
||||
from btdockerModelV2.dockerSock import container
|
||||
sk_container = container.dockerContainer()
|
||||
sk_container_list = sk_container.get_container()
|
||||
|
||||
data = []
|
||||
for container in sk_container_list:
|
||||
if not "running" in container["State"]: continue
|
||||
|
||||
port_list = []
|
||||
for p in container["Ports"]:
|
||||
if not "PublicPort" in p: continue
|
||||
if not p["PublicPort"] in port_list:
|
||||
port_list.append(p["PublicPort"])
|
||||
|
||||
data.append({
|
||||
"id": container["Id"],
|
||||
"name": container["Names"][0].replace("/", ""),
|
||||
"status": container["State"],
|
||||
"image": container["Image"],
|
||||
"created_time": container["Created"],
|
||||
"ports": port_list,
|
||||
})
|
||||
|
||||
return public.return_message(0, 0, data)
|
||||
else:
|
||||
get.row = 10000
|
||||
installed_apps = self.get_installed_apps(get)['message']
|
||||
not_allow_category = ("Database", "System")
|
||||
if installed_apps and installed_apps.get('data', []):
|
||||
for app in installed_apps["data"]:
|
||||
if not "running" in app["status"]:
|
||||
installed_apps["data"].remove(app)
|
||||
if app["apptype"] in not_allow_category:
|
||||
installed_apps["data"].remove(app) if app in installed_apps["data"] else None
|
||||
|
||||
# return public.returnResult(status=installed_apps["status"], data=installed_apps["data"])
|
||||
return public.return_message(0, 0, installed_apps)
|
||||
|
||||
def generate_apphub(self, get):
|
||||
'''
|
||||
@name 解析外部应用列表
|
||||
@author csj <2025/7/9>
|
||||
@return dict{"status":True/False,"msg":"提示信息"}
|
||||
'''
|
||||
return AppHub().generate_apphub(get)
|
||||
|
||||
def create_app(self,get):
|
||||
'''
|
||||
@name 创建应用
|
||||
@author csj <2025/7/9>
|
||||
@return dict{"status":True/False,"msg":"提示信息"}
|
||||
'''
|
||||
if get.get("appid","0") == "-1": # 从apphub创建应用
|
||||
self.templates_path = os.path.join(AppHub.hub_home_path, "templates")
|
||||
self.apps_json_file = os.path.join(AppHub.hub_home_path, "apps.json")
|
||||
|
||||
version = get.get("version","latest")
|
||||
app_name = get.get("app_name","")
|
||||
|
||||
if not os.path.exists(self.templates_path):
|
||||
os.makedirs(self.templates_path)
|
||||
|
||||
#/www/dk_project/dk_app/apphub/apphub/templates/app_name/version
|
||||
app_version_path = os.path.join(AppHub.hub_home_path, app_name, version)
|
||||
if not os.path.exists(app_version_path):
|
||||
return public.return_message(-1, 0, public.lang("Version {} for applying {} does not exist", (version, app_name)))
|
||||
|
||||
# /www/dk_project/dk_app/apphub/apphub/templates/app_name
|
||||
app_template_path = os.path.join(self.templates_path, app_name)
|
||||
|
||||
public.ExecShell("\cp -r {} {}".format(app_version_path,app_template_path))
|
||||
|
||||
return super().create_app(get)
|
||||
|
||||
def get_apphub_config(self, get):
|
||||
'''
|
||||
@name 获取apphub配置
|
||||
@author csj <2025/7/9>
|
||||
@return dict{"status":True/False,"data":{}}
|
||||
'''
|
||||
return public.return_message(0, 0, AppHub.get_config())
|
||||
|
||||
def set_apphub_git(self,get):
|
||||
'''
|
||||
@name 设置外部应用的git地址
|
||||
@author csj <2025/7/9>
|
||||
@param get: git_url, git_branch, user, password
|
||||
'''
|
||||
if not hasattr(get, 'git_url') or not get.git_url:
|
||||
return public.return_message(-1, 0, public.lang("GIT ADDRESS IS NOT SET"))
|
||||
if not hasattr(get, 'git_branch') or not get.git_branch:
|
||||
return public.return_message(-1, 0, public.lang("The branch name is not set"))
|
||||
|
||||
return AppHub().set_apphub_git(get)
|
||||
|
||||
def import_git_apphub(self,get):
|
||||
'''
|
||||
@name 从git导入外部应用
|
||||
@author csj <2025/7/9>
|
||||
'''
|
||||
return AppHub().import_git_apphub(get)
|
||||
|
||||
def install_apphub(self,get):
|
||||
'''
|
||||
@name 安装apphub所需环境
|
||||
@author csj <2025/7/9>
|
||||
'''
|
||||
return AppHub().install_apphub(get)
|
||||
|
||||
def import_zip_apphub(self,get):
|
||||
'''
|
||||
@name 从zip包导入外部应用
|
||||
@author csj <2025/7/9>
|
||||
@param get: sfile: zip文件路径
|
||||
'''
|
||||
if not hasattr(get, 'sfile') or not get.sfile:
|
||||
return public.return_message(-1, 0, public.lang("The zip file path is not set"))
|
||||
|
||||
return AppHub().import_zip_apphub(get)
|
||||
|
||||
def parser_zip_apphub(self,get):
|
||||
'''
|
||||
@name 解析zip包
|
||||
@author csj <2025/7/9>
|
||||
@param get: sfile: zip文件路径
|
||||
@return dict{"status":True/False,"data":[]}
|
||||
'''
|
||||
if not hasattr(get, 'sfile') or not get.sfile:
|
||||
return public.return_message(-1, 0, public.lang("Please select the file path"))
|
||||
|
||||
app_list = []
|
||||
files = get.sfile.split(',')
|
||||
for sfile in files:
|
||||
get.sfile = sfile
|
||||
|
||||
apps = AppHub().parser_zip_apphub(get)
|
||||
app_list.extend(apps)
|
||||
public.print_log(app_list)
|
||||
return public.return_message(0, 0, app_list)
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
pass
|
||||
Reference in New Issue
Block a user