openNAMU__openNAMU/route/bbs_w_delete.py

from .tool.func import *

from .api_bbs_w_post import api_bbs_w_post

def bbs_w_delete(bbs_num = '', post_num = '', comment_num = ''):
    with get_db_connect() as conn:
        curs = conn.cursor()

        curs.execute(db_change('select set_data from bbs_set where set_id = ? and set_name = "bbs_name"'), [bbs_num])
        db_data = curs.fetchall()
        if not db_data:
            return redirect('/bbs/main')

        bbs_name = db_data[0][0]
        
        bbs_num_str = str(bbs_num)
        post_num_str = str(post_num)

        if admin_check() != 1:
            return redirect('/bbs/w/' + bbs_num_str)
        
        temp_dict = json.loads(api_bbs_w_post(bbs_num_str + '-' + post_num_str).data)
        if not 'user_id' in temp_dict:
            return redirect('/bbs/main')
        
        if flask.request.method == 'POST':
            if comment_num == '':
                curs.execute(db_change('delete from bbs_data where set_code = ? and set_id = ?'), [post_num_str, bbs_num_str])
                curs.execute(db_change('delete from bbs_set where set_code = ? and set_id = ?'), [post_num_str, bbs_num_str])
                curs.execute(db_change('delete from bbs_data where set_id = ? or set_id like ?'), [bbs_num_str + '-' + post_num_str, bbs_num_str + '-' + post_num_str + '-%'])
                
                return redirect('/bbs/w/' + bbs_num_str)
            else:
                comment_num_split = comment_num.split('-')
                
                set_id = bbs_num_str + '-' + post_num_str
                set_id_sub = '-'.join(comment_num_split[:-1])
                if set_id_sub != '':
                    set_id += '-' + set_id_sub

                set_code = comment_num_split[len(comment_num_split) - 1]

                print(set_id, set_code)

                curs.execute(db_change("update bbs_data set set_data = '' where set_code = ? and set_id = ?"), [set_code, set_id])
                
                return redirect('/bbs/w/' + bbs_num_str + '/' + post_num_str)
        else:
            sub = '(' + bbs_name + ')'
            sub += ' (' + post_num_str + ')'
            
            name = load_lang('bbs_comment_delete')
            if comment_num == '':
                name = load_lang('bbs_post_delete')
            else:
                sub += ' (' + comment_num + ')'

            return easy_minify(flask.render_template(skin_check(),
                imp = [name, wiki_set(), wiki_custom(), wiki_css([sub, 0])],
                data = render_simple_set('''
                    <form method="post">
                        <span>''' + load_lang('delete_warning') + '''</span>
                        <hr class="main_hr">
                        <button type="submit">''' + load_lang('delete') + '''</button>
                    </form>
                '''),
                menu = [['bbs/w/' + bbs_num_str + '/' + post_num_str, load_lang('return')]]
            ))