# -*- coding: utf-8 -*- from __future__ import unicode_literals import os.path import logging from pelican import signals logger = logging.getLogger(__name__) class AliasGenerator(object): TEMPLATE = """
""" def __init__(self, context, settings, path, theme, output_path, *args): self.output_path = output_path self.context = context self.alias_delimiter = settings.get('ALIAS_DELIMITER', ',') def create_alias(self, page, alias): # If path starts with a /, remove it if alias[0] == '/': relative_alias = alias[1:] else: relative_alias = alias path = os.path.join(self.output_path, relative_alias) directory, filename = os.path.split(path) try: os.makedirs(directory) except OSError: pass if filename == '': path = os.path.join(path, 'index.html') logger.info('[alias] Writing to alias file %s' % path) with open(path, 'w') as fd: fd.write(self.TEMPLATE.format(destination_path=page.url)) def generate_output(self, writer): pages = self.context['pages'] + self.context['articles'] for page in pages: aliases = page.metadata.get('alias', []) if type(aliases) != list: aliases = aliases.split(self.alias_delimiter) for alias in aliases: alias = alias.strip() logger.info('[alias] Processing alias %s' % alias) self.create_alias(page, alias) def get_generators(generators): return AliasGenerator def register(): signals.get_generators.connect(get_generators)