From 0426fce3b39ebedf6b92bfe6b18aa08987b11ea2 Mon Sep 17 00:00:00 2001 From: Scott Bahling Date: Sat, 23 Mar 2013 17:28:08 +0100 Subject: Improve header id generation --- panfry/document.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'panfry/document.py') diff --git a/panfry/document.py b/panfry/document.py index 7bc65c2..edf1cf8 100755 --- a/panfry/document.py +++ b/panfry/document.py @@ -42,18 +42,23 @@ class Document: ''' re_link1 = re.compile(r'(\[(?P[^\]^]+)\])\((?P#.*)\)', re.S) re_link2 = re.compile(r'(\[(?P[^\]^]+)\][^(])', re.S) + re_link3 = re.compile(r'(\[(?P[^\]^]+)\])\[(?P[^\]].*)\]', re.S) source = re_link1.sub(self._get_cross_link, source) source = re_link2.sub(self._get_cross_link, source) + source = re_link3.sub(self._get_cross_link, source) return source def _get_cross_link(self, m): - ref_txt = m.group('s') + if 'h' in m.groupdict(): + heading = m.group('h') + else: + heading = m.group('s') for page in self.pages: for tocitem in page.toc: - if tocitem.heading == ref_txt.replace('\n', ' '): - return '[%s](%s#%s)' % (ref_txt, + if tocitem.heading == heading.replace('\n', ' '): + return '[%s](%s#%s)' % (heading, page.htmlfile, tocitem.header_id, ) -- cgit v1.2.3