|
|
@@ -3,6 +3,7 @@ from .set_mark.namumark import namumark, link_fix
|
|
|
from .set_mark.tool import *
|
|
|
|
|
|
import re
|
|
|
+import os
|
|
|
import html
|
|
|
import sqlite3
|
|
|
import asyncio
|
|
|
@@ -10,6 +11,12 @@ import threading
|
|
|
import urllib.parse
|
|
|
import multiprocessing
|
|
|
|
|
|
+if os.path.exists('route/tool/set_mark/custom.py'):
|
|
|
+ from .set_mark.custom import custom_mark
|
|
|
+else:
|
|
|
+ def custom_mark(conn, data, title, num, include):
|
|
|
+ return [data, '', []]
|
|
|
+
|
|
|
def load_conn2(data):
|
|
|
global conn
|
|
|
global curs
|
|
|
@@ -42,6 +49,8 @@ def render_do(title, data, num, include):
|
|
|
rep_data = curs.fetchall()
|
|
|
if rep_data[0][0] == 'namumark':
|
|
|
data = namumark(conn, data, title, num, include)
|
|
|
+ elif rep_data[0][0] == 'custom':
|
|
|
+ data = custom_mark(conn, data, title, num, include)
|
|
|
elif rep_data[0][0] == 'raw':
|
|
|
data = [data, '', []]
|
|
|
else:
|