initial revision

1 files changed, 31 insertions, 0 deletions

diff --git a/pysnmp/entity/rfc3413/context.py b/pysnmp/entity/rfc3413/context.py
new file mode 100644
index 0000000..a73227f
--- /dev/null
+++ b/pysnmp/entity/rfc3413/context.py
@@ -0,0 +1,31 @@
+# SNMP entity context
+from pysnmp import error
+
+class SnmpContext:
+    def __init__(self, snmpEngine, contextEngineId=None):
+        self.contextEngineId = contextEngineId
+        self.contextNames = {
+            '': snmpEngine.msgAndPduDsp.mibInstrumController # Default name
+            } 
+
+    def registerContextName(self, contextName, mibInstrum=None):
+        if self.contextNames.has_key(contextName):
+            raise error.PySnmpError(
+                'Duplicate contextName %s' % contextName
+                )
+        if mibInstrum is None:
+            self.contextName[contextName] = self.contextName['']
+        else:
+            self.contextName[contextName] = mibInstrum
+            
+    def unregisterContextName(self, contextName):
+        if self.contextNames.has_key(contextName):
+            del self.contextName[contextName]
+
+    def getMibInstrum(self, contextName):
+        if not self.contextNames.has_key(contextName):
+            raise error.PySnmpError(
+                'Missing contextName %s' % contextName
+                )
+        else:
+            return self.contextNames[contextName]