share_post.py 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  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. diaspora_link = 'https://sharetodiaspora.github.io/?title=%s&url=%s' % (title, url)
  32. facebook_link = 'http://www.facebook.com/sharer/sharer.php?u=%s' % url
  33. gplus_link = 'https://plus.google.com/share?url=%s' % url
  34. twitter_link = 'https://twitter.com/intent/tweet?text=%s&url=%s' % (title, url)
  35. linkedin_link = 'https://www.linkedin.com/shareArticle?mini=true&url=%s&title=%s&summary=%s&source=%s' % (
  36. url, title, summary, url
  37. )
  38. mail_link = 'mailto:?subject=%s&body=%s' % (title, url)
  39. share_links = {
  40. 'diaspora': diaspora_link,
  41. 'twitter': twitter_link,
  42. 'facebook': facebook_link,
  43. 'google-plus': gplus_link,
  44. 'linkedin': linkedin_link,
  45. 'email': mail_link
  46. }
  47. content.share_post = share_links
  48. def run_plugin(generators):
  49. for generator in generators:
  50. if isinstance(generator, ArticlesGenerator):
  51. for article in generator.articles:
  52. share_post(article)
  53. elif isinstance(generator, PagesGenerator):
  54. for page in generator.pages:
  55. share_post(page)
  56. def register():
  57. try:
  58. signals.all_generators_finalized.connect(run_plugin)
  59. except AttributeError:
  60. # NOTE: This results in #314 so shouldn't really be relied on
  61. # https://github.com/getpelican/pelican-plugins/issues/314
  62. signals.content_object_init.connect(share_post)