openNAMU__openNAMU/route/now_update.py

from .tool.func import *

def now_update_2(conn):
    curs = conn.cursor()

    if admin_check(None, 'update') != 1:
       return re_error('/error/3')

    curs.execute('select data from other where name = "update"')
    up_data = curs.fetchall()
    if up_data:
        up_data = up_data[0][0]
    else:
        up_data = 'stable'

    if platform.system() == 'Linux':
        print('Update')

        os.system('git remote rm origin')
        os.system('git remote add origin https://github.com/2DU/opennamu.git')
        ok = os.system('git fetch origin ' + up_data)
        ok = os.system('git reset --hard origin/' + up_data)
        if ok == 0:
            return redirect('/restart')
    else:
        if platform.system() == 'Windows':
            print('Update')

            urllib.request.urlretrieve('https://github.com/2DU/opennamu/archive/' + up_data + '.zip', 'update.zip')
            zipfile.ZipFile('update.zip').extractall('')
            ok = os.system('xcopy /y /r opennamu-' + up_data + ' .')
            if ok == 0:
                print('Remove')
                os.system('rd /s /q opennamu-' + up_data)
                os.system('del update.zip')

                return redirect('/restart')

    return easy_minify(flask.render_template(skin_check(), 
        imp = [load_lang('update'), wiki_set(), custom(), other2([0, 0])],
        data = load_lang("update_error") + ' <a href="https://github.com/2DU/opennamu">(Github)</a>',
        menu = [['manager/1', load_lang('return')]]
    ))