openNAMU__openNAMU/set_mark/link.py

import json
import sqlite3
import re
from urllib import parse
import hashlib

json_data = open('set.json').read()
set_data = json.loads(json_data)

conn = sqlite3.connect(set_data['db'] + '.db')
curs = conn.cursor()

def url_pas(data):
    return(parse.quote(data).replace('/','%2F'))

def sha224(data):
    return(hashlib.sha224(bytes(data, 'utf-8')).hexdigest())

def link(title, data, num, category, backlink):
    data = data.replace('\', '\\')
    
    m = re.findall("\[\[(분류:(?:(?:(?!\]\]).)*))\]\]", data)
    for g in m:
        if(title != g):
            if(num == 1):
                backlink += [[title, g, 'cat']]
                
            if(category == ''):
                curs.execute("select title from data where title = ?", [g])
                exists = curs.fetchall()
                if(exists):
                    red = ""
                else:
                    red = 'class="not_thing"'
                    
                category += '<a ' + red + ' href="/w/' + url_pas(g) + '">' + re.sub("분류:", "", g) + '</a>'
            else:
                curs.execute("select title from data where title = ?", [g])
                exists = curs.fetchall()
                if(exists):
                    red = ""
                else:
                    red = 'class="not_thing"'
                    
                category += ' / ' + '<a ' + red + ' href="/w/' + url_pas(g) + '">' + re.sub("분류:", "", g) + '</a>'
        
        data = re.sub("\[\[(분류:(?:(?:(?!\]\]).)*))\]\]", '', data, 1)
    
    test = re.findall('\[\[wiki:([^|\]]+)(?:\|([^\]]+))?\]\]', data)
    if(test):
        for wiki in test:
            if(wiki[1]):
                data = re.sub('\[\[wiki:([^|\]]+)(?:\|([^\]]+))?\]\]', '<a id="inside" href="/' + wiki[0] + '">' + wiki[1] + '</a>', data, 1)
            else:
                data = re.sub('\[\[wiki:([^|\]]+)(?:\|([^\]]+))?\]\]', '<a id="inside" href="/' + wiki[0] + '">' + wiki[0] + '</a>', data, 1)
     
    data = re.sub("\[\[(?::(?P<in>(?:분류|파일):(?:(?:(?!\]\]).)*)))\]\]", "[[\g<in>]]", data)

    a = re.findall('\[\[\.\.\/(\|(?:(?!]]).)+)?]]', data)
    for i in a:
        b = re.search('(.*)\/', title)
        if(b):
            m = b.groups()
            if(i):
                data = re.sub('\[\[\.\.\/(\|((?!]]).)+)?]]', '[[' + m[0] + i + ']]', data, 1)
            else:
                data = re.sub('\[\[\.\.\/(\|((?!]]).)+)?]]', '[[' + m[0] + ']]', data, 1)
        else:
            if(i):
                data = re.sub('\[\[\.\.\/(\|((?!]]).)+)?]]', '[[' + title + i + ']]', data, 1)
            else:
                data = re.sub('\[\[\.\.\/(\|((?!]]).)+)?]]', '[[' + title + ']]', data, 1)

    data = re.sub('\[\[(?P<in>\/(?:(?!]]|\|).)+)(?P<out>\|(?:(?:(?!]]).)+))?]]', '[[' + title + '\g<in>\g<out>]]', data)
                
    link = re.compile('\[\[((?:(?!\[\[|\]\]|\|).)*)(?:\|((?:(?!\[\[|\]\]).)*))?\]\]')
    while(1):
        l_d = link.search(data)
        if(l_d):
            d = l_d.groups()
            if(re.search('^(?:파일|외부):', d[0])):
                width = ''
                height = ''
                align = ''
                span = ['', '']
                
                try:        
                    w_d = re.search('width=([0-9]+(?:[a-z%]+)?)', d[1])
                    if(w_d):
                        width = 'width="' + w_d.groups()[0] + '" '
                    
                    h_d = re.search('height=([0-9]+(?:[a-z%]+)?)', d[1])
                    if(h_d):
                        height = 'height="' + h_d.groups()[0] + '" '
                        
                    a_d = re.search('align=(center|right)', d[1])
                    if(a_d):
                        span[0] = '<span style="display: block; text-align: ' + a_d.groups()[0] + ';">'
                        span[1] = '</span>'
                except:
                    pass
                    
                f_d = re.search('^파일:([^.]+)\.(.+)$', d[0])
                if(f_d):
                    if(not re.search("^파일:([^\n]*)", title)):
                        if(num == 1):
                            backlink += [[title, d[0], 'file']]
                        
                    img = span[0] + '<img src="/image/' + sha224(f_d.groups()[0]) + '.' + f_d.groups()[1] + '" ' + width + height + '>' + span[1]
                    data = link.sub(img, data, 1)
                else:
                    img = span[0] + '<img src="' + re.sub('^외부:', '', d[0]) + '" ' + width + height + '>' + span[1]
                    data = link.sub(img, data, 1)
                                    
            elif(re.search('^https?:\/\/', d[0])):
                view = d[0]
                try:
                    if(re.search('(.+)', d[1])):
                        view = d[1]
                except:
                    pass
                
                data = link.sub('<a class="out_link" rel="nofollow" href="' + d[0] + '">' + view + '</a>', data, 1)
            else:
                view = d[0].replace('\\\\', '<slash>').replace('\\', '').replace('<slash>', '\\')
                try:
                    if(re.search('(.+)', d[1])):
                        view = d[1].replace('\\\\', '<slash>').replace('\\', '').replace('<slash>', '\\')
                except:
                    pass        
                    
                sh = ''
                s_d = re.search('#((?:(?!x27;|#).)+)$', d[0])
                if(s_d):
                    href = re.sub('#((?:(?!x27;|#).)+)$', '', d[0])
                    sh = '#' + s_d.groups()[0]
                else:
                    href = d[0]
                    
                if(d[0] == title):
                    data = link.sub('<b>' + view + '</b>', data, 1)
                elif(re.search('^#', d[0])):
                    data = link.sub('<a title="' + sh + '" href="' + sh + '">' + view + '</a>', data, 1)
                else:                    
                    a = href.replace('&#x27;', "'").replace('&quot;', '"').replace('\\\\', '<slash>').replace('\\', '').replace('<slash>', '\\')
                    
                    if(num == 1):
                        backlink += [[title, a, '']]
                    
                    curs.execute("select title from data where title = ?", [a])
                    if(not curs.fetchall()):
                        no = 'class="not_thing"'
                        
                        if(num == 1):
                            backlink += [[title, a, 'no']]
                    else:
                        no = ''
                    
                    data = link.sub('<a ' + no + ' title="' + href + sh + '" href="/w/' + url_pas(a) + sh + '">' + view.replace('\\', '\\\\') + '</a>', data, 1)
        else:
            break
            
    data = data.replace('\\', '&#92;')

    return([data, category, backlink])