def to_compact_dict(self, compact_keys=True, compact_values=True):
dict_ = self.serialize_to_dict()
if compact_values:
dict_ = self.to_compact_values(dict_)
if compact_keys:
dict_ = replace_keys(dict_, COMPACT_KEY_MAP)
return dict_
def to_compact_dict(self, compact_keys=True, compact_values=True):
dict_ = self.serialize_to_dict()
if compact_values:
dict_ = map_values(dict_, COMPACT_VALUE_MAP)
dict_ = map_values(dict_, COMPACT_SUBKEY_MAP, subkeys=True)
if compact_keys:
dict_ = replace_keys(dict_, COMPACT_KEY_MAP)
return dict_
def from_compact_dict(cls,
compact_dict,
compact_keys=True,
compact_values=True):
dict_ = compact_dict
if compact_keys:
dict_ = replace_keys(dict_, UNCOMPACT_KEY_MAP)
if compact_values:
dict_ = cls.from_compact_values(dict_)
return cls.deserialize_from_dict(dict_)
def from_compact_dict(cls,
compact_dict,
compact_keys=True,
compact_values=True):
dict_ = compact_dict
if compact_keys:
dict_ = replace_keys(dict_, UNCOMPACT_KEY_MAP)
if compact_values:
dict_ = map_values(dict_, UNCOMPACT_VALUE_MAP)
dict_ = map_values(dict_, UNCOMPACT_SUBKEY_MAP, subkeys=True)
return cls.deserialize_from_dict(dict_)
def test_replace_keys():
replace_map = {
'long_a': 'a',
'long_b': 'b',
'long_c': 'c',
'long_d': 'd',
'long_e': 'e',
'long_f': 'f',
}
assert replace_keys({'long_a': 1}, replace_map) == {'a': 1}
assert replace_keys([{'long_a': 1}], replace_map) == [{'a': 1}]
assert replace_keys({'long_a': 1, 'long_b': 2}, replace_map) == {'a': 1, 'b': 2}
assert replace_keys({
'long_a': 1,
'long_b': 2,
'long_c': [1, 2, 3]
}, replace_map) == {
'a': 1,
'b': 2,
'c': [1, 2, 3]
}
assert replace_keys({
'long_a': {
'long_c': 1
},
'long_b': 2,
'long_c': [1, 2, 3]
}, replace_map) == {
'a': {
'c': 1
},
'b': 2,
'c': [1, 2, 3]
}
assert replace_keys({
'long_a': {
'long_c': 1
},
'long_b': 2,
'long_c': [{
'long_d': 4
}, 2, 3]
}, replace_map) == {
'a': {
'c': 1
},
'b': 2,
'c': [{
'd': 4
}, 2, 3]
}
assert replace_keys({
'long_a': {
'long_c': [1, 5, 6]
},
'long_b': 2,
'long_c': [{
'long_d': [1, 2, 3]
}, 2, 3]
}, replace_map) == {
'a': {
'c': [1, 5, 6]
},
'b': 2,
'c': [{
'd': [1, 2, 3]
}, 2, 3]
}