Index: cms/converters.py |
=================================================================== |
--- a/cms/converters.py |
+++ b/cms/converters.py |
@@ -13,16 +13,17 @@ |
# You should have received a copy of the GNU General Public License |
# along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
from __future__ import unicode_literals |
import os |
import HTMLParser |
import re |
+import urlparse |
import jinja2 |
import markdown |
# Monkey-patch Markdown's isBlockLevel function to ensure that no paragraphs |
# are inserted into the <head> tag |
orig_isBlockLevel = markdown.util.isBlockLevel |
@@ -391,16 +392,17 @@ |
'toclist': self.toclist, |
} |
globals = { |
'get_string': self.get_string, |
'has_string': self.has_string, |
'get_page_content': self.get_page_content, |
'get_pages_metadata': self.get_pages_metadata, |
+ 'get_canonical_url': self.get_canonical_url, |
} |
for dirname, dictionary in [('filters', filters), |
('globals', globals)]: |
for filename in self._params['source'].list_files(dirname): |
root, ext = os.path.splitext(filename) |
if ext.lower() != '.py': |
continue |
@@ -512,16 +514,32 @@ |
filter_value = [filter_value] |
for option in filter_value: |
if str(option) not in metadata[filter_name]: |
return False |
elif filter_value != metadata[filter_name]: |
return False |
return True |
+ def get_canonical_url(self, page): |
+ """Return canonical URL for the page (without locale code)""" |
+ try: |
+ base_url = self._params['site_url'] |
+ except KeyError: |
+ raise Exception('You must configure `siteurl` to use' |
+ '`get_canonical_url()`') |
+ |
+ locale, page_url = self._params['source'].resolve_link( |
+ page, self._params['locale'] |
+ ) |
+ # Remove the locale component that `resolve_link` adds at the |
+ # beginning. |
+ page_url = page_url[len(locale) + 1:] |
+ return urlparse.urljoin(base_url, page_url) |
+ |
def toclist(self, content): |
toc_re = r'<h(\d)\s[^<>]*\bid="([^<>"]+)"[^<>]*>(.*?)</h\1>' |
flat = [] |
for match in re.finditer(toc_re, content, re.S): |
flat.append({ |
'level': int(match.group(1)), |
'anchor': jinja2.Markup(match.group(2)).unescape(), |
'title': jinja2.Markup(match.group(3)).unescape(), |