File size: 2,533 Bytes
593db5a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
"""
An implementation of MutableAttr.
"""
try:
from collections.abc import Mapping
except ImportError:
from collections import Mapping
import six
from attrdict.mixins import MutableAttr
__all__ = ['AttrMap']
class AttrMap(MutableAttr):
"""
An implementation of MutableAttr.
"""
def __init__(self, items=None, sequence_type=tuple):
if items is None:
items = {}
elif not isinstance(items, Mapping):
items = dict(items)
self._setattr('_sequence_type', sequence_type)
self._setattr('_mapping', items)
self._setattr('_allow_invalid_attributes', False)
def _configuration(self):
"""
The configuration for an attrmap instance.
"""
return self._sequence_type
def __getitem__(self, key):
"""
Access a value associated with a key.
"""
return self._mapping[key]
def __setitem__(self, key, value):
"""
Add a key-value pair to the instance.
"""
self._mapping[key] = value
def __delitem__(self, key):
"""
Delete a key-value pair
"""
del self._mapping[key]
def __len__(self):
"""
Check the length of the mapping.
"""
return len(self._mapping)
def __iter__(self):
"""
Iterated through the keys.
"""
return iter(self._mapping)
def __repr__(self):
"""
Return a string representation of the object.
"""
# sequence type seems like more trouble than it is worth.
# If people want full serialization, they can pickle, and in
# 99% of cases, sequence_type won't change anyway
return six.u("AttrMap({mapping})").format(mapping=repr(self._mapping))
def __getstate__(self):
"""
Serialize the object.
"""
return (
self._mapping,
self._sequence_type,
self._allow_invalid_attributes
)
def __setstate__(self, state):
"""
Deserialize the object.
"""
mapping, sequence_type, allow_invalid_attributes = state
self._setattr('_mapping', mapping)
self._setattr('_sequence_type', sequence_type)
self._setattr('_allow_invalid_attributes', allow_invalid_attributes)
@classmethod
def _constructor(cls, mapping, configuration):
"""
A standardized constructor.
"""
return cls(mapping, sequence_type=configuration)
|