diff options
Diffstat (limited to 'paste/util/quoting.py')
-rw-r--r-- | paste/util/quoting.py | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/paste/util/quoting.py b/paste/util/quoting.py new file mode 100644 index 0000000..df0d9da --- /dev/null +++ b/paste/util/quoting.py @@ -0,0 +1,85 @@ +# (c) 2005 Ian Bicking and contributors; written for Paste (http://pythonpaste.org) +# Licensed under the MIT license: http://www.opensource.org/licenses/mit-license.php + +import cgi +import six +import re +from six.moves import html_entities +from six.moves.urllib.parse import quote, unquote + + +__all__ = ['html_quote', 'html_unquote', 'url_quote', 'url_unquote', + 'strip_html'] + +default_encoding = 'UTF-8' + +def html_quote(v, encoding=None): + r""" + Quote the value (turned to a string) as HTML. This quotes <, >, + and quotes: + """ + encoding = encoding or default_encoding + if v is None: + return '' + elif isinstance(v, six.binary_type): + return cgi.escape(v, 1) + elif isinstance(v, six.text_type): + if six.PY3: + return cgi.escape(v, 1) + else: + return cgi.escape(v.encode(encoding), 1) + else: + if six.PY3: + return cgi.escape(six.text_type(v), 1) + else: + return cgi.escape(six.text_type(v).encode(encoding), 1) + +_unquote_re = re.compile(r'&([a-zA-Z]+);') +def _entity_subber(match, name2c=html_entities.name2codepoint): + code = name2c.get(match.group(1)) + if code: + return six.unichr(code) + else: + return match.group(0) + +def html_unquote(s, encoding=None): + r""" + Decode the value. + + """ + if isinstance(s, six.binary_type): + s = s.decode(encoding or default_encoding) + return _unquote_re.sub(_entity_subber, s) + +def strip_html(s): + # should this use html_unquote? + s = re.sub('<.*?>', '', s) + s = html_unquote(s) + return s + +def no_quote(s): + """ + Quoting that doesn't do anything + """ + return s + +_comment_quote_re = re.compile(r'\-\s*\>') +# Everything but \r, \n, \t: +_bad_chars_re = re.compile('[\x00-\x08\x0b-\x0c\x0e-\x1f]') +def comment_quote(s): + """ + Quote that makes sure text can't escape a comment + """ + comment = str(s) + #comment = _bad_chars_re.sub('', comment) + #print('in ', repr(str(s))) + #print('out', repr(comment)) + comment = _comment_quote_re.sub('->', comment) + return comment + +url_quote = quote +url_unquote = unquote + +if __name__ == '__main__': + import doctest + doctest.testmod() |