Linux下一键启停Python服务脚本的超实用方法

$ ./manager.sh start
 ##: 启动成功(Start successfully) ~

$ ./manager.sh stop
 ##: 已停止(stopped) ...

$ ./manager.sh restart
 ##: 已停止(stopped) ...
 ##: 启动成功(Start successfully) ~

$ ./manager.sh start
 ##: 服务已经在运行中(The service is already running)

#!/bin/bash

# 要管理的 Python 服务文件名（可修改）
server_name="server.py"

# 彩色输出定义
RED='\033[31m'
GREEN='\033[32m'
YELLOW='\033[33m'
CYAN='\033[36m'
RESET='\033[0m'

start() {
    # 检查是否已有进程在运行（取第一个匹配的 PID）
    pid=$(ps -ef | grep -v grep | grep ${server_name} | awk 'NR==1{print $2}')
    
    if [[ -z "$pid" ]]; then
        echo -e "${YELLOW}正在启动 ${server_name} ...${RESET}"
        nohup python3 ${server_name} > server.log 2>&1 &
        sleep 2  # 给进程一点启动时间
        
        # 再次检查是否真的启动成功
        pid=$(ps -ef | grep -v grep | grep ${server_name} | awk 'NR==1{print $2}')
        if [[ -n "$pid" ]]; then
            echo -e "${CYAN} ##: 启动成功(Start successfully) ~${RESET}"
        else
            echo -e "${RED} 启动失败(failed to activate)，请查看 server.log${RESET}"
        fi
    else
        echo -e "${GREEN} ##: 服务已经在运行中(The service is already running)，PID: $pid${RESET}"
    fi
}

stop() {
    pid=$(ps -ef | grep -v grep | grep ${server_name} | awk 'NR==1{print $2}')
    
    if [[ -n "$pid" ]]; then
        kill -9 $pid
        echo -e "${GREEN} ##: 已停止(stopped)，PID $pid 已结束${RESET}"
    else
        echo -e "${YELLOW} ##: 未检测到运行中的服务${RESET}"
    fi
}

restart() {
    echo -e "${YELLOW}正在重启服务...${RESET}"
    stop
    sleep 1
    start
}

# 使用说明
usage() {
    echo "Usage: $0 {start|stop|restart}"
    exit 1
}

case "$1" in
    start)
        start
        ;;
    stop)
        stop
        ;;
    restart)
        restart
        ;;
    *)
        usage
        ;;
esac

exit 0

chmod +x manager.sh

./manager.sh start    # 启动
./manager.sh stop     # 停止
./manager.sh restart  # 重启