share_post.py 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. """
  2. Share Post
  3. ==========
  4. This plugin adds share URL to article. These links are textual which means no
  5. online tracking of your readers.
  6. """
  7. from bs4 import BeautifulSoup
  8. try:
  9. from urllib.parse import quote
  10. except ImportError:
  11. from urllib import quote
  12. from pelican import signals, contents
  13. from pelican.generators import ArticlesGenerator, PagesGenerator
  14. def article_title(content):
  15. main_title = BeautifulSoup(content.title, 'html.parser').get_text().strip()
  16. sub_title = ''
  17. if hasattr(content, 'subtitle'):
  18. sub_title = ' ' + BeautifulSoup(content.subtitle, 'html.parser').get_text().strip()
  19. return quote(('%s%s' % (main_title, sub_title)).encode('utf-8'))
  20. def article_url(content):
  21. site_url = content.settings['SITEURL']
  22. return quote(('%s/%s' % (site_url, content.url)).encode('utf-8'))
  23. def article_summary(content):
  24. return quote(BeautifulSoup(content.summary, 'html.parser').get_text().strip().encode('utf-8'))
  25. def share_post(content):
  26. if isinstance(content, contents.Static):
  27. return
  28. title = article_title(content)
  29. url = article_url(content)
  30. summary = article_summary(content)
  31. tweet = ('%s%s%s' % (title, quote(' '), url)).encode('utf-8')
  32. diaspora_link = 'https://sharetodiaspora.github.io/?title=%s&url=%s' % (title, url)
  33. facebook_link = 'http://www.facebook.com/sharer/sharer.php?u=%s' % url
  34. gplus_link = 'https://plus.google.com/share?url=%s' % url
  35. twitter_link = 'http://twitter.com/home?status=%s' % tweet
  36. linkedin_link = 'https://www.linkedin.com/shareArticle?mini=true&url=%s&title=%s&summary=%s&source=%s' % (
  37. url, title, summary, url
  38. )
  39. mail_link = 'mailto:?subject=%s&body=%s' % (title, url)
  40. share_links = {
  41. 'diaspora': diaspora_link,
  42. 'twitter': twitter_link,
  43. 'facebook': facebook_link,
  44. 'google-plus': gplus_link,
  45. 'linkedin': linkedin_link,
  46. 'email': mail_link
  47. }
  48. content.share_post = share_links
  49. def run_plugin(generators):
  50. for generator in generators:
  51. if isinstance(generator, ArticlesGenerator):
  52. for article in generator.articles:
  53. share_post(article)
  54. elif isinstance(generator, PagesGenerator):
  55. for page in generator.pages:
  56. share_post(page)
  57. def register():
  58. try:
  59. signals.all_generators_finalized.connect(run_plugin)
  60. except AttributeError:
  61. # NOTE: This results in #314 so shouldn't really be relied on
  62. # https://github.com/getpelican/pelican-plugins/issues/314
  63. signals.content_object_init.connect(share_post)