Add methods to handle internal data for context

Author: c00kiemon5ter

src/satosa/context.py

@@ -1,6 +1,3 @@
-"""
-Holds methods for sending internal data through the satosa proxy
-"""
 from .exception import SATOSAError
 
 
@@ -11,6 +8,9 @@ class SATOSABadContextError(SATOSAError):
     pass
 
 
+"""
+Holds methods for sending internal data through the satosa proxy
+"""
 class Context(object):
     """
     Holds information about the current request.
@@ -60,3 +60,19 @@ def path(self, p):
         elif p.startswith('/'):
             raise ValueError("path can't start with '/'")
         self._path = p
+
+    def decorate(self, key, value):
+        """
+        Add information to the context
+        """
+
+        self.internal_data[key] = value
+        return self
+
+    def get_decoration(self, key):
+        """
+        Retrieve information from the context
+        """
+
+        value = self.internal_data.get(key)
+        return value