openNAMU__openNAMU/route/main_sys_restart.py

from .tool.func import *

def main_sys_restart_do():
    print('Restart')

    time.sleep(3)

    python_ver = ''
    python_ver = str(sys.version_info.major) + '.' + str(sys.version_info.minor)

    run_list = [
        sys.executable,
        'python' + python_ver,
        'python3',
        'python',
        'py -' + python_ver
    ]

    for exe_name in run_list:
        try:
            subprocess.Popen([exe_name] + sys.argv)
            break
        except:
            continue
    
    os._exit(0)

async def main_sys_restart(golang_process):
    with get_db_connect() as conn:
        if await acl_check('', 'owner_auth', '', '') == 1:
            return await re_error(conn, 3)

        if flask.request.method == 'POST':
            await acl_check(tool = 'owner_auth', memo = 'restart')

            if golang_process.poll() is None:
                golang_process.terminate()
                try:
                    golang_process.wait(timeout = 5)
                except subprocess.TimeoutExpired:
                    golang_process.kill()
                    try:
                        golang_process.wait(timeout = 5)
                    except subprocess.TimeoutExpired:
                        print('Golang process not terminated properly.')

            threading.Thread(target = main_sys_restart_do).start()
            return flask.Response(await get_lang("warning_restart"), status = 200)
        else:
            return await render_template(
                await get_lang('wiki_restart'),
                '''
                    <form method="post">
                        <button type="submit">''' + await get_lang('restart') + '''</button>
                    </form>
                ''',
                0,
                [['manager', await get_lang('return')]]
            )