Make neighbors plugin support translated articles.

This commit is contained in:
zhouji
2014-01-23 09:26:47 +08:00
parent c4d7a761c8
commit 211d835f30

View File

@@ -18,10 +18,22 @@ def iter3(seq):
nxt, cur = cur, prv
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):
for nxt, cur, prv in iter3(generator.articles):
cur.next_article = nxt
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():
signals.article_generator_finalized.connect(neighbors)