[omemo] Refactor storeSession()
This commit is contained in:
@@ -296,13 +296,19 @@ class LiteAxolotlStore(AxolotlStore):
|
|||||||
return self._con.execute(query).fetchall()
|
return self._con.execute(query).fetchall()
|
||||||
|
|
||||||
def storeSession(self, recipientId, deviceId, sessionRecord):
|
def storeSession(self, recipientId, deviceId, sessionRecord):
|
||||||
self.deleteSession(recipientId, deviceId)
|
|
||||||
|
|
||||||
query = '''INSERT INTO sessions(recipient_id, device_id, record)
|
query = '''INSERT INTO sessions(recipient_id, device_id, record)
|
||||||
VALUES(?,?,?)'''
|
VALUES(?,?,?)'''
|
||||||
|
try:
|
||||||
self._con.execute(query, (recipientId,
|
self._con.execute(query, (recipientId,
|
||||||
deviceId,
|
deviceId,
|
||||||
sessionRecord.serialize()))
|
sessionRecord.serialize()))
|
||||||
|
except sqlite3.IntegrityError:
|
||||||
|
query = '''UPDATE sessions SET record = ?
|
||||||
|
WHERE recipient_id = ? AND device_id = ?'''
|
||||||
|
self._con.execute(query, (sessionRecord.serialize(),
|
||||||
|
recipientId,
|
||||||
|
deviceId))
|
||||||
|
|
||||||
self._con.commit()
|
self._con.commit()
|
||||||
|
|
||||||
def containsSession(self, recipientId, deviceId):
|
def containsSession(self, recipientId, deviceId):
|
||||||
|
|||||||
Reference in New Issue
Block a user