blob: fbb24260eb9bb4dd9f46066079b814e381682878 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
"""
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)
|
