mdserver-web/web/admin/setup/init_cmd.py

# coding:utf-8

# ---------------------------------------------------------------------------------
# MW-Linux面板
# ---------------------------------------------------------------------------------
# copyright (c) 2018-∞(https://github.com/midoks/mdserver-web) All rights reserved.
# ---------------------------------------------------------------------------------
# Author: midoks <midoks@163.com>
# ---------------------------------------------------------------------------------

import os
import shutil

import core.mw as mw

def cmdContent():
    script = mw.getPanelDir() + '/scripts/init.d/mw.tpl'
    content = mw.readFile(script)
    content = content.replace("{$SERVER_PATH}", mw.getPanelDir())
    content += "\n# make:{0}".format(mw.formatDate())
    return content


def init_cmd():
    cmd_content = cmdContent()
    script_bin = mw.getPanelDir() + '/scripts/init.d/mw'
    mw.writeFile(script_bin, cmd_content)
    mw.execShell('chmod +x ' + script_bin)

    # 在linux系统中,确保/etc/init.d存在
    if not mw.isAppleSystem() and not os.path.exists("/etc/rc.d/init.d"):
        mw.execShell('mkdir -p /etc/rc.d/init.d')

    if not mw.isAppleSystem() and not os.path.exists("/etc/init.d"):
        mw.execShell('mkdir -p /etc/init.d')

    sys_name = mw.getOsName()

    # initd
    if os.path.exists('/etc/rc.d/init.d'):
        initd_bin = '/etc/rc.d/init.d/mw'
        mw.writeFile(initd_bin, cmd_content)
        mw.execShell('chmod +x ' + initd_bin)
        # 加入自启动
        mw.execShell('which chkconfig && chkconfig --add mw')


    if os.path.exists('/etc/init.d'):
        initd_bin = '/etc/init.d/mw'
        mw.writeFile(initd_bin, cmd_content)
        mw.execShell('chmod +x ' + initd_bin)
        # 加入自启动
        mw.execShell('which update-rc.d && update-rc.d -f mw defaults')

    # if sys_name == 'opensuse':
    #     init_cmd_systemd()
    return True


def init_cmd_systemd():
    systemd_dir = mw.systemdCfgDir()

    systemd_mw = systemd_dir + '/mw.service'
    systemd_mw_task = systemd_dir + '/mw-task.service'

    systemd_mw_tpl = mw.getPanelDir() + '/scripts/init.d/mw.service.tpl'
    systemd_mw_task_tpl = mw.getPanelDir() + '/scripts/init.d/mw-task.service.tpl'

    if os.path.exists(systemd_mw):
        os.remove(systemd_mw)
    if os.path.exists(systemd_mw_task):
        os.remove(systemd_mw_task)

    contentReplace(systemd_mw_tpl, systemd_mw)
    contentReplace(systemd_mw_task_tpl, systemd_mw_task)

    mw.execShell('systemctl enable mw')
    mw.execShell('systemctl enable mw-task')
    mw.execShell('systemctl daemon-reload')