Changeset 6730 for TI12-security/trunk/NDGSecurity/python/ndg_security_common/ndg/security/common/utils/__init__.py

Timestamp:

16/03/10 08:37:55 (11 years ago)

Author:

pjkersha

Message:

• Working Credential Wallet refactored for Python 2.6
• Pruned out more old code: ZSI and Twisted SAOP/WSDL wrappers, Session Manager and WS-Security, Test 'Site B' Attribute Authority - for testing role mapping.
• Started XACML package ElementTree based parser.

File:

• TI12-security/trunk/NDGSecurity/python/ndg_security_common/ndg/security/common/utils/__init__.py (modified) (1 diff)

TI12-security/trunk/NDGSecurity/python/ndg_security_common/ndg/security/common/utils/__init__.py

@@ -76 +76 @@
     
         return super(TypedList, self).append(item)
+
+class RestrictedKeyNamesDict(dict):
+    """Utility class for holding a constrained list of key names
+    """
+    
+    def __init__(self, *arg, **kw):
+        """Alter standard dict() initialisation to fix key names set at 
+        initialisation
+        """
+        super(RestrictedKeyNamesDict, self).__init__(*arg, **kw)
+        self.__keyNames = self.keys() 
+          
+    def __setitem__(self, key, val):
+        if key not in self.__keyNames:
+            raise KeyError('Key name %r not recognised.  Valid key names '
+                           'are: %r' % (key, self.__keyNames))
+            
+        dict.__setitem__(self, key, val)
+
+    def update(self, d, **kw):        
+        for dictArg in (d, kw):
+            for k in dictArg:
+                if k not in self.__keyNames:
+                    raise KeyError('Key name "%s" not recognised.  Valid '
+                                   'key names are: %s' % 
+                                   self.__keyNames)
+        
+        dict.update(self, d, **kw)