Source code for wolframclient.utils.encoding
from __future__ import absolute_import, print_function, unicode_literals
from wolframclient.utils import six
from wolframclient.utils.functional import map
[docs]def force_text(s, encoding="utf-8", errors="strict"):
"""
Similar to smart_text, except that lazy instances are resolved to
strings, rather than kept as lazy objects.
If strings_only is True, don't convert (some) non-string-like objects.
"""
# Handle the common case first for performance reasons.
if isinstance(s, six.text_type):
return s
if not isinstance(s, six.string_types):
if six.PY3:
if isinstance(s, bytes):
s = six.text_type(s, encoding, errors)
else:
s = six.text_type(s)
elif hasattr(s, "__unicode__"):
s = six.text_type(s)
else:
s = six.text_type(bytes(s), encoding, errors)
else:
# Note: We use .decode() here, instead of six.text_type(s, encoding,
# errors), so that if s is a SafeBytes, it ends up being a
# SafeText at the end.
s = s.decode(encoding, errors)
return s
[docs]def force_bytes(s, encoding="utf-8", errors="strict"):
"""
If strings_only is True, don't convert (some) non-string-like objects.
"""
# Handle the common case first for performance reasons.
if isinstance(s, bytes):
return s
if isinstance(s, six.buffer_types):
return bytes(s)
if not isinstance(s, six.string_types):
try:
if six.PY3:
return six.text_type(s).encode(encoding)
else:
return bytes(s)
except UnicodeEncodeError:
if isinstance(s, Exception):
# An Exception subclass containing non-ASCII data that doesn't
# know how to print itself properly. We shouldn't raise a
# further exception.
return b" ".join(force_bytes(arg, encoding, errors=errors) for arg in s)
return six.text_type(s).encode(encoding, errors)
else:
return s.encode(encoding, errors)
[docs]def safe_force_text(obj):
try:
return force_text(obj, errors="ignore")
except Exception as e:
return "<unprintable obj: %s>" % e
# this function is supposed to be the most efficient byte concatenation that can be archived in python
# used by the serializers
# join seems to be the winner
# https://gist.github.com/smcl/7462529818bb77baad32727a9e5ff44c
# https://blog.mclemon.io/python-efficient-string-concatenation-in-python-2016-edition
if six.PY2:
def concatenate_bytes(iterable):
return b"".join(map(six.binary_type, iterable))
else:
concatenate_bytes = b"".join