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

419 lines
15 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import os
import io
import sys
import platform
import shutil
import time
import subprocess
import json
import datetime
import socket
import re
from threading import Thread
from api.utils import shell_execute, docker, const
from api.model.app import App
from api.model.response import Response
from api.utils import lock
from api.utils.common_log import myLogger
# 获取所有app的信息
def get_my_app():
ret = Response(code=const.RETURN_FAIL, message="App query failed!")
# get all info
cmd = "docker compose ls -a --format json"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
output_list = json.loads(output["result"])
app_list, has_add = get_apps_from_compose(output_list)
list = get_apps_from_queue(app_list, has_add)
ret = Response(code=const.RETURN_SUCCESS, message="The app query is successful.", data=list)
ret = ret.dict()
return ret
# 获取具体某个app的信息
def get_app_detail(app_id):
ret = {}
ret['code'] = const.RETURN_FAIL
ret['message'] = 'App query failed!'
ret['data'] = None
if docker.check_app_id(app_id):
# get all info
app_name = split_app_id(app_id)
info, code = if_app_exits(app_id)
if code:
cmd = "docker compose ls -a --format json"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
output_list = json.loads(output["result"])
app_list, has_add = get_apps_from_compose(output_list)
list = get_apps_from_queue(app_list, has_add)
flag = 0
app_info = None
for app in list:
if app["app_id"] == app_id:
list.clear()
list.append(app)
app_info = app
flag = 1
break
if flag == 1:
ret['code'] = const.RETURN_SUCCESS
ret['message'] = "The app query is successful."
ret['data'] = app_info
else:
ret['message'] = 'This app is not currently installed.'
else:
ret['message'] = "AppID is not legal!"
return ret
# 查询某个正在安装的app的 具体状态waiting等待安装pulling拉取镜像initializing初始化running正常运行
def install_app_process(app_id):
ret = Response(code=const.RETURN_FAIL, message=" ")
app_name = split_app_id(app_id)
if docker.check_app_id(app_id):
var_path = "/data/apps/" + app_name + "/variables.json"
real_name = docker.read_var(var_path, 'name')
info, code = if_app_exits(app_id)
if code:
percentage = docker.get_process_perc(app_name, real_name)
ret.code = const.RETURN_SUCCESS
ret.message = percentage
else:
ret.message = "This app is not currently installed."
else:
ret.message = "AppID is not legal!"
ret = ret.dict()
return ret
def install_app(app_name, customer_app_name, app_version):
ret = Response(code=const.RETURN_FAIL, message=" ")
ret.code, ret.message = check_app(app_name, customer_app_name, app_version)
if ret.code == const.RETURN_SUCCESS:
ret.code, ret.message = prepare_app(app_name, customer_app_name)
if ret.code == const.RETURN_SUCCESS:
t1 = Thread(target=install_app_job, args=(customer_app_name, app_version,))
t1.start()
ret.message = "The app is starting, please check again in a few minutes."
ret = ret.dict()
return ret
def start_app(app_id):
ret = Response(code=const.RETURN_FAIL, message="")
if docker.check_app_id(app_id):
app_name = split_app_id(app_id)
info, code = if_app_exits(app_id)
if code:
app_path = info.split()[-1].rsplit('/', 1)[0]
docker.check_app_compose(app_path + '/.env')
cmd = "docker compose -f " + app_path + "/docker-compose.yml start"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = const.RETURN_SUCCESS
ret.message = "The app starts successfully."
else:
ret.message = "The app failed to start!"
else:
ret.message = "This app is not currently installed."
else:
ret.message = "AppID is not legal!"
ret = ret.dict()
return ret
def stop_app(app_id):
ret = Response(code=const.RETURN_FAIL, message="")
if docker.check_app_id(app_id):
app_name = split_app_id(app_id)
info, code = if_app_exits(app_id)
if code:
app_path = info.split()[-1].rsplit('/', 1)[0]
cmd = "docker compose -f " + app_path + "/docker-compose.yml stop"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = const.RETURN_SUCCESS
ret.message = "The app stopped successfully."
else:
ret.message = "App stop failed!"
else:
ret.message = "This app is not currently installed."
else:
ret.message = 'AppID is not legal!'
ret = ret.dict()
return ret
def restart_app(app_id):
ret = Response(code=const.RETURN_FAIL, message="")
if docker.check_app_id(app_id):
app_name = split_app_id(app_id)
info, code = if_app_exits(app_id)
if code:
app_path = info.split()[-1].rsplit('/', 1)[0]
cmd = "docker compose -f " + app_path + "/docker-compose.yml restart"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = const.RETURN_SUCCESS
ret.message = "The app restarts successfully."
else:
ret.message = "App restart failed!"
else:
ret.message = "This app is not currently installed."
else:
ret.message = 'AppID is not legal!'
ret = ret.dict()
return ret
def uninstall_app(app_id):
ret = Response(code=const.RETURN_FAIL, message="")
if docker.check_app_id(app_id):
app_name = split_app_id(app_id)
info, code = if_app_exits(app_id)
if code:
app_path = info.split()[-1].rsplit('/', 1)[0]
if_stopped = stop_app(app_id) # stop_app
app_name = split_app_id(app_id)
if if_stopped["code"] == 0:
cmd = "docker compose -f " + app_path + "/docker-compose.yml down -v"
lib_path = '/data/library/apps/' + app_name
if app_path != lib_path:
cmd = cmd + " && sudo rm -rf " + app_path
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
ret.code = 0
ret.message = "The app is deleted successfully"
else:
ret.message = "App deletion failed!"
else:
ret.message = if_stopped["message"]
else:
ret.message = 'AppID is not legal!'
ret = ret.dict()
return ret
def check_app(app_name, customer_app_name, app_version):
message = " "
code = const.RETURN_FAIL
if app_name == None or customer_app_name == None or app_version == None:
message = "Please fill in the APP information completely!"
elif not docker.check_app_directory(app_name):
message = "Installing the app is not supported!"
elif re.match('^[a-z0-9]+$', customer_app_name) == None:
message = "App names must be lowercase letters and numbers!"
elif docker.check_directory("/data/apps/" + customer_app_name):
message = "The APP name is already in use, please specify a different name to reinstall."
elif not docker.check_vm_resource(app_name):
message = "System resources (memory, CPU, disk) are insufficient, and continuing to install may cause the app to not run or the server to be abnormal!"
else:
code = const.RETURN_SUCCESS
return code, message
def prepare_app(app_name, customer_app_name):
library_path = "/data/library/apps/" + app_name
install_path = "/data/apps/" + customer_app_name
message = " "
code = const.RETURN_SUCCESS
output = shell_execute.execute_command_output_all("cp -r " + library_path + " " + install_path)
if int(output["code"]) != 0:
message = "creating" + customer_app_name + "directory failed!"
code = const.RETURN_FAIL
return code, message
def install_app_job(customer_app_name, app_version):
# write running_apps.txt
file_path = "/data/apps/running_apps.txt"
shell_execute.execute_command_output_all("echo " + customer_app_name + " >> " + file_path)
# modify env
env_path = "/data/apps/" + customer_app_name + "/.env"
docker.modify_env(env_path, 'APP_NAME', customer_app_name)
docker.modify_env(env_path, "APP_VERSION", app_version)
# check port
docker.check_app_compose(env_path)
# modify running_apps.txt
cmd = "cd /data/apps/" + customer_app_name + " && sudo docker compose up --pull always -d"
shell_execute.execute_command_output_all(cmd)
# delete
output = shell_execute.execute_command_output_all("sed -n \'/^" + customer_app_name + "/=\' " + file_path)
if int(output["code"]) == 0 and output["result"] != "":
line_num = output["result"].split("\n")[0]
shell_execute.execute_command_output_all("sed -i \'" + line_num + "d\' " + file_path)
def if_app_exits(app_id):
app_name = app_id.split('_')[1]
real_name = app_id.split('_')[0]
flag = False
info = ""
cmd = "docker compose ls -a | grep \'/" + app_name + "/\'"
output = shell_execute.execute_command_output_all(cmd)
if int(output["code"]) == 0:
info = output["result"]
app_path = info.split()[-1].rsplit('/', 1)[0]
is_official = check_if_official_app(app_path + '/variables.json')
if is_official:
name = docker.read_var(app_path + '/variables.json', 'name')
if name == real_name:
flag = True
elif real_name == app_name:
flag = True
myLogger.info_logger("APP info: " + info)
return info, flag
def split_app_id(app_id):
return app_id.split("_")[1]
def get_apps_from_compose(output_list):
ip_result = shell_execute.execute_command_output_all("curl ifconfig.me")
ip = ip_result["result"]
app_list = []
has_add = []
for app_info in output_list:
volume = app_info["ConfigFiles"] # volume
app_path = volume.rsplit('/', 1)[0]
app_name = volume.split('/')[-2]
app_id = app_name + "_" + app_name # app_id
real_name = ""
trade_mark = ""
port = 0
url = ""
admin_url = ""
image_url = ""
user_name = ""
password = ""
official_app = False
if app_name in ['appmanage', 'nginxproxymanager']:
continue
# get code
case = app_info["Status"].split("(")[0] # case
if case == "running":
case_code = const.APP_RUNNING # case_code
elif case == "exited":
case = "stop"
case_code = const.APP_STOP
elif case == "created":
case_code = const.APP_READY
case = "installing"
else:
case_code = const.APP_ERROR
var_path = app_path + "/variables.json"
official_app = check_if_official_app(var_path)
if official_app:
real_name = docker.read_var(var_path, 'name')
app_id = real_name + "_" + app_name # app_id
# get trade_mark
trade_mark = docker.read_var(var_path, 'trademark')
image_url = get_Image_url(real_name)
# get env info
path = app_path + "/.env"
# get port and url
try:
http_port = list(docker.read_env(
path, "APP_HTTP_PORT").values())[0]
port = int(http_port)
easy_url = "http://" + ip + ":" + str(port)
url = get_url(real_name, easy_url)
admin_url = get_admin_url(real_name, url)
except IndexError:
try:
db_port = list(docker.read_env(
path, "APP_DB.*_PORT").values())[0]
port = int(db_port)
except IndexError:
pass
# get user_name
try:
user_name = list(docker.read_env(path, "APP_USER").values())[0]
except IndexError:
pass
# get password
try:
password = list(docker.read_env(
path, "POWER_PASSWORD").values())[0]
except IndexError:
pass
has_add.append(app_name)
app = App(app_id=app_id, name=real_name, customer_name=app_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, official_app=official_app)
app_list.append(app.dict())
return app_list, has_add
def check_if_official_app(var_path):
if docker.check_directory(var_path):
if docker.read_var(var_path, 'name') != "" and docker.read_var(var_path, 'trademark') != "" and docker.read_var(
var_path, 'requirements') != "":
requirements = docker.read_var(var_path, 'requirements')
try:
cpu = requirements['cpu']
mem = requirements['memory']
return True
except:
return False
else:
return False
def get_apps_from_queue(app_list, has_add):
file_path = "/data/apps/running_apps.txt"
if docker.check_directory(file_path):
output = shell_execute.execute_command_output_all("cat " + file_path)
apps = output["result"].split("\n")
for running_app_name in apps:
running_app_name = re.sub("\n", "", running_app_name)
if running_app_name not in has_add and running_app_name != "":
var_path = "/data/apps/" + running_app_name + "/variables.json"
trade_mark = docker.read_var(var_path, 'trademark')
real_name = docker.read_var(var_path, 'name')
image_url = get_Image_url(real_name)
app = App(app_id=real_name + "_" + running_app_name, name=real_name, customer_name=running_app_name,
status_code=const.APP_READY, status="installing", port=0, volume="",
url="", image_url=image_url, admin_url="", trade_mark=trade_mark, user_name="",
password="", official_app=True)
app_list.append(app.dict())
return app_list
def get_Image_url(app_name):
image_url = "static/images/" + app_name + "-websoft9.png"
return image_url
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