websoft9/appmanage/api/service/manage.py
2023-03-10 08:31:01 +08:00

220 lines
7.4 KiB
Python

import os, io, sys, platform, shutil, time, subprocess, json, datetime
import socket
from threading import Thread
from api.utils import shell_execute, network, docker, const
from api.model.app import App
from api.model.response import Response
# 获取所有app的信息
def get_my_app(app_name=None):
#{"name":"id",...}
ret = Response(code=const.RETURN_FAIL, message="app查询失败")
# get all info
cmd = "sudo docker compose ls -a"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
output_list = output["result"].split()
list = []
output_list = output_list[4:]
num = int(len(output_list) / 3)
list = set_app_info(output_list, num)
flag = 0
if app_name != None:
for app in list:
if app["name"] == app_name:
list.clear()
list.append(app)
flag = 1
break
if app_name == None or flag == 1:
ret = Response(code=const.RETURN_SUCCESS, message="app查询成功", data=list)
ret = ret.dict()
return ret
def set_app_info(output_list, num):
ip_result = shell_execute.execute_command_output_all("curl ifconfig.me")
ip = ip_result["result"]
list = []
for i in range(0, num):
name = output_list[3 * i] # name
image_url = "https://libs.websoft9.com/Websoft9/logo/product/" + name + "-websoft9.png"
# get trade_mark
trade_mark = ""
var_path = "/data/apps/" + name + "/variables.json"
try:
f = open(var_path, 'r', encoding='utf-8')
var = json.load(f)
try:
trade_mark = var["trademark"]
except KeyError:
pass
except FileNotFoundError:
pass
id = 0 # id
case = output_list[3 * i + 1].split("(")[0] # case
if case == "running":
case_code = const.RETURN_RUNNING # case_code
elif case == "exited":
case = "stop"
case_code = const.RETURN_STOP
elif case == "ready":
case_code = const.RETURN_READY
else:
case_code = const.RETURN_ERROR
volume = output_list[3 * i + 2] # volume
j = 2
while not volume.startswith("/"):
volume = output_list[3 * i + j]
j = j + 1
# get env info
path = "/data/apps/" + name + "/.env"
http_port = docker.read_env(path, "APP_HTTP_PORT")
db_port = docker.read_env(path, "APP_DB.*_PORT")
# get port and url
port = 0
url = "-"
admin_url = "-"
if http_port != "":
port = int(http_port)
easy_url = "http://" + ip + ":" + str(port)
url = get_url(app_name, easy_url)
admin_url = get_admin_url(app_name, url)
elif db_port != "":
port = int(db_port)
# get user_name
user_name = docker.read_env(path, "APP_USER")
if user_name == "":
user_name = "-"
# get password
password = docker.read_env(path, "POWER_PASSWORD")
if password == "":
password = "-"
app = App(id=id, name=name, status_code=case_code, status=case, port=port, volume=volume, url=url,
image_url=image_url, admin_url=admin_url, trade_mark=trade_mark, user_name=user_name, password=password)
list.append(app.dict())
return list
def get_url(app_name,easy_url):
url = easy_url
if app_name == "joomla":
url = easy_url + "/administrator"
elif app_name == "other":
url = easy_url + "/administrator"
else:
url = easy_url
return url
def get_admin_url(app_name,url):
admin_url = "-"
if app_name == "wordpress":
admin_url = url + "/wp-admin"
elif app_name == "other":
admin_url = url + "/admin"
else:
admin_url = "-"
return admin_url
def install_app_process(app_name):
if docker.check_app_directory(app_name):
percentage = docker.get_process_perc(app_name)
ret = Response(code=const.RETURN_SUCCESS, message=percentage)
ret = ret.dict()
else:
ret = Response(code=const.RETURN_FAIL , message="目前没有安装此App")
ret = ret.dict()
return ret
def install_app(app_name, app_version):
# check directory
if docker.check_app_directory(app_name):
# check port
docker.check_app_compose(app_name)
if app_version != None:
path = "/data/apps/"+app_name+"/.env"
docker.modify_env(path, "APP_VERSION", app_version)
cmd = "cd /data/apps/"+app_name+" && sudo docker compose up -d"
t1 = Thread(target=shell_execute.execute_command_output_all, args=(cmd,))
t1.start()
ret = Response(code=const.RETURN_SUCCESS, message="应用正在启动中,请过几分钟再查询")
ret = ret.dict()
else:
ret = Response(code=const.RETURN_FAIL , message="目前不支持安装此App")
ret = ret.dict()
return ret
def if_app_exits(app_name):
cmd = "docker compose ls -a | grep \'"+app_name+"\\b\'"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == -1:
return False
else:
return True
def start_app(app_name):
ret = Response(code=const.RETURN_FAIL, message="")
if if_app_exits(app_name):
docker.check_app_compose(app_name)
cmd = "docker compose -f /data/apps/"+app_name+"/docker-compose.yml start"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = const.RETURN_SUCCESS
ret.message = "应用启动成功"
else:
ret.message = "应用启动失败"
else:
ret.message = "应用不存在"
ret = ret.dict()
return ret
def stop_app(app_name):
ret = Response(code=const.RETURN_FAIL, message="")
if if_app_exits(app_name):
cmd = "docker compose -f /data/apps/"+app_name+"/docker-compose.yml stop"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = const.RETURN_SUCCESS
ret.message = "应用停止成功"
else:
ret.message = "应用停止失败"
else:
ret.message = "应用不存在"
ret = ret.dict()
return ret
def restart_app(app_name):
ret = Response(code=const.RETURN_FAIL, message="")
if if_app_exits(app_name):
cmd = "docker compose -f /data/apps/"+app_name+"/docker-compose.yml restart"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = const.RETURN_SUCCESS
ret.message = "应用重启成功"
else:
ret.message = "应用重启失败"
else:
ret.message = "应用不存在"
ret = ret.dict()
return ret
def delete_app(app_name):
ret = Response(code=const.RETURN_FAIL, message="")
if_stopped = stop_app(app_name)
if if_stopped["code"] == 0:
cmd = "docker compose -f /data/apps/"+app_name+"/docker-compose.yml down"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = 0
ret.message = "应用删除成功"
else:
ret.message = "应用删除失败"
else:
ret.message = if_stopped["message"]
ret = ret.dict()
return ret