sambuc/pelican-plugins
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #131 from calfzhou/neighbors

Browse Source 
Make neighbors plugin support translated articles.
This commit is contained in:
Justin Mayer
2014-01-23 14:59:14 -08:00
parent c4d7a761c8 211d835f30
commit 5e6307d9e2
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
neighbors/neighbors.py
View File

@@ -18,10 +18,22 @@ def iter3(seq):
nxt, cur = cur, prv nxt, cur = cur, prv
yield nxt, cur, None yield nxt, cur, None
def get_translation(article, prefered_language):
if not article:
return None
for translation in article.translations:
if translation.lang == prefered_language:
return translation
return article
def neighbors(generator): def neighbors(generator):
for nxt, cur, prv in iter3(generator.articles): for nxt, cur, prv in iter3(generator.articles):
cur.next_article = nxt cur.next_article = nxt
cur.prev_article = prv cur.prev_article = prv
for translation in cur.translations:
translation.next_article = get_translation(nxt, translation.lang)
translation.prev_article = get_translation(prv, translation.lang)
def register(): def register():
signals.article_generator_finalized.connect(neighbors) signals.article_generator_finalized.connect(neighbors)