import os
import sqlite3
import cherrypy
+from random import randint
+import sys
class Store(object):
def get_user_preferences(self, user):
return self._load_user_prefs(self._user_dbname, user)
+ def save_user_preferences(self, user, options):
+ SELECT = "SELECT option, value FROM users WHERE name=?"
+ UPDATE = "UPDATE users SET value=? WHERE name=? AND option=?"
+ INSERT = "INSERT INTO users VALUES(?,?,?)"
+ con = None
+ try:
+ con = sqlite3.connect(self._user_dbname)
+ cur = con.cursor()
+ cur.execute("""
+ CREATE TABLE IF NOT EXISTS users(name TEXT,
+ option TEXT,
+ value TEXT)
+ """)
+ curvals = dict()
+ for row in cur.execute(SELECT, (user,)):
+ curvals[row[0]] = row[1]
+
+ for name in options:
+ if name in curvals:
+ cur.execute(UPDATE, (options[name], user, name))
+ else:
+ cur.execute(INSERT, (user, name, options[name]))
+
+ con.commit()
+ except sqlite3.Error, e:
+ if con:
+ con.rollback()
+ cherrypy.log.error("Failed to store config: [%s]" % e)
+ raise
+ finally:
+ if con:
+ con.close()
+
def get_plugins_config(self, facility):
con = None
rows = []
if con:
con.close()
+ def new_datum(self, plugin, datum):
+ ID = "(SELECT IFNULL(MAX(id), 0) + 1 FROM %s_data)" % plugin
+ INSERT_NEW = "INSERT INTO %s_data VALUES(%s,?,?)" % (plugin, ID)
+ INSERT = "INSERT INTO %s_data VALUES(?,?,?)" % plugin
+ SELECT = "SELECT id FROM %s_data WHERE name=? AND value=?" % plugin
+ DELETE = "DELETE FROM %s_data WHERE name=? AND value=?" % plugin
+ con = None
+ try:
+ con = sqlite3.connect(self._admin_dbname)
+ cur = con.cursor()
+ tmpid = ('new', str(randint(0, sys.maxint)))
+ cur.execute(INSERT_NEW, tmpid)
+ cur.execute(SELECT, tmpid)
+ rows = cur.fetchall()
+ idval = rows[0][0]
+ for name in datum:
+ cur.execute(INSERT, (idval, name, datum[name]))
+ cur.execute(DELETE, tmpid)
+ con.commit()
+ except sqlite3.Error, e:
+ if con:
+ con.rollback()
+ cherrypy.log.error("Failed to store %s data: [%s]" % (plugin, e))
+ raise
+ finally:
+ if con:
+ con.close()
+
+ def del_datum(self, plugin, idval):
+ DELETE = "DELETE FROM %s_data WHERE id=?" % plugin
+ con = None
+ try:
+ con = sqlite3.connect(self._admin_dbname)
+ cur = con.cursor()
+ cur.execute(DELETE, (idval,))
+ con.commit()
+ except sqlite3.Error, e:
+ if con:
+ con.rollback()
+ cherrypy.log.error("Failed to delete %s data: [%s]" % (plugin, e))
+ raise
+ finally:
+ if con:
+ con.close()
+
def wipe_data(self, plugin):
# Try to backup old data first, just in case
try: