openNAMU__openNAMU/route/server_indexing.py

from .tool.func import *

def server_indexing_2(conn, db_type):
    curs = conn.cursor()

    if db_type == 'sqlite':
        if admin_check() != 1:
            return re_error('/error/3')

        if flask.request.method == 'POST':
            admin_check(None, 'indexing')

            curs.execute(db_change("select name from sqlite_master where type = 'index'"))
            data = curs.fetchall()
            if data:
                for delete_index in data:
                    print('Delete : ' + delete_index[0])

                    sql = 'drop index if exists ' + delete_index[0]
                    
                    try:
                        curs.execute(db_change(sql))
                    except:
                        pass
            else:
                curs.execute(db_change("select name from sqlite_master where type in ('table', 'view') and name not like 'sqlite_%' union all select name from sqlite_temp_master where type in ('table', 'view') order by 1;"))
                for table in curs.fetchall():            
                    curs.execute(db_change('select sql from sqlite_master where name = ?'), [table[0]])
                    cul = curs.fetchall()
                    
                    r_cul = re.findall('(?:([^ (]*) text)', str(cul[0]))
                    
                    for n_cul in r_cul:
                        print('Create : index_' + table[0] + '_' + n_cul)

                        sql = 'create index index_' + table[0] + '_' + n_cul + ' on ' + table[0] + '(' + n_cul + ')'
                        try:
                            curs.execute(db_change(sql))
                        except:
                            pass

            conn.commit()
            
            return redirect()  
        else:
            curs.execute(db_change("select name from sqlite_master where type = 'index'"))
            data = curs.fetchall()
            if data:
                b_data = load_lang('delete')
            else:
                b_data = load_lang('create')

            return easy_minify(flask.render_template(skin_check(), 
                imp = [load_lang('indexing'), wiki_set(), custom(), other2([0, 0])],
                data =  '''
                        <form method="post">
                            <button type="submit">''' + b_data + '''</button>
                        </form>
                        ''',
                menu = [['manager', load_lang('return')]]
            ))   
    else:
        return redirect('/')