ZZY
a4ebe33fca
- 添加了新的命令行参数和功能： - --nginx-reload: 重新加载 Nginx 配置 - --docker-compose-restart: 重启 Docker Compose 服务 - --docker-ps: 列出运行中的 Docker 容器 - 重构了 main 函数，使其支持更灵活的参数配置 - 优化了配置文件读取和解析逻辑 - 更新了 .gitignore 文件，添加了 requirements.txt - 修正了拼写错误：INNER_NGINX_OUTPATH -> inner-nginx.conf
117 lines
5.1 KiB
Python
117 lines
5.1 KiB
Python
"""
|
|
this is the main file
|
|
"""
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
from pathlib import Path
|
|
from dotenv import load_dotenv
|
|
from src.jinja2_render import render_dataclass
|
|
from src.config_reader import nginx_config, docker_config, docker2nginx
|
|
|
|
load_dotenv()
|
|
|
|
def run_command(command):
|
|
"run a command and use stdout"
|
|
try:
|
|
result = subprocess.run(command, shell=True, check=True,
|
|
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
print(result.stdout.decode())
|
|
return True
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Command failed with error: {e.stderr.decode()}")
|
|
return False
|
|
|
|
def docker_compose_action(config_path, output_path):
|
|
"读取并渲染配置文件"
|
|
(auto_config, other) = docker_config(config_path)
|
|
render_dataclass(auto_config, Path('./src/docker-compose.j2'), output_path, other=other)
|
|
|
|
def inner_nginx_action(config_path, output_path):
|
|
"读取并渲染配置文件"
|
|
(auto_config, other) = docker2nginx(config_path)
|
|
render_dataclass(auto_config, Path('./src/nginx-conf.j2'), output_path, other=other)
|
|
|
|
def outer_nginx_action(config_path, output_path):
|
|
"读取并渲染配置文件"
|
|
(auto_config, other) = nginx_config(config_path)
|
|
render_dataclass(auto_config, Path('./src/nginx-conf.j2'), output_path, other=other)
|
|
|
|
def nginx_reload():
|
|
"""reload nginx config"""
|
|
res = True
|
|
if res:
|
|
res = run_command("docker exec -it nginx nginx -t")
|
|
if res:
|
|
res = run_command("docker exec -it nginx nginx -s reload")
|
|
|
|
def docker_compose_restart():
|
|
"docker compose restart"
|
|
run_command("docker compose restart")
|
|
|
|
def docker_ps():
|
|
"list running docker containers by my format"
|
|
cmd = "docker ps --format "\
|
|
"\"table {{.Names}}\t{{.Image}}\t{{.Command}}\t{{.CreatedAt}}\t{{.Status}}\""
|
|
run_command(cmd)
|
|
|
|
def main():
|
|
"main function"
|
|
parser = argparse.ArgumentParser(description="Automate Docker and Nginx configuration.")
|
|
parser.add_argument('-a', '--all', action='store_true', default=True,
|
|
help='Perform all actions')
|
|
parser.add_argument('-c', '--docker-compose', action='store_true',
|
|
help='Generate Docker Compose')
|
|
parser.add_argument('-i', '--inner-nginx', action='store_true',
|
|
help='Generate Inner Nginx (docker network)')
|
|
parser.add_argument('-o', '--outer-nginx', action='store_true',
|
|
help='Generate Outer Nginx (host machine)')
|
|
parser.add_argument('-r', '--nginx-reload', action='store_true',
|
|
help='Reload Nginx configuration')
|
|
parser.add_argument('-s', '--docker-compose-restart', action='store_true',
|
|
help='Restart Docker Compose services')
|
|
parser.add_argument('-p', '--docker-ps', action='store_true',
|
|
help='List running Docker containers')
|
|
parser.add_argument('--docker-config-path', type=Path,
|
|
default=Path(os.getenv('DOCKER_INIPATH', 'docker.ini')),
|
|
help='Path to the Docker INI configuration file')
|
|
parser.add_argument('--docker-output-path', type=Path,
|
|
default=Path(os.getenv('DOCKER_COMPOSE_OUTPATH', 'docker-compose.conf')),
|
|
help='Path to the output Docker Compose configuration file')
|
|
parser.add_argument('--inner-config-path', type=Path,
|
|
default=Path(os.getenv('DOCKER_INIPATH', 'docker.ini')),
|
|
help='Path to the Docker INI configuration file for inner Nginx')
|
|
parser.add_argument('--inner-output-path', type=Path,
|
|
default=Path(os.getenv('INNER_NGINX_OUTPATH', 'inner-nginx.conf')),
|
|
help='Path to the output Inner Nginx configuration file')
|
|
parser.add_argument('--outer-config-path', type=Path,
|
|
default=Path(os.getenv('NGINX_INIPATH', 'nginx.ini')),
|
|
help='Path to the Nginx INI configuration file for outer Nginx')
|
|
parser.add_argument('--outer-output-path', type=Path,
|
|
default=Path(os.getenv('OUTER_NGINX_OUTPATH', 'outer-nginx.conf')),
|
|
help='Path to the output Outer Nginx configuration file')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.all:
|
|
docker_compose_action(args.docker_config_path, args.docker_output_path)
|
|
inner_nginx_action(args.inner_config_path, args.inner_output_path)
|
|
outer_nginx_action(args.outer_config_path, args.outer_output_path)
|
|
nginx_reload()
|
|
else:
|
|
if args.docker_compose:
|
|
docker_compose_action(args.docker_config_path, args.docker_output_path)
|
|
if args.inner_nginx:
|
|
inner_nginx_action(args.inner_config_path, args.inner_output_path)
|
|
if args.outer_nginx:
|
|
outer_nginx_action(args.outer_config_path, args.outer_output_path)
|
|
if args.nginx_reload:
|
|
nginx_reload()
|
|
if args.docker_compose_restart:
|
|
docker_compose_restart()
|
|
if args.docker_ps:
|
|
docker_ps()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|