1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
|
#
# This file is part of pysnmp software.
#
# Copyright (c) 2005-2019, Ilya Etingof <etingof@gmail.com>
# License: https://www.pysnmp.com/pysnmp/license.html
#
from pysnmp import nextid
from pysnmp.proto import error
class Cache:
__stateReference = nextid.Integer(0xffffff)
def __init__(self):
self.__cacheEntries = {}
def push(self, **securityData):
stateReference = self.__stateReference()
self.__cacheEntries[stateReference] = securityData
return stateReference
def pop(self, stateReference):
if stateReference in self.__cacheEntries:
securityData = self.__cacheEntries[stateReference]
else:
raise error.ProtocolError(
f'Cache miss for stateReference={stateReference} at {self}'
)
del self.__cacheEntries[stateReference]
return securityData
|