from __future__ import absolute_import, print_function, unicode_literals
from itertools import chain
from wolframclient.serializers.base import FormatSerializer
from wolframclient.serializers.utils import py_encode_decimal, py_encode_text
from wolframclient.utils import six
from wolframclient.utils.api import base64
from wolframclient.utils.encoding import force_bytes, force_text
[docs]def yield_with_separators(iterable, separator=b", ", first=None, last=None):
if first:
yield first
for i, arg in enumerate(iterable):
if i:
yield separator
for sub in arg:
yield sub
if last:
yield last
[docs]class WLSerializer(FormatSerializer):
def __init__(self, normalizer=None, indent=None, **opts):
super(WLSerializer, self).__init__(normalizer=normalizer, **opts)
self.indent = indent
[docs] def generate_bytes(self, data):
return self.encode(data)
[docs] def serialize_function(self, head, args, **opts):
return chain(head, yield_with_separators(args, first=b"[", last=b"]"))
[docs] def serialize_symbol(self, name):
yield force_bytes(name)
[docs] def serialize_string(self, string):
return py_encode_text(string)
[docs] def serialize_bytes(self, bytes, as_byte_array=not six.PY2):
# by default we are serializing as_byte_array for PY3,
# py2 is by default using strings
if as_byte_array:
return self.serialize_function(
self.serialize_symbol(b"ByteArray"), ((b'"', base64.b64encode(bytes), b'"'),)
)
else:
return self.serialize_string(force_text(bytes, "iso-8859-1"))
[docs] def serialize_decimal(self, number):
yield py_encode_decimal(number)
[docs] def serialize_float(self, number):
yield (b"%.13f" % number).rstrip(b"0")
[docs] def serialize_int(self, number):
yield b"%i" % number
[docs] def serialize_rule(self, lhs, rhs):
return yield_with_separators((lhs, rhs), separator=b" -> ")
[docs] def serialize_rule_delayed(self, lhs, rhs):
return yield_with_separators((lhs, rhs), separator=b" :> ")
[docs] def serialize_mapping(self, mapping, **opts):
return yield_with_separators(
(self.serialize_rule(key, value) for key, value in mapping),
first=b"<|",
last=b"|>",
)
[docs] def serialize_association(self, mapping, **opts):
return self.serialize_mapping(mapping, **opts)
[docs] def serialize_iterable(self, iterable, **opts):
return yield_with_separators(iterable, first=b"{", last=b"}")