18 TaggedObject = collections.namedtuple(
26 def _make_tag_only_serializer(tag, expected_value):
30 assert to.instance == expected_value
35 def _make_struct_serializer(fmt):
37 packer = functools.partial(struct.pack, fmt)
40 return packer(to.tag, to.instance)
44 _TAGS_TO_SERIALIZERS = {
45 TAG_NULL: _make_tag_only_serializer(TAG_NULL, None),
46 TAG_TRUE: _make_tag_only_serializer(TAG_TRUE, True),
47 TAG_FALSE: _make_tag_only_serializer(TAG_FALSE, False),
48 TAG_UINT8: _make_struct_serializer('B'),
49 TAG_UINT16: _make_struct_serializer('H'),
50 TAG_UINT32: _make_struct_serializer('I'),
51 TAG_UINT64: _make_struct_serializer('Q'),
52 TAG_INT8: _make_struct_serializer('b'),
53 TAG_INT16: _make_struct_serializer('h'),
54 TAG_INT32: _make_struct_serializer('i'),
55 TAG_INT64: _make_struct_serializer('q'),
59 return _TAGS_TO_SERIALIZERS[to.tag](to)
61 def _make_tag_only_parser(tag, value):
63 return TaggedObject(tag = tag, instance = value)
67 def _make_struct_deserializer(tag, fmt):
69 size = struct.calcsize(fmt)
70 unpacker = functools.partial(struct.unpack, fmt)
75 return TaggedObject(tag = tag, instance = unpacker(b)[0])
80 TAG_NULL: _make_tag_only_parser(TAG_NULL, None),
81 TAG_TRUE: _make_tag_only_parser(TAG_TRUE, True),
82 TAG_FALSE: _make_tag_only_parser(TAG_FALSE, False),
83 TAG_UINT8: _make_struct_deserializer(TAG_UINT8, 'B'),
84 TAG_UINT16: _make_struct_deserializer(TAG_UINT16, 'H'),
85 TAG_UINT32: _make_struct_deserializer(TAG_UINT32, 'I'),
86 TAG_UINT64: _make_struct_deserializer(TAG_UINT64, 'Q'),
87 TAG_INT8: _make_struct_deserializer(TAG_INT8, 'b'),
88 TAG_INT16: _make_struct_deserializer(TAG_INT16, 'h'),
89 TAG_INT32: _make_struct_deserializer(TAG_INT32, 'i'),
90 TAG_INT64: _make_struct_deserializer(TAG_INT64, 'q'),
94 if isinstance(b, bytes):
99 result = _TAGS_TO_PARSERS[tag](b)
103 if len(remainder) == 0:
106 raise Exception('Unable to parse remainder: {}'.format(remainder))