# -*- coding: utf-8 -*-
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.encoding import force_bytes
[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_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'}')
