summaryrefslogtreecommitdiff
path: root/src/leap/mx/util
diff options
context:
space:
mode:
authorIsis Lovecruft <isis@torproject.org>2013-02-15 19:36:43 +0000
committerIsis Lovecruft <isis@torproject.org>2013-02-15 19:36:43 +0000
commitb32d659b85901a8e04c6c921928483fda3b3e6e0 (patch)
treeeee7c9ca9abdeafc3a401d99a9ffe24a997aea44 /src/leap/mx/util
parent2fc0d3ffaae3681177e023cd7a8a77b5ead8a403 (diff)
Add the storage utility for parsing the config file structure in a more
readable fashion.
Diffstat (limited to 'src/leap/mx/util')
-rw-r--r--src/leap/mx/util/storage.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/src/leap/mx/util/storage.py b/src/leap/mx/util/storage.py
new file mode 100644
index 0000000..c4c797a
--- /dev/null
+++ b/src/leap/mx/util/storage.py
@@ -0,0 +1,42 @@
+
+class Storage(dict):
+ """
+ A Storage object is like a dictionary except `obj.foo` can be used
+ in addition to `obj['foo']`.
+
+ >>> o = Storage(a=1)
+ >>> o.a
+ 1
+ >>> o['a']
+ 1
+ >>> o.a = 2
+ >>> o['a']
+ 2
+ >>> del o.a
+ >>> o.a
+ None
+ """
+ def __getattr__(self, key):
+ try:
+ return self[key]
+ except KeyError, k:
+ return None
+
+ def __setattr__(self, key, value):
+ self[key] = value
+
+ def __delattr__(self, key):
+ try:
+ del self[key]
+ except KeyError, k:
+ raise AttributeError, k
+
+ def __repr__(self):
+ return '<Storage ' + dict.__repr__(self) + '>'
+
+ def __getstate__(self):
+ return dict(self)
+
+ def __setstate__(self, value):
+ for (k, v) in value.items():
+ self[k] = v