func_render.py 4.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. from .func_tool import *
  2. from .func_render_namumark import class_do_render_namumark
  3. # 커스텀 마크 언젠간 다시 추가 예정
  4. class class_do_render:
  5. def __init__(self, conn, lang_data = {}, markup = ''):
  6. self.conn = conn
  7. if lang_data == '{}':
  8. lang_data = {
  9. 'toc' : 'toc',
  10. 'category' : 'category'
  11. }
  12. self.lang_data = lang_data
  13. self.markup = markup
  14. def do_render(self, doc_name, doc_data, data_type, data_in):
  15. curs = self.conn.cursor()
  16. doc_set = {}
  17. if data_type == 'from':
  18. doc_set['doc_from'] = 'O'
  19. data_type = 'view'
  20. data_in = (data_in + '_') if data_in != '' else ''
  21. doc_set['doc_include'] = data_in
  22. rep_data = self.markup
  23. if rep_data == '' and doc_name != '':
  24. curs.execute(db_change("select set_data from data_set where doc_name = ? and set_name = 'document_markup'"), [doc_name])
  25. db_data = curs.fetchall()
  26. if db_data and db_data[0][0] != '' and db_data[0][0] != 'normal':
  27. rep_data = db_data[0][0]
  28. if rep_data == '':
  29. curs.execute(db_change('select data from other where name = "markup"'))
  30. db_data = curs.fetchall()
  31. rep_data = db_data[0][0] if db_data else 'namumark'
  32. if rep_data == 'namumark' or rep_data == 'namumark_beta':
  33. data_end = class_do_render_namumark(curs, doc_name, doc_data, doc_set, self.lang_data)()
  34. elif rep_data == 'raw':
  35. data_end = [html.escape(doc_data).replace('\n', '<br>'), '', {}]
  36. else:
  37. data_end = [doc_data, '', {}]
  38. if data_type == 'thread' or data_type == 'api_thread':
  39. def do_thread_a_change(match):
  40. data = match[2].replace('#', '')
  41. data_split = data.split('-')
  42. if match[1] == 'topic_a' or len(data_split) == 1:
  43. return '<a href="' + match[2] + '">' + match[2] + '</a>'
  44. elif match[1] == 'topic_a_post' and len(data_split) == 3:
  45. return '<a href="/bbs/w/' + data_split[2] + '/' + data_split[1] + '#' + data_split[0] + '">#' + data_split[0] + '-' + data_split[1] + '</a>'
  46. elif len(data_split) == 2:
  47. return '<a href="/thread/' + data_split[1] + '#' + data_split[0] + '">' + match[2] + '</a>'
  48. else:
  49. return ''
  50. data_end[0] = re.sub(r'&lt;(topic_a(?:_post|_thread)?)&gt;((?:(?!&lt;\/topic_a(?:_post|_thread)?&gt;).)+)&lt;\/topic_a(?:_post|_thread)?&gt;', do_thread_a_change, data_end[0])
  51. data_end[0] = re.sub(r'&lt;topic_call&gt;@(?P<in>(?:(?!&lt;\/topic_call&gt;).)+)&lt;\/topic_call&gt;', '<a href="/w/user:\\g<in>">@\\g<in></a>', data_end[0])
  52. if data_type == 'backlink':
  53. curs.execute(db_change("delete from back where link = ?"), [doc_name])
  54. curs.execute(db_change("delete from back where title = ? and type = 'no'"), [doc_name])
  55. curs.execute(db_change("delete from data_set where doc_name = ? and set_name = 'link_count'"), [doc_name])
  56. curs.execute(db_change("delete from data_set where doc_name = ? and set_name = 'doc_type'"), [doc_name])
  57. backlink = data_end[2]['backlink'] if 'backlink' in data_end[2] else []
  58. if backlink != []:
  59. curs.executemany(db_change("insert into back (link, title, type, data) values (?, ?, ?, ?)"), data_end[2]['backlink'])
  60. curs.execute(db_change("delete from back where title = ? and type = 'no'"), [doc_name])
  61. link_count = 0
  62. if 'link_count' in data_end[2]:
  63. link_count = data_end[2]['link_count']
  64. curs.execute(db_change("insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'link_count', ?)"), [doc_name, link_count])
  65. if 'redirect' in data_end[2] and data_end[2]['redirect'] == 1:
  66. curs.execute(db_change("insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'doc_type', 'redirect')"), [doc_name])
  67. else:
  68. curs.execute(db_change("insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'doc_type', '')"), [doc_name])
  69. self.conn.commit()
  70. return [data_end[0], data_end[1], data_end[2]]