websoft9/cli/controller.py
chendelin1982 cf777bc15c compose
2021-11-02 20:54:10 +08:00

149 lines
4.4 KiB
Python

from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Type, Union
import model, os, sys, subprocess, re
from model import GitOp
path_repo = "./data/application.list"
path_project = ""
# for Git clone HA
github_url = ("https://github.com", "https://hub.fastgit.org", "https://github.com.cnpmjs.org")
class Print:
'''display all any information you want'''
def __init__(self, status:str):
self.status = status
def printRepo():
model.FileOp.printJson(path_repo)
def lsProject(self):
ls = model.DockerOp(self.status)
ls.lsProject()
class Create:
def __init__(self, app_name: str, project_name: str):
self.folder = None
self.app_name = app_name
self.project_name = project_name
if self.project_name != None:
self.folder = self.project_name
else:
self.folder = self.app_name
def downRepo(self):
'''download repository'''
geturl = model.SmoothUrl()
gitop = model.GitOp()
cmd = "git clone --depth=1 " + geturl.res(github_url) + "/websoft9/docker-" + self.app_name + " " + self.folder
if os.path.exists("./"+self.folder):
print(os.path.abspath(self.folder)+" folder already exists")
sys.exit(0)
else:
gitop.gitClone(cmd)
def setEnv(self):
'''reset the password | port | container_name for application'''
fileop=model.FileOp(self.folder+'/.env')
securityop=model.SecurityOp()
netop=model.NetOp()
env_dict = fileop.fileToDict()
env_str = fileop.fileToString()
port_list = []
for key in list(env_dict.keys()):
if env_dict[key] in ["", "True", "False"]:
del env_dict[key]
for key,value in env_dict.items():
# replace password
if re.match('\w*PASSWORD',key,re.I) != None:
env_str = env_str.replace(key+"="+value, key+"="+securityop.randomPass())
# replace port
if re.match('\w*PORT',key,re.I) != None:
port = int(value)
while port in port_list or not netop.checkPort(port):
port = port + 1
port_list.append(port)
env_str = env_str.replace(key+"="+value, key+"="+str(port))
# replace app_container
if re.match('\w*APP_CONTAINER_NAME',key,re.I) != None:
env_str = env_str.replace(key+"="+value, key+"="+self.folder)
# replace app_network
if re.match('\w*APP_NETWORK',key,re.I) != None:
env_str = env_str.replace(key+"="+value, key+"="+self.folder)
fileop.stringToFile(env_str)
def upRepo(self):
'''docker-compose up repository'''
cmd_up = "docker-compose up -d"
cmd_down = "docker-compose down -v"
os.chdir(self.folder)
try:
os.system(cmd_up)
except:
print("Create failed")
os.system(cmd_down)
sys.exit(0)
def printResult(self):
pass
class Status:
'''The status operation of project, e.g Start | Stop | Restart | Erase'''
def __init__(self, project_name: str, **parameter):
self.project_name = project_name
self.dockerop = model.DockerOp()
projectdict = self.dockerop.getProject()
if self.project_name != None:
try:
self.project_path = projectdict[self.project_name]
except:
print("No this application!")
sys.exit(0)
else
try:
self.project_path = parameter[project-directory]
except:
print("Directory not exist, please check it")
sys.exit(0)
self.dockercomposeop = model.DockerComposeOp(self.project_path)
def stopApp(self):
pass
def upApp(self):
print("It up the application at directory: "+self.project_path)
self.dockercomposeop.execute()
def startApp(self):
pass
def restartApp(self):
self.dockercomposeop.execute()
def deleteApp(self):
self.dockercomposeop.execute()