398 lines
16 KiB
Python
398 lines
16 KiB
Python
|
# 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
|