##// END OF EJS Templates
Fixed small issue with reflinks to non-existing posts
neko259 -
r314:18a208f8 default
parent child Browse files
Show More
@@ -1,106 +1,106 b''
1 1 from django.core.urlresolvers import reverse
2 2 import markdown
3 3 from markdown.inlinepatterns import Pattern
4 4 from markdown.util import etree
5 5 import boards
6 6
7 7 __author__ = 'neko259'
8 8
9 9
10 10 AUTOLINK_PATTERN = r'(https?://\S+)'
11 11 QUOTE_PATTERN = r'^(?<!>)(>[^>].+)$'
12 12 REFLINK_PATTERN = r'((>>)(\d+))'
13 13 SPOILER_PATTERN = r'%%(.+)%%'
14 14 COMMENT_PATTERN = r'^(//(.+))'
15 15
16 16 class AutolinkPattern(Pattern):
17 17 def handleMatch(self, m):
18 18 link_element = etree.Element('a')
19 19 href = m.group(2)
20 20 link_element.set('href', href)
21 21 link_element.text = href
22 22
23 23 return link_element
24 24
25 25
26 26 class QuotePattern(Pattern):
27 27 def handleMatch(self, m):
28 28 quote_element = etree.Element('span')
29 29 quote_element.set('class', 'quote')
30 30 quote_element.text = m.group(2)
31 31
32 32 return quote_element
33 33
34 34
35 35 class ReflinkPattern(Pattern):
36 36 def handleMatch(self, m):
37 37 post_id = m.group(4)
38 38
39 39 posts = boards.models.Post.objects.filter(id=post_id)
40 40 if posts.count() > 0:
41 41 ref_element = etree.Element('a')
42 42
43 43 post = posts[0]
44 44 if post.thread:
45 45 link = reverse(boards.views.thread, kwargs={'post_id': post.thread.id}) \
46 46 + '#' + post_id
47 47 else:
48 48 link = reverse(boards.views.thread, kwargs={'post_id': post_id})
49 49
50 50 ref_element.set('href', link)
51 51 ref_element.text = m.group(2)
52 52
53 return ref_element
53 return ref_element
54 54
55 55
56 56 class SpoilerPattern(Pattern):
57 57 def handleMatch(self, m):
58 58 quote_element = etree.Element('span')
59 59 quote_element.set('class', 'spoiler')
60 60 quote_element.text = m.group(2)
61 61
62 62 return quote_element
63 63
64 64
65 65 class CommentPattern(Pattern):
66 66 def handleMatch(self, m):
67 67 quote_element = etree.Element('span')
68 68 quote_element.set('class', 'comment')
69 69 quote_element.text = '//' + m.group(3)
70 70
71 71 return quote_element
72 72
73 73
74 74 class NeboardMarkdown(markdown.Extension):
75 75 def extendMarkdown(self, md, md_globals):
76 76 self._add_neboard_patterns(md)
77 77 self._delete_patterns(md)
78 78
79 79 def _delete_patterns(self, md):
80 80 del md.parser.blockprocessors['quote']
81 81
82 82 del md.inlinePatterns['image_link']
83 83 del md.inlinePatterns['image_reference']
84 84
85 85 def _add_neboard_patterns(self, md):
86 86 autolink = AutolinkPattern(AUTOLINK_PATTERN, md)
87 87 quote = QuotePattern(QUOTE_PATTERN, md)
88 88 reflink = ReflinkPattern(REFLINK_PATTERN, md)
89 89 spoiler = SpoilerPattern(SPOILER_PATTERN, md)
90 90 comment = CommentPattern(COMMENT_PATTERN, md)
91 91
92 92 md.inlinePatterns[u'autolink_ext'] = autolink
93 93 md.inlinePatterns[u'spoiler'] = spoiler
94 94 md.inlinePatterns[u'comment'] = comment
95 95 md.inlinePatterns[u'reflink'] = reflink
96 96 md.inlinePatterns[u'quote'] = quote
97 97
98 98
99 99 def makeExtension(configs=None):
100 100 return NeboardMarkdown(configs=configs)
101 101
102 102 neboard_extension = makeExtension()
103 103
104 104
105 105 def markdown_extended(markup):
106 106 return markdown.markdown(markup, [neboard_extension], safe_mode=True)
General Comments 0
You need to be logged in to leave comments. Login now