openNAMU__openNAMU/mark.py

from set_mark.start import start

import re
import html
import sqlite3
from urllib import parse
import time
import threading

def load_conn2(data):
    global conn
    global curs

    conn = data
    curs = conn.cursor()

def send_parser(data):
    data = html.escape(data)
    
    javascript = re.compile('javascript:', re.I)
    
    data = javascript.sub('', data)
    data = re.sub('&lt;a href="(?:[^"]*)"&gt;(?P<in>(?:(?!&lt;).)*)&lt;\/a&gt;', '<a href="' + parse.quote('\g<in>').replace('/','%2F') + '">\g<in></a>', data)  
    
    return data
    
def plusing(name, link, backtype):
    curs.execute("select title from back where title = ? and link = ? and type = ?", [link, name, backtype])
    if not curs.fetchall():
        curs.execute("insert into back (title, link, type) values (?, ?, ?)", [link, name, backtype])

def namumark(title, data, num):
    data = start(conn, data, title)
    if num == 1:        
        for back_data in data[2]:
            thread_start = threading.Thread(target = plusing, args = [back_data[0], back_data[1], back_data[2]])
            thread_start.start()
            thread_start.join()

        conn.commit()
        
    return data[0] + data[1]