sapamja · August 14, 2014 23:55 · Aug 14, 2014
diff --git a/storage_class.py b/storage_class.py
@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+"""
+This file contains Storage class to storage data.
+"""
+
+class Storage(dict):
+     """
+     A Storage object is like a dictionary except `obj.foo` can be used
+     in addition to `obj['foo']`.
+
+     >>> obj = Storage(a=1)
+     >>> obj.val
+     1
+     >>> obj['val']
+     1
+     >>> obj.val = 2
+     >>> obj['val']
+     2
+     >>> del obj.val
+     >>> obj.val
+     None
+     """
+     def __getattr__(self, key):
+         try:
+             return self[key]
+         except KeyError:
+             return None
+
+     def __setattr__(self, key, value):
+         self[key] = value
+
+     def __delattr__(self, key):
+         try:
+             del self[key]
+         except KeyError as error:
+             raise AttributeError(error)
+
+     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
No results found