fmouhart/pelican-autopages
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support pages for categories and tags, too

Browse Source
This commit is contained in:
Dan Crosta 2015-05-06 23:04:28 -04:00
parent 5e4257c91b
commit 63fafa802d
1 changed files with 33 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
autopages.py
View File

@ -1,9 +1,12 @@
import logging
import os import os
import os.path import os.path
from pelican import signals from pelican import signals
logger = logging.getLogger("autopages")
def yield_files(root): def yield_files(root):
root = os.path.realpath(os.path.abspath(root)) root = os.path.realpath(os.path.abspath(root))
for dirpath, dirnames, filenames in os.walk(root): for dirpath, dirnames, filenames in os.walk(root):
@ -19,21 +22,41 @@ def yield_files(root):
continue continue
yield os.path.join(dirpath, filename) yield os.path.join(dirpath, filename)
def test(article_generator): def make_page(readers, filename):
base_path, filename = os.path.split(filename)
page = readers.read_file(base_path, filename)
slug, _ = os.path.splitext(filename)
return slug, page
def make_pages(readers, path):
pages = {}
for filename in yield_files(path):
try:
slug, page = make_page(readers, filename)
except Exception:
logger.exception("Could not make autopage for %r", filename)
continue
pages[slug] = page
return pages
def create_autopages(article_generator):
settings = article_generator.settings settings = article_generator.settings
readers = article_generator.readers readers = article_generator.readers
path = settings.get("AUTHOR_PAGE_PATH", "authors")
author_pages = {} authors_path = settings.get("AUTHOR_PAGE_PATH", "authors")
for filename in yield_files(path): categories_path = settings.get("CATEGORY_PAGE_PATH", "categories")
base_path, filename = os.path.split(filename) tags_path = settings.get("TAG_PAGE_PATH", "tags")
page = readers.read_file(base_path, filename)
slug, _ = os.path.splitext(filename) author_pages = make_pages(readers, authors_path)
author_pages[slug] = page category_pages = make_pages(readers, categories_path)
tag_pages = make_pages(readers, tags_path)
for author, _ in article_generator.authors: for author, _ in article_generator.authors:
print "set author.page for %s to %r" % (author.slug, author_pages.get(author.slug, ""))
author.page = author_pages.get(author.slug, "") author.page = author_pages.get(author.slug, "")
for category, _ in article_generator.categories:
category.page = category_pages.get(category.slug, "")
for tag in article_generator.tags:
tag.page = tag_pages.get(tag.slug, "")
def register(): def register():
signals.article_generator_finalized.connect(test) signals.article_generator_finalized.connect(create_autopages)