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
|
# -*- coding: utf-8 -*-
from ..state.prekeystore import PreKeyStore
from ..state.prekeyrecord import PreKeyRecord
from ..invalidkeyidexception import InvalidKeyIdException
class InMemoryPreKeyStore(PreKeyStore):
def __init__(self):
self.store = {}
def loadPreKey(self, preKeyId):
if preKeyId not in self.store:
raise InvalidKeyIdException("No such prekeyRecord!")
return PreKeyRecord(serialized=self.store[preKeyId])
def storePreKey(self, preKeyId, preKeyRecord):
self.store[preKeyId] = preKeyRecord.serialize()
def containsPreKey(self, preKeyId):
return preKeyId in self.store
def removePreKey(self, preKeyId):
if preKeyId in self.store:
del self.store[preKeyId]
|
