openNAMU__openNAMU/route/tool/set_mark/markdown.py

from . import tool

import datetime
import html
import re

def markdown(conn, data, title, include_num):
    backlink = []
    include_num = include_num + '_' if include_num else ''
    plus_data = '' + \
        'get_link_state("' + include_num + '");\n' + \
        'get_file_state("' + include_num + '");\n' + \
    ''

    data = html.escape(data)
    data = data.replace('\r\n', '\n')
    data = '\n' + data

    class head_render:
        def __init__(self):
            pass

        def __call__(self, match):
            head_len = str(len(match[1]))
            head_data = match[2]

            return '<h' + head_len + '>' + head_data + '</h' + head_len + '>'

    head_r = r'\n(#{1,6}) ?([^\n]+)'
    head_do = head_render()
    data = re.sub(head_r, head_do, data)

    class link_render:
        def __init__(self):
            pass

        def __call__(self, match):
            if match[1] == '!':
                if re.search(r'^http(s)?:\/\/', match[3], flags = re.I):
                    return '<img alt="' + match[2] + '" src="' + match[3] + '">'
                else:
                    file_name = re.search(r'^([^.]+)\.([^.]+)$', match[3])
                    if file_name:
                        file_end = file_name.group(2)
                        file_name = file_name.group(1)
                    else:
                        file_name = 'Test'
                        file_end = 'jpg'

                    file_src = '/image/' + tool.sha224_replace(file_name) + '.' + file_end
                    file_alt = 'file:' + file_name + '.' + file_end

                    return '' + \
                        '<img class="' + include_num + 'file_finder_1" alt="' + match[2] + '" src="' + file_src + '">' + \
                        '<a class="' + include_num + 'file_finder_2" id="not_thing" href="/upload?name=' + tool.url_pas(file_name) + '">' + file_alt + '</a>' + \
                    ''
            else:
                if re.search(r'^http(s)?:\/\/', match[3], flags = re.I):
                    return '<a id="out_link" href="' + match[3] + '">' + match[2] + '</a>'
                else:
                    return '<a class="' + include_num + 'link_finder" href="/w/' + match[3] + '">' + match[2] + '</a>'

    link_r = r'(!)?\[((?:(?!\]\().)+)\]\(([^\]]+)\)'
    link_do = link_render()
    data = re.sub(link_r, link_do, data)

    data = re.sub(r'\*\*((?:(?!\*\*).)+)\*\*', '<b>\1</b>', data)
    data = re.sub(r'__((?:(?!__).)+)__', '<i>\1</i>', data)

    
    return [data, plus_data, backlink]