websoft9/appmanage/api/utils/docker.py

import os, io, sys, platform, shutil, time, json, datetime
import re
from api.utils import shell_execute
from api.utils import network

from dotenv import load_dotenv, find_dotenv
import dotenv
from pathlib import Path

def create_app_directory(app_name):
    # 判断/data/apps/app_name是否已经存在，如果已经存在，方法结束
    print("checking dir...")
    path = "/data/apps/"+app_name
    isexsits = os.path.exists(path)
    if isexsits:
        return

    if not os.path.exists("/tmp/docker-library"):
        shell_execute.execute_command_output_all("git clone https://ghproxy.com/https://github.com/Websoft9/docker-library.git /tmp/docker-library")

    shell_execute.execute_command_output_all("cp -r /tmp/docker-library/apps/"+app_name+" /data/apps")

def check_app_compose(app_name):
    print("checking port...")
    path = "/data/apps/" + app_name + "/.env"
    http_port_env, http_port = read_env(path, "APP_HTTP_PORT")
    db_port_env, db_port = read_env(path, "APP_DB.*_PORT")
    #1.判断/data/apps/app_name/.env中的port是否占用，没有被占用，方法结束（network.py的get_start_port方法）
    if http_port != "":
        print("check http port...")
        http_port = network.get_start_port(http_port)
        modify_port(path, http_port_env, http_port)
    if db_port != "":
        print("check db port...")
        db_port = network.get_start_port(db_port)
        modify_port(path, db_port_env, db_port)
    print("port check complete")
    return

def read_env(path, key):
    output = shell_execute.execute_command_output_all("cat " + path + "|grep "+ key+ "|head -1")
    code = output["code"]
    env = ""    #the name of environment var
    ret = ""    #the value of environment var
    if int(code) == 0 and output["result"] != "":
        ret = output["result"]
        env = ret.split("=")[0]
        ret = ret.split("=")[1]
        ret = re.sub("'","",ret)
        ret = re.sub("\n","",ret)
    return env, ret

def modify_port(path, env_name, port):
    file_data = ""
    with open(path, "r", encoding="utf-8") as f:
        for line in f:
            if env_name in line:
                line = line.replace(line, env_name + "=" + port+"\n")
            file_data += line
    with open(path, "w", encoding="utf-8") as f:
        f.write(file_data)