openNAMU__openNAMU/route/go_api_w_render.py

from .tool.func import *

class api_w_render_include:
    def __init__(self, data_option):
        self.include_change_list = data_option

    def __call__(self, match):
        match_org = match.group(0)
        match = match.groups()

        if len(match) < 3:
            match = list(match) + ['']

        if match[2] == '\\':
            return match_org
        else:
            slash_add = ''
            if match[0]:
                if len(match[0]) % 2 == 1:
                    slash_add = '\\' * (len(match[0]) - 1)
                else:
                    slash_add = match[0]

            if match[1] in self.include_change_list:
                return slash_add + self.include_change_list[match[1]]
            else:
                return slash_add + match[2]

async def api_w_render(name = '', tool = '', request_method = '', request_data = {}):
    with get_db_connect() as conn:
        curs = conn.cursor()

        flask_data = flask_data_or_variable(flask.request.form, request_data)
        request_method = flask.request.method if request_method == '' else request_method

        if request_method == 'POST':
            name = flask_data.get('name', '')
            data_org = flask_data.get('data', '')
            data_option = flask_data.get('option', '')

            markup = ''
            if tool in ('', 'from', 'include'):
                curs.execute(db_change("select set_data from data_set where doc_name = ? and set_name = 'document_markup'"), [name])
                db_data = curs.fetchall()
                if db_data and db_data[0][0] != '' and db_data[0][0] != 'normal':
                    markup = db_data[0][0]

                if markup == '':
                    curs.execute(db_change('select data from other where name = "markup"'))
                    db_data = curs.fetchall()
                    markup = db_data[0][0] if db_data else 'namumark'

            data_type = ''
            if tool == '':
                data_type = 'api_view'
            elif tool == 'from':
                data_type = 'api_from'
            elif tool == 'include':
                data_type = 'api_include'
            elif tool == 'backlink':
                data_type = 'backlink'
            else:
                data_type = 'api_thread'

            if markup in ('', 'namumark', 'namumark_beta') and data_option != '':
                data_option = orjson.loads(data_option)

                data_option_func = api_w_render_include(data_option)

                # parameter replace
                data_org = re.sub(r'(\\+)?@([ㄱ-힣a-zA-Z0-9_]+)=((?:\\@|[^@\n])+)@', data_option_func, data_org)
                data_org = re.sub(r'(\\+)?@([ㄱ-힣a-zA-Z0-9_]+)@', data_option_func, data_org)

                # remove end br
                data_org = re.sub('^\n+', '', data_org)

            if markup in ('', 'namumark'):
                data_pas = render_set(conn, 
                    doc_name = name, 
                    doc_data = data_org, 
                    data_type = data_type
                )

                return flask.jsonify({
                    "data" : data_pas[0], 
                    "js_data" : data_pas[1]
                })
            else:
                other_set = {}
                other_set["doc_name"] = name
                other_set["render_type"] = data_type
                other_set["data"] = data_org

                return flask.Response(response = (await python_to_golang(sys._getframe().f_code.co_name, other_set)), status = 200, mimetype = 'application/json')
        else:
            return flask.jsonify({})