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
31
32
|
# -*- coding: utf-8 -*-
from ..state.signedprekeystore import SignedPreKeyStore
from ..state.signedprekeyrecord import SignedPreKeyRecord
from ..invalidkeyidexception import InvalidKeyIdException
class InMemorySignedPreKeyStore(SignedPreKeyStore):
def __init__(self):
self.store = {}
def loadSignedPreKey(self, signedPreKeyId):
if signedPreKeyId not in self.store:
raise InvalidKeyIdException("No such signedprekeyrecord! %s " % signedPreKeyId)
return SignedPreKeyRecord(serialized=self.store[signedPreKeyId])
def loadSignedPreKeys(self):
results = []
for serialized in self.store.values():
results.append(SignedPreKeyRecord(serialized=serialized))
return results
def storeSignedPreKey(self, signedPreKeyId, signedPreKeyRecord):
self.store[signedPreKeyId] = signedPreKeyRecord.serialize()
def containsSignedPreKey(self, signedPreKeyId):
return signedPreKeyId in self.store
def removeSignedPreKey(self, signedPreKeyId):
del self.store[signedPreKeyId]
|
