pep/slixmpp-omemo
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

xep_0384: store sessions to file

Browse source 
Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2018-12-01 18:22:34 +00:00
parent 23ac4a5bff
commit 05ef6d75c3
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
storage.py
View file

@ -67,12 +67,24 @@ class SyncFileStorage(omemo.Storage):
json.dump(self.__state, f) json.dump(self.__state, f)
def loadSession(self, _callback, bare_jid: str, device_id: int): def loadSession(self, _callback, bare_jid: str, device_id: int):
if not self.__sessions:
try:
filepath = os.path.join(self.storage_dir, 'sessions.json')
with open(filepath, 'r') as f:
self.__sessions = json.load(f)
except OSError:
return None
return self.__sessions.get(bare_jid, {}).get(device_id, None) return self.__sessions.get(bare_jid, {}).get(device_id, None)
def storeSession(self, callback, bare_jid: str, device_id: int, session) -> None: def storeSession(self, callback, bare_jid: str, device_id: int, session) -> None:
self.__sessions[bare_jid] = self.__sessions.get(bare_jid, {}) self.__sessions[bare_jid] = self.__sessions.get(bare_jid, {})
self.__sessions[bare_jid][device_id] = session self.__sessions[bare_jid][device_id] = session
filepath = os.path.join(self.storage_dir, 'sessions.json')
with open(filepath, 'w') as f:
json.dump(self.__sessions, f)
def loadActiveDevices(self, _callback, bare_jid: str) -> Union[None, List[int]]: def loadActiveDevices(self, _callback, bare_jid: str) -> Union[None, List[int]]:
if not self.__devices: if not self.__devices:
try: try: