1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980 |
- """
- 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 = {}
- for thread in threads:
- thread_dict[thread['id']] = thread['title']
-
- posts = Paginator(disqus.posts.list,
- forum=generator.settings['DISQUS_SITENAME'])
-
- child_dict = {}
- for post in posts:
- if post['id'] not in child_dict.keys():
- child_dict[post['id']] = []
- if post['parent'] is not None:
- if str(post['parent']) not in child_dict.keys():
- child_dict[str(post['parent'])] = []
- child_dict[str(post['parent'])].append(post)
-
- post_dict = {}
- for post in posts:
- build_post_dict(post_dict, child_dict, thread_dict, post)
- for article in generator.articles:
- if article.title in post_dict:
- article.disqus_comments = post_dict[article.title]
- article.disqus_comment_count = sum([
- postcounter(post) for post in post_dict[article.title]])
- def postcounter(node):
- return 1 + sum([postcounter(n) for n in node['children']])
- def build_post_dict(post_dict, child_dict, thread_dict, post):
- if post['thread'] not in thread_dict.keys():
- return
- build_child_dict(child_dict, post)
- if post['parent'] is not None:
- return
- if thread_dict[post['thread']] not in post_dict.keys():
- post_dict[thread_dict[post['thread']]] = []
- post_dict[thread_dict[post['thread']]].append(post)
- def build_child_dict(child_dict, post):
- post['children'] = child_dict[post['id']]
- for child in child_dict[post['id']]:
- build_child_dict(child_dict, child)
- def register():
- signals.initialized.connect(initialized)
- signals.article_generator_finalized.connect(disqus_static)
|