Added constant for struct encoding

This commit is contained in:
Philip Trauner 2016-10-12 16:27:11 +02:00
parent 55f9dd4425
commit d2571f6d50

View file

@ -29,10 +29,8 @@ INVALID_METADATA_LAYOUT = 2
INVALID_DATA_TYPE = 3 INVALID_DATA_TYPE = 3
META_ROUTE = "meta" META_ROUTE = "meta"
COMPRESSION_LEVEL = 3 COMPRESSION_LEVEL = 3
PACK_FORMAT = "Bh"
class ConvertFailedError(ValueError): class ConvertFailedError(ValueError):
@ -76,7 +74,7 @@ class Meta:
def create_metadata(data_type, converted_route, indexed_dict=False): def create_metadata(data_type, converted_route, indexed_dict=False):
return struct.pack("bh", return struct.pack(PACK_FORMAT,
DATA_TYPES[data_type] if not indexed_dict else DATA_TYPES[INDEXED_DICT], DATA_TYPES[data_type] if not indexed_dict else DATA_TYPES[INDEXED_DICT],
converted_route) converted_route)
@ -108,7 +106,7 @@ def pack_message(data, exchange_route, compression_level,
def parse_metadata(message): def parse_metadata(message):
metadata = struct.unpack("bh", message[:4]) metadata = struct.unpack(PACK_FORMAT, message[:4])
return Metadata(metadata[0], metadata[1]) return Metadata(metadata[0], metadata[1])