forked from varia/varia.website
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
|
"""
|
||
|
|
Related posts plugin for Pelican
|
||
|
|
================================
|
||
|
|
|
||
|
|
Adds related_posts variable to article's context
|
||
|
|
"""
|
||
|
|
|
||
|
|
from pelican import signals
|
||
|
|
from collections import Counter
|
||
|
|
from itertools import chain
|
||
|
|
|
||
|
|
|
||
|
|
def add_related_posts(generator):
|
||
|
|
# get the max number of entries from settings
|
||
|
|
# or fall back to default (5)
|
||
|
|
numentries = generator.settings.get('RELATED_POSTS_MAX', 5)
|
||
|
|
# Skip all posts in the same category as the article
|
||
|
|
skipcategory = generator.settings.get('RELATED_POSTS_SKIP_SAME_CATEGORY', False)
|
||
|
|
for article in chain(generator.articles, generator.drafts):
|
||
|
|
# set priority in case of forced related posts
|
||
|
|
if hasattr(article,'related_posts'):
|
||
|
|
# split slugs
|
||
|
|
related_posts = article.related_posts.split(',')
|
||
|
|
posts = []
|
||
|
|
# get related articles
|
||
|
|
for slug in related_posts:
|
||
|
|
i = 0
|
||
|
|
slug = slug.strip()
|
||
|
|
for a in generator.articles:
|
||
|
|
if i >= numentries: # break in case there are max related psots
|
||
|
|
break
|
||
|
|
if a.slug == slug:
|
||
|
|
posts.append(a)
|
||
|
|
i += 1
|
||
|
|
|
||
|
|
article.related_posts = posts
|
||
|
|
else:
|
||
|
|
# no tag, no relation
|
||
|
|
if not hasattr(article, 'tags'):
|
||
|
|
continue
|
||
|
|
|
||
|
|
# score = number of common tags
|
||
|
|
related = chain(*(generator.tags[tag] for tag in article.tags))
|
||
|
|
if skipcategory:
|
||
|
|
related = (other for other in related
|
||
|
|
if other.category != article.category)
|
||
|
|
scores = Counter(related)
|
||
|
|
|
||
|
|
# remove itself
|
||
|
|
scores.pop(article, None)
|
||
|
|
|
||
|
|
article.related_posts = [other for other, count
|
||
|
|
in scores.most_common(numentries)]
|
||
|
|
|
||
|
|
def register():
|
||
|
|
signals.article_generator_finalized.connect(add_related_posts)
|