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.2 KiB
48 lines
2.2 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 sqlite3
|
|
from collections import namedtuple
|
|
|
|
class Keystore:
|
|
SCHEMA = '''
|
|
PRAGMA synchronous=FULL;
|
|
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.executescript(self.SCHEMA)
|
|
|
|
def __del__(self):
|
|
self._db.close()
|
|
|
|
# fetch all entries with known fingerprints: `load()` or specific entry `load(jid = jid)`
|
|
def load(self, **args):
|
|
sql = "SELECT * FROM keystore WHERE %s" % (not args and "fingerprint IS NOT NULL" or " AND ".join(["{0}='{1}'".format(*arg) for arg in args.items()]))
|
|
return (args) and self._db.execute(sql).fetchone() or self._db.execute(sql).fetchall()
|
|
|
|
# save entry to database: save(jid=jid, fingerprint=fingerprint)
|
|
def save(self, **args):
|
|
sql = "REPLACE INTO keystore({0}) VALUES({1})".format(",".join(args.keys()), ",".join(["'%s'"%s for s in args.values()]))
|
|
self._db.execute(sql)
|
|
|
|
# delete entry from database: `delete(jid=jid) `
|
|
def delete(self, **args):
|
|
sql = "DELETE FROM keystore WHERE %s" % " AND ".join(["{0}='{1}'".format(*a) for a in args.items()])
|
|
self._db.execute(sql)
|