71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Read More Link
|
|
===========================
|
|
|
|
This plugin inserts an inline "read more" or "continue" link into the last html element of the object summary.
|
|
|
|
For more information, please visit: http://vuongnguyen.com/creating-inline-read-more-link-python-pelican-lxml.html
|
|
|
|
"""
|
|
|
|
from pelican import signals, contents
|
|
from pelican.utils import truncate_html_words
|
|
|
|
try:
|
|
from lxml.html import fragment_fromstring, fragments_fromstring, tostring
|
|
from lxml.etree import ParserError
|
|
except ImportError:
|
|
raise Exception("Unable to find lxml. To use READ_MORE_LINK, you need lxml")
|
|
|
|
|
|
def insert_into_last_element(html, element):
|
|
"""
|
|
function to insert an html element into another html fragment
|
|
example:
|
|
html = '<p>paragraph1</p><p>paragraph2...</p>'
|
|
element = '<a href="/read-more/">read more</a>'
|
|
---> '<p>paragraph1</p><p>paragraph2...<a href="/read-more/">read more</a></p>'
|
|
"""
|
|
try:
|
|
item = fragment_fromstring(element)
|
|
except ParserError, TypeError:
|
|
item = fragment_fromstring('<span></span>')
|
|
|
|
try:
|
|
doc = fragments_fromstring(html)
|
|
doc[-1].append(item)
|
|
|
|
return ''.join(tostring(e) for e in doc)
|
|
except ParserError, TypeError:
|
|
return ''
|
|
|
|
def insert_read_more_link(instance):
|
|
"""
|
|
Insert an inline "read more" link into the last element of the summary
|
|
:param instance:
|
|
:return:
|
|
"""
|
|
|
|
# only deals with Article type
|
|
if type(instance) != contents.Article: return
|
|
|
|
|
|
SUMMARY_MAX_LENGTH = instance.settings.get('SUMMARY_MAX_LENGTH')
|
|
READ_MORE_LINK = instance.settings.get('READ_MORE_LINK', None)
|
|
READ_MORE_LINK_FORMAT = instance.settings.get('READ_MORE_LINK_FORMAT',
|
|
'<a class="read-more" href="/{url}">{text}</a>')
|
|
|
|
if not (SUMMARY_MAX_LENGTH and READ_MORE_LINK and READ_MORE_LINK_FORMAT): return
|
|
|
|
if hasattr(instance, '_summary') and instance._summary:
|
|
summary = instance._summary
|
|
else:
|
|
summary = truncate_html_words(instance.content, SUMMARY_MAX_LENGTH)
|
|
|
|
if summary<instance.content:
|
|
read_more_link = READ_MORE_LINK_FORMAT.format(url=instance.url, text=READ_MORE_LINK)
|
|
instance._summary = insert_into_last_element(summary, read_more_link)
|
|
|
|
def register():
|
|
signals.content_object_init.connect(insert_read_more_link) |