You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
2.1 KiB
48 lines
2.1 KiB
# Copyright (C) 2019 Philipp Hörist <philipp AT hoerist.com>
|
|
# Copyright (C) 2019 Pavel R. <pd at narayana dot im>
|
|
|
|
# This file is part of Gajim OTR Plugin.
|
|
#
|
|
# Gajim OTR Plugin is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published
|
|
# by the Free Software Foundation; version 3 only.
|
|
#
|
|
# This software is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You can always obtain full license text at <http://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
import sqlite3
|
|
from collections import namedtuple
|
|
|
|
class Keystore:
|
|
|
|
__TABLE_LAYOUT__ = '''
|
|
CREATE TABLE IF NOT EXISTS keystore (jid TEXT, privatekey TEXT, fingerprint TEXT, trust INTEGER, timestamp INTEGER, comment TEXT, UNIQUE(privatekey)); CREATE UNIQUE INDEX IF NOT EXISTS jid_fingerprint ON keystore (jid, fingerprint);
|
|
'''
|
|
|
|
def __init__(self, db):
|
|
self._db = sqlite3.connect(db, isolation_level=None)
|
|
self._db.row_factory = lambda cur,row : namedtuple("Row", [col[0] for col in cur.description])(*row)
|
|
self._db.execute("PRAGMA synchronous=FULL;")
|
|
self._db.executescript(self.__TABLE_LAYOUT__)
|
|
|
|
def load(self, item = {'fingerprint IS NOT NULL; --': None}):
|
|
sql = "SELECT * FROM keystore WHERE " + " AND ".join(["%s = '%s'" % (str(key), str(value)) for key,value in item.items()])
|
|
if next(iter(item.values())): return self._db.execute(sql).fetchone() # return fetchone() if `item` arg is set
|
|
return self._db.execute(sql).fetchall() or () # else return fetchall() or empty iterator
|
|
|
|
def save(self, item):
|
|
sql = "REPLACE INTO keystore(%s) VALUES(%s)" % (",".join(item.keys()), ",".join(["'%s'" % x for x in item.values()]) )
|
|
return self._db.execute(sql)
|
|
|
|
def forgot(self, item):
|
|
sql = "DELETE FROM keystore WHERE " + " AND ".join(["%s='%s'" % (str(key),str(value)) for key,value in item.items()])
|
|
return self._db.execute(sql)
|
|
|
|
def close(self):
|
|
self._db.close()
|