src/pyams_utils/text.py
changeset 289 c8e21d7dd685
child 292 b338586588ad
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/pyams_utils/text.py	Wed Dec 05 12:45:56 2018 +0100
@@ -0,0 +1,203 @@
+#
+# Copyright (c) 2008-2015 Thierry Florac <tflorac AT ulthar.net>
+# All Rights Reserved.
+#
+# This software is subject to the provisions of the Zope Public License,
+# Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.
+# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
+# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
+# FOR A PARTICULAR PURPOSE.
+#
+
+__docformat__ = 'restructuredtext'
+
+import html
+
+import docutils.core
+from pyramid.interfaces import IRequest
+from zope.interface import Interface
+from zope.schema.vocabulary import SimpleTerm, SimpleVocabulary
+
+from pyams_utils.adapter import ContextRequestAdapter, ContextRequestViewAdapter, adapter_config
+from pyams_utils.interfaces.tales import ITALESExtension
+from pyams_utils.interfaces.text import IHTMLRenderer
+from pyams_utils.request import check_request
+from pyams_utils.vocabulary import vocabulary_config
+
+
+def get_text_start(text, length, max=0):
+    """Get first words of given text with maximum given length
+
+    If *max* is specified, text is shortened only if remaining text is longer this value
+
+    :param str text: initial text
+    :param integer length: maximum length of resulting text
+    :param integer max: if > 0, *text* is shortened only if remaining text is longer than max
+
+    >>> from pyams_utils.text import get_text_start
+    >>> get_text_start('This is a long string', 10)
+    'This is a&#133;'
+    >>> get_text_start('This is a long string', 20)
+    'This is a long&#133;'
+    >>> get_text_start('This is a long string', 20, 7)
+    'This is a long string'
+    """
+    result = text or ''
+    if length > len(result):
+        return result
+    index = length - 1
+    text_length = len(result)
+    while (index > 0) and (result[index] != ' '):
+        index -= 1
+    if (index > 0) and (text_length > index + max):
+        return result[:index] + '&#133;'
+    return text
+
+
+@adapter_config(name='truncate', context=(Interface, Interface, Interface), provides=ITALESExtension)
+class TruncateCharsTalesExtension(ContextRequestViewAdapter):
+    """extension:truncate(value, length, max) TALES expression
+
+    Truncates a sentence if it is longer than the specified 'length' characters.
+    Truncated strings will end with an ellipsis character (“…”)
+    See also 'get_text_start'
+    """
+
+    @staticmethod
+    def render(value, length=50, max=0):
+        if not value:
+            return ''
+        return get_text_start(value, length, max=max)
+
+
+@adapter_config(name='raw', context=(str, IRequest), provides=IHTMLRenderer)
+class BaseHTMLRenderer(ContextRequestAdapter):
+    """Raw text HTML renderer
+
+    This renderer renders input text 'as is', mainly for use in a <pre> tag.
+    """
+
+    def render(self, **kwargs):
+        return self.context
+
+
+@adapter_config(name='text', context=(str, IRequest), provides=IHTMLRenderer)
+class TextRenderer(BaseHTMLRenderer):
+    """Basic text HTML renderer
+
+    This renderer only replace newlines with HTML breaks.
+    """
+
+    def render(self, **kwargs):
+        return html.escape(self.context).replace('\n', '<br />\n')
+
+
+@adapter_config(name='js', context=(str, IRequest), provides=IHTMLRenderer)
+class JsRenderer(BaseHTMLRenderer):
+    """Custom Javascript HTML renderer
+
+    This renderer replaces single quotes with escaped ones
+    """
+
+    def render(self, **kwargs):
+        return self.context.replace("'", "\\'")
+
+
+@adapter_config(name='rest', context=(str, IRequest), provides=IHTMLRenderer)
+class ReStructuredTextRenderer(BaseHTMLRenderer):
+    """reStructuredText HTML renderer
+
+    This renderer is using *docutils* to render HTML output.
+    """
+
+    def render(self, **kwargs):
+        """Render reStructuredText to HTML"""
+        overrides = {
+            'halt_level': 6,
+            'input_encoding': 'unicode',
+            'output_encoding': 'unicode',
+            'initial_header_level': 3,
+        }
+        if 'settings' in kwargs:
+            overrides.update(kwargs['settings'])
+        parts = docutils.core.publish_parts(self.context,
+                                            writer_name='html',
+                                            settings_overrides=overrides)
+        return ''.join((parts['body_pre_docinfo'], parts['docinfo'], parts['body']))
+
+
+def text_to_html(text, renderer='text'):
+    """Convert text to HTML using the given renderer
+
+    Renderer name can be any registered HTML renderer adapter
+    """
+    request = check_request()
+    registry = request.registry
+    renderer = registry.queryMultiAdapter((text, request), IHTMLRenderer, name=renderer)
+    if renderer is not None:
+        return renderer.render()
+
+
+empty_marker = object()
+
+
+@adapter_config(name='html', context=(Interface, Interface, Interface), provides=ITALESExtension)
+class HTMLTalesExtension(ContextRequestViewAdapter):
+    """*extension:html* TALES expression
+
+    If first *context* argument of the renderer is an object for which an :py:class:`IHTMLRenderer`
+    adapter can be found, this adapter is used to render the context to HTML; if *context* is a string,
+    it is converted to HTML using the renderer defined as second parameter; otherwise, context is just
+    converted to string using the :py:func:`str` function.
+    """
+
+    def render(self, context=empty_marker, renderer='text'):
+        if context is empty_marker:
+            context = self.context
+        if not context:
+            return ''
+        registry = self.request.registry
+        adapter = registry.queryMultiAdapter((context, self.request, self.view), IHTMLRenderer)
+        if adapter is None:
+            adapter = registry.queryMultiAdapter((context, self.request), IHTMLRenderer)
+        if adapter is not None:
+            return adapter.render()
+        elif isinstance(context, str):
+            return text_to_html(context, renderer)
+        else:
+            return str(context)
+
+
+@vocabulary_config(name='PyAMS HTML renderers')
+class RenderersVocabulary(SimpleVocabulary):
+    """Text renderers vocabulary"""
+
+    def __init__(self):
+        request = check_request()
+        registry = request.registry
+        translate = registry.localizer.translate
+        terms = [SimpleTerm(name, name, translate(adapt.title).label)
+                 for name, adapt in registry.getAdapters(('', request), IHTMLRenderer)]
+        super(RenderersVocabulary, self).__init__(terms)
+
+
+@adapter_config(name='br', context=(Interface, Interface, Interface), provides=ITALESExtension)
+class BrTalesExtension(ContextRequestViewAdapter):
+    """extension:br(value, class) TALES expression
+
+    This expression can be used to context a given character ('|' by default) into HTML
+    breaks with given CSS class.
+    """
+
+    @staticmethod
+    def render(value, css_class='', character='|', start_tag=None, end_tag=None):
+        if not value:
+            return ''
+        br = '<br {0} />'.format('class="{0}"'.format(css_class) if css_class else '')
+        elements = value.split(character)
+        if start_tag:
+            elements[0] = '<{0}>{1}</{0}>'.format(start_tag, elements[0])
+        if end_tag:
+            elements[-1] = '<{0}>{1}</{0}>'.format(end_tag, elements[-1])
+        return br.join(elements)