openNAMU__openNAMU/route/edit_move.py

from .tool.func import *

async def edit_move(name):
    with get_db_connect() as conn:
        curs = conn.cursor()

        if await acl_check(name, 'document_move') == 1:
            return await re_error(conn, 0)
        
        if do_title_length_check(conn, name) == 1:
            return await re_error(conn, 38)

        if flask.request.method == 'POST':
            move_title = flask.request.form.get('title', 'test')
            if await acl_check(move_title) == 1:
                return await re_error(conn, 0)

            if await captcha_post(conn, flask.request.form.get('g-recaptcha-response', flask.request.form.get('g-recaptcha', ''))) == 1:
                return await re_error(conn, 13)

            if await do_edit_slow_check(conn) == 1:
                return await re_error(conn, 24)
            
            send = flask.request.form.get('send', '')
            agree = flask.request.form.get('copyright_agreement', '')

            time = get_time()
            ip = ip_check()
            
            has_error = 0

            move_option = flask.request.form.get('move_option', 'none')
            move_option_topic = flask.request.form.get('move_topic_option', 'none')
            document_set_option = flask.request.form.get('document_set_option', 'none')
            
            if await do_edit_send_check(conn, send) == 1:
                return await re_error(conn, 37)
            
            if do_edit_text_bottom_check_box_check(conn, agree) == 1:
                return await re_error(conn, 29)

            # 역링크 관련 패치 해야할 듯

            # 문서 이동 파트 S
            curs.execute(db_change("select title from history where title = ?"), [move_title])
            if curs.fetchall():
                if move_option == 'merge' and await acl_check(tool = 'owner_auth', memo = 'merge documents (' + name + ') (' + move_title + ')') != 1:
                    curs.execute(db_change("select data from data where title = ?"), [move_title])
                    data = curs.fetchall()
                    if data:
                        curs.execute(db_change("delete from data where title = ?"), [move_title])
                        curs.execute(db_change("delete from back where link = ?"), [move_title])

                    curs.execute(db_change("select data from data where title = ?"), [name])
                    data = curs.fetchall()
                    data_in = data[0][0] if data else ''

                    curs.execute(db_change("update data set title = ? where title = ?"), [move_title, name])
                    curs.execute(db_change("update back set link = ? where link = ?"), [move_title, name])

                    # 역링크 S
                    # 문서 합치기이므로 기존 문서 쪽은 no 역링크 생성, 이동하는 곳에는 no 역링크 제거
                    curs.execute(db_change("select distinct link from back where title = ?"), [name])
                    backlink = [[for_a[0], name, 'no', ''] for for_a in curs.fetchall()]
                    curs.executemany(db_change("insert into back (link, title, type, data) values (?, ?, ?, ?)"), backlink)
                    curs.execute(db_change("delete from back where title = ? and type = 'no'"), [move_title])
                    # 역링크 E

                    curs.execute(db_change("select id from history where title = ? order by id + 0 desc limit 1"), [move_title])
                    num = curs.fetchall()[0][0]

                    curs.execute(db_change("select id from history where title = ? order by id + 0 asc"), [name])
                    data = curs.fetchall()
                    for move in data:
                        curs.execute(db_change("update rc set title = ?, id = ? where title = ? and id = ?"), [move_title, str(int(num) + int(move[0])), name, move[0]])
                        curs.execute(db_change("update history set title = ?, id = ? where title = ? and id = ?"), [move_title, str(int(num) + int(move[0])), name, move[0]])

                    history_plus(conn, 
                        move_title, 
                        data_in, 
                        time, 
                        ip, 
                        send, 
                        '0',
                        t_check = '<a>' + name + '</a> ↔ <a>' + move_title + '</a>',
                        mode = 'move'
                    )
                elif move_option == 'reverse':
                    # 전체적인 구조 변경 필요
                    # 중간 문서 거치지 않고 불러와서 바로 변경하도록
                    # 문서 이동 말고 나머지도 그렇게 변경 필요함
                    i = 0
                    var_name = ''
                    while var_name == '':
                        temp_title = 'test ' + load_random_key() + ' ' + str(i)
                        curs.execute(db_change("select title from history where title = ? limit 1"), [temp_title])
                        if not curs.fetchall():
                            var_name = temp_title
                        else:
                            i += 1

                    for title_name in [[name, var_name], [move_title, name], [var_name, move_title]]:
                        curs.execute(db_change("update data set title = ? where title = ?"), [title_name[1], title_name[0]])
                        curs.execute(db_change("update back set link = ? where link = ?"), [title_name[1], title_name[0]])

                        curs.execute(db_change("update history set title = ? where title = ?"), [title_name[1], title_name[0]])
                        curs.execute(db_change("update rc set title = ? where title = ?"), [title_name[1], title_name[0]])

                    for title_name in [[name, move_title], [move_title, name]]:
                        curs.execute(db_change("select data from data where title = ?"), [name])
                        data = curs.fetchall()
                        data_in = data[0][0] if data else ''

                        history_plus(conn, 
                            title_name[0], 
                            data_in, 
                            time, 
                            ip, 
                            send, 
                            '0',
                            t_check = '<a>' + title_name[0] + '</a> ⇋ <a>' + title_name[1] + '</a>',
                            mode = 'move'
                        )
                elif move_option != 'none':
                    has_error = 1
            elif move_option != 'none':
                curs.execute(db_change("select data from data where title = ?"), [name])
                data = curs.fetchall()
                data_in = data[0][0] if data else ''

                curs.execute(db_change("update data set title = ? where title = ?"), [move_title, name])
                curs.execute(db_change("update back set link = ? where link = ?"), [move_title, name])

                # 역링크 S
                # 문서 합치기 쪽 역링크와 동일하게
                curs.execute(db_change("select distinct link from back where title = ?"), [name])
                backlink = [[for_a[0], name, 'no', ''] for for_a in curs.fetchall()]
                curs.executemany(db_change("insert into back (link, title, type, data) values (?, ?, ?, ?)"), backlink)
                curs.execute(db_change("delete from back where title = ? and type = 'no'"), [move_title])
                # 역링크 E

                # 역사와 최근 변경 이동 S
                curs.execute(db_change("update history set title = ? where title = ?"), [move_title, name])
                curs.execute(db_change("update rc set title = ? where title = ?"), [move_title, name])
                # 역사와 최근 변경 이동 E

                history_plus(conn, 
                    move_title, 
                    data_in, 
                    time, 
                    ip, 
                    send,
                    '0',
                    t_check = '<a>' + name + '</a> → <a>' + move_title + '</a>',
                    mode = 'move'
                )

            # 문서 이동 파트 E
            
            # 토론 이동 파트 S
            curs.execute(db_change("select title from rd where title = ?"), [move_title])
            if curs.fetchall():
                if move_option_topic == 'merge' and await acl_check(tool = 'owner_auth', memo = 'merge document\'s topics (' + name + ') (' + move_title + ')') != 1:
                    curs.execute(db_change("update rd set title = ? where title = ?"), [move_title, name])
                elif move_option_topic == 'reverse':
                    i = 0
                    var_name = ''
                    while var_name == '':
                        temp_title = 'test ' + load_random_key() + ' ' + str(i)
                        curs.execute(db_change("select title from rd where title = ? limit 1"), [temp_title])
                        if not curs.fetchall():
                            var_name = temp_title
                        else:
                            i += 1
                    
                    for title_name in [[name, var_name], [move_title, name], [var_name, move_title]]:
                        curs.execute(db_change("update rd set title = ? where title = ?"), [title_name[1], title_name[0]])
                else:
                    has_error = 1
            elif move_option_topic != 'none':
                curs.execute(db_change("update rd set title = ? where title = ?"), [move_title, name])

            # 토론 이동 파트 E

            # data_set 이동 파트 S
            if document_set_option == 'reverse':
                i = 0
                var_name = ''
                while var_name == '':
                    temp_title = 'test ' + load_random_key() + ' ' + str(i)
                    curs.execute(db_change("select title from history where title = ? limit 1"), [temp_title])
                    if not curs.fetchall():
                        var_name = temp_title
                    else:
                        i += 1
                
                for title_name in [[name, var_name], [move_title, name], [var_name, move_title]]:
                    curs.execute(db_change("update data_set set doc_name = ? where doc_name = ?"), [title_name[1], title_name[0]])
            elif document_set_option == 'normal':
                curs.execute(db_change("delete from data_set where doc_name = ?"), [move_title])
                curs.execute(db_change("delete from acl where title = ?"), [move_title])

                curs.execute(db_change("update data_set set doc_name = ? where doc_name = ?"), [move_title, name])
                curs.execute(db_change("update acl set title = ? where title = ?"), [move_title, name])

            # data_set 이동 파트 E

            if has_error == 0:
                return redirect(conn, '/w/' + url_pas(move_title))
            else:
                return await re_error(conn, 19)
        else:
            owner_auth = await acl_check(tool = 'owner_auth')
            owner_auth = 1 if owner_auth == 0 else 0

            return await render_template(
                name,
                '''
                    <form method="post">
                        <span>''' + await get_lang('document_name') + '''</span>
                        <hr class="main_hr">
                        <input class="__ON_INPUT__" value="''' + name + '''" name="title" type="text">
                        <hr class="main_hr">
                        
                        <input class="__ON_INPUT__" placeholder="''' + await get_lang('why') + '''" name="send" type="text">
                        <hr class="main_hr">
                        
                        <h2>''' + await get_lang('document') + '''</h2>
                        <select class="__ON_SELECT__" name="move_option">
                            <option value="normal"> ''' + await get_lang('normal') + '''</option>
                            <option value="none"> ''' + await get_lang('dont_move') + '''</option>
                            <option value="reverse"> ''' + await get_lang('replace_move') + '''</option>
                            ''' + ('<option value="merge"> ' + await get_lang('merge_move') + '</option>' if owner_auth == 1 else '') + '''
                        </select>
                        <hr class="main_hr">
                        <!-- <label><input class="__ON_INPUT__" type="checkbox" name="move_redirect_make"> ''' + await get_lang('move_redirect_make') + '''</label>
                        <hr class="main_hr"> -->
                        
                        <h2>''' + await get_lang('discussion') + '''</h2>
                        <select class="__ON_SELECT__" name="move_topic_option">
                            <option value="none"> ''' + await get_lang('dont_move') + '''</option>
                            <option value="normal"> ''' + await get_lang('normal') + '''</option>
                            <option value="reverse"> ''' + await get_lang('replace_move') + '''</option>
                            ''' + ('<option value="merge"> ' + await get_lang('merge_move') + '</option>' if owner_auth == 1 else '') + '''
                        </select>
                        <hr class="main_hr">

                        ''' + ((
                            '''<h2>''' + await get_lang('document_set') + '''</h2>
                            <select class="__ON_SELECT__" name="document_set_option">
                                <option value="none"> ''' + await get_lang('dont_move') + '''</option>
                                <option value="normal"> ''' + await get_lang('normal') + '''</option>
                                <option value="reverse"> ''' + await get_lang('replace_move') + '''</option>
                            </select>
                            <hr class="main_hr">
                            '''
                        ) if owner_auth == 1 else '') + '''

                        ''' + await captcha_get(conn) + await ip_warning(conn) + get_edit_text_bottom_check_box(conn) + get_edit_text_bottom(conn, 'move')  + '''
                        
                        <button class="__ON_BUTTON__" type="submit">''' + await get_lang('move') + '''</button>
                    </form>
                ''',
                '(' + await get_lang('move') + ')',
                [['w/' + url_pas(name), await get_lang('return')], ['move_all', await get_lang('multiple_move')]]
            )