Init project

This commit is contained in:
2025-07-10 16:41:46 +08:00
commit f589307abe
8 changed files with 240 additions and 0 deletions

1
.python-version Normal file
View File

@ -0,0 +1 @@
3.12

0
README.md Normal file
View File

7
pyproject.toml Normal file
View File

@ -0,0 +1,7 @@
[project]
name = "managescripts"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = []

140
servicesManager.py Executable file
View File

@ -0,0 +1,140 @@
#!/usr/bin/env python3
import subprocess
import sys
import os
import logging
import colorlog
# init color log
handler = colorlog.StreamHandler()
handler.setFormatter(colorlog.ColoredFormatter(
'%(log_color)s%(asctime)s - %(levelname)s - %(message)s',
log_colors={
'DEBUG': 'cyan',
'INFO': 'green',
'WARNING': 'yellow',
'ERROR': 'red',
'CRITICAL': 'bold_red',
}
))
logger = colorlog.getLogger(__name__)
logger.setLevel(logging.INFO)
logger.addHandler(handler)
def get_operation():
while True:
choice = input("请选择操作 (0 停止, 1 重启, q 退出): ").strip().lower()
if choice == 'q':
return 'q'
try:
op = int(choice)
if op in (0, 1):
return op
except ValueError:
pass
logger.warning("无效输入,请输入 0、1 或 q。")
def manage_service(name, command_template):
logger.info(f"===== {name} 管理 =====")
logger.info("0. 停止")
logger.info("1. 启动")
op = get_operation()
if op == 'q':
return False
action = 'down' if op == 0 else 'up -d'
cmd = command_template.format(action=action)
try:
subprocess.run(cmd, shell=True, check=True)
logger.info(f"{name} {('停止' if op == 0 else '重启')} 完成")
except subprocess.CalledProcessError as e:
logger.error(f"{name} 管理失败: {e}")
return True
def nginx_manager():
logger.info("===== Nginx 管理 =====")
logger.info("0. 停止")
logger.info("1. 重启")
op = get_operation()
if op == 'q':
return False
action = 'stop' if op == 0 else 'restart'
cmd = f"sudo systemctl {action} nginx"
try:
subprocess.run(cmd, shell=True, check=True)
logger.info(f"Nginx {('停止' if op == 0 else '重启')} 完成")
except subprocess.CalledProcessError as e:
logger.error(f"Nginx 管理失败: {e}")
return True
def homepage_manager():
path = os.path.expanduser("~/web/homepageService")
return manage_service("Homepage", f"cd {path} && docker compose {{action}}")
def status_manager():
path = os.path.expanduser("~/web/statusService")
return manage_service("Status", f"cd {path} && docker compose {{action}}")
def gitea_manager():
path = os.path.expanduser("~/web/giteaService")
return manage_service("Gitea", f"cd {path} && docker compose {{action}}")
def main():
services = {
0: ("Nginx", nginx_manager),
1: ("Homepage", homepage_manager),
2: ("Status", status_manager),
3: ("Gitea", gitea_manager),
}
while True:
logger.info("=== 服务管理 ===")
for key, (name, _) in services.items():
logger.info(f"{key}. {name}")
logger.info("a. 全部")
logger.info("q. 退出")
choice = input("请选择要管理的服务或 a 全部: ").strip().lower()
if choice == 'q':
logger.info("退出程序")
break
if choice == 'a':
logger.info("=== 批量操作: 全部服务 ===")
logger.info("0. 停止")
logger.info("1. 重启")
op = get_operation()
if op == 'q':
continue
action_nginx = 'stop' if op == 0 else 'restart'
action_others = 'down' if op == 0 else 'restart'
commands = [
("Nginx", f"sudo systemctl {action_nginx} nginx"),
("Homepage", f"cd {os.path.expanduser('~/web/homepageService')} && docker compose {action_others}"),
("Status", f"cd {os.path.expanduser('~/web/statusService')} && docker compose {action_others}"),
("Gitea", f"cd {os.path.expanduser('~/web/giteaService')} && docker compose {action_others}"),
]
for name, cmd in commands:
try:
logger.info(f"{name} 开始 {('停止' if op==0 else '重启')}")
subprocess.run(cmd, shell=True, check=True)
logger.info(f"{name} 完成")
except subprocess.CalledProcessError as e:
logger.error(f"{name} 操作失败: {e}")
continue
try:
idx = int(choice)
if idx in services:
cont = services[idx][1]()
if cont is False:
continue
else:
logger.warning("无效选项")
except ValueError:
logger.warning("无效输入,请输入数字或 q")
return 0
if __name__ == "__main__":
sys.exit(main())

24
shell/install.sh Executable file
View File

@ -0,0 +1,24 @@
#!/bin/bash
set -ue
sudo apt update
## Docker
# curl -fsSL https://get.docker.com | bash -s docker
# sudo apt install docker-compose
# # sudo groupadd docker # 若尚不存在 docker 组,则需先创建
# sudo usermod -aG docker $USER
## Nginx
sudo apt install nginx
sudo systemctl enable nginx
## Certbot
sudo apt install certbot
## Cron
sudo apt install cron
## Python
sudo apt install python3
sudo apt install pip
curl -LsSf https://astral.sh/uv/install.sh | sh # astral uv

3
shell/sslrenew.sh Executable file
View File

@ -0,0 +1,3 @@
#!/bin/sh
sudo certbot renew

64
src/ServicesManager.py Normal file
View File

@ -0,0 +1,64 @@
import subprocess
import sys
import os
import logging
import colorlog
# init color log
handler = colorlog.StreamHandler()
handler.setFormatter(colorlog.ColoredFormatter(
'%(log_color)s%(asctime)s - %(levelname)s - %(message)s',
log_colors={
'DEBUG': 'cyan',
'INFO': 'green',
'WARNING': 'yellow',
'ERROR': 'red',
'CRITICAL': 'bold_red',
}
))
logger = colorlog.getLogger(__name__)
logger.setLevel(logging.INFO)
logger.addHandler(handler)
def get_operation():
while True:
choice = input("请选择操作 (0 停止, 1 重启, q 退出): ").strip().lower()
if choice == 'q':
return 'q'
try:
op = int(choice)
if op in (0, 1):
return op
except ValueError:
pass
logger.warning("无效输入,请输入 0、1 或 q。")
class Service:
"""A template of webservices
Attributes:
tag (str): The tag that mark the service instance use which way to deploy.
name (str): The name of the service instance.
dir (str): The configuration & data directory of the service instance.
"""
def __init__(self, tag: str, name: str, dir: str):
self.tag = tag
self.name = name
self.dir = dir
def command_gen(self):
"""Generate services's management commands according to services's tag
"""
system_services_command = "sudo systemctl"
docker_services_command = f"cd {os.path.expanduser(f'{self.dir}')} && docker compose"
if self.tag == "sys":
return system_services_command
elif self.tag == "docker":
return docker_services_command
else:
raise ValueError("The service tag {self.tag} was not included")

1
src/__init__.py Normal file
View File

@ -0,0 +1 @@