123456789101112131415161718192021222324252627282930313233343536373839404142434445 |
- # -*- coding: utf-8 -*-
- """
- Disqus static comment plugin for Pelican
- ====================================
- This plugin adds a disqus_comments property to all articles.
- Comments are fetched at generation time using disqus API.
- """
- from disqusapi import DisqusAPI, Paginator
- from pelican import signals
- def initialized(pelican):
- from pelican.settings import _DEFAULT_CONFIG
- _DEFAULT_CONFIG.setdefault('DISQUS_SECRET_KEY', '')
- _DEFAULT_CONFIG.setdefault('DISQUS_PUBLIC_KEY', '')
- if pelican:
- pelican.settings.setdefault('DISQUS_SECRET_KEY', '')
- pelican.settings.setdefault('DISQUS_PUBLIC_KEY', '')
- def disqus_static(generator):
- disqus = DisqusAPI(generator.settings['DISQUS_SECRET_KEY'],
- generator.settings['DISQUS_PUBLIC_KEY'])
- threads = Paginator(disqus.threads.list,
- forum=generator.settings['DISQUS_SITENAME'])
- thread_dict = {} # disqus thread id => title
- for thread in threads:
- thread_dict[thread['id']] = thread['title']
- posts = Paginator(disqus.posts.list, forum=generator.settings['DISQUS_SITENAME'])
- post_dict = {} # title => [post1, post2, ...]
- for post in posts:
- if post['thread'] not in thread_dict.keys():
- continue
- if thread_dict[post['thread']] not in post_dict.keys():
- post_dict[thread_dict[post['thread']]] = []
- post_dict[thread_dict[post['thread']]].append(post)
- for article in generator.articles:
- if article.title in post_dict:
- article.disqus_comments = post_dict[article.title]
- def register():
- signals.initialized.connect(initialized)
- signals.article_generator_finalized.connect(disqus_static)
|