diff options
Diffstat (limited to 'services/datastore/datastore.py')
-rw-r--r-- | services/datastore/datastore.py | 333 |
1 files changed, 333 insertions, 0 deletions
diff --git a/services/datastore/datastore.py b/services/datastore/datastore.py new file mode 100644 index 0000000..92270e4 --- /dev/null +++ b/services/datastore/datastore.py @@ -0,0 +1,333 @@ +#!/bin/python +# Copyright (C) 2006, Red Hat, Inc. +# +# This program 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; either version 2 of the License, or +# (at your option) any later version. +# +# This program 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 should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + +import os +import dbus, dbus.glib, gobject +import logging +import sqlite +import dbus_helpers + +have_sugar = False +try: + from sugar import env + have_sugar = True +except ImportError: + pass + + +_DS_SERVICE = "org.laptop.sugar.DataStore" +_DS_DBUS_INTERFACE = "org.laptop.sugar.DataStore" +_DS_OBJECT_PATH = "/org/laptop/sugar/DataStore" + +_DS_OBJECT_DBUS_INTERFACE = "org.laptop.sugar.DataStore.Object" +_DS_OBJECT_OBJECT_PATH = "/org/laptop/sugar/DataStore/Object" + +class NotFoundError(Exception): + pass + +def _create_op(uid): + return "%s/%d" % (_DS_OBJECT_OBJECT_PATH, uid) + +def _get_uid_from_op(op): + if not op.startswith(_DS_OBJECT_OBJECT_PATH + "/"): + raise ValueError("Invalid object path %s." % op) + item = op[len(_DS_OBJECT_OBJECT_PATH + "/"):] + return int(item) + +def _get_data_as_string(data): + if isinstance(data, list): + data_str = "" + for item in data: + data_str += chr(item) + return data_str + elif isinstance(data, int): + return str(data) + elif isinstance(data, str): + return data + elif isinstance(data, unicode): + return str(data) + else: + raise ValueError("Unsupported data type") + +class DataStoreDBusHelper(dbus.service.Object): + def __init__(self, parent, bus_name): + self._parent = parent + self._bus_name = bus_name + dbus.service.Object.__init__(self, bus_name, _DS_OBJECT_PATH) + + @dbus.service.method(_DS_DBUS_INTERFACE, + in_signature="x", out_signature="o") + def get(self, uid): + uid = self._parent.get(uid) + return self._create_op(uid) + + @dbus.service.method(_DS_DBUS_INTERFACE, + in_signature="aya{sv}", out_signature="o") + def create(self, data, prop_dict): + uid = self._parent.create(data, prop_dict) + return _create_op(uid) + + @dbus.service.method(_DS_DBUS_INTERFACE, + in_signature="o", out_signature="i") + def delete(self, op): + uid = _get_uid_from_op(op) + self._parent.delete(uid) + return 0 + + @dbus.service.method(_DS_DBUS_INTERFACE, + in_signature="a{sv}", out_signature="ao") + def find(self, prop_dict): + uids = self._parent.find(prop_dict) + ops = [] + for uid in uids: + ops.append(_create_op(uid)) + return ops + +class ObjectDBusHelper(dbus_helpers.FallbackObject): + def __init__(self, parent, bus_name): + self._parent = parent + self._bus_name = bus_name + dbus_helpers.FallbackObject.__init__(self, bus_name, _DS_OBJECT_OBJECT_PATH) + + @dbus_helpers.method(_DS_OBJECT_DBUS_INTERFACE, + in_signature="", out_signature="ay", dbus_message_keyword="dbus_message") + def get_data(self, dbus_message=None): + if not dbus_message: + raise RuntimeError("Need the dbus message.") + uid = _get_uid_from_op(dbus_message.get_path()) + return self._parent.get_data(uid) + + @dbus_helpers.method(_DS_OBJECT_DBUS_INTERFACE, + in_signature="ay", out_signature="i", dbus_message_keyword="dbus_message") + def set_data(self, data, dbus_message=None): + if not dbus_message: + raise RuntimeError("Need the dbus message.") + uid = _get_uid_from_op(dbus_message.get_path()) + self._parent.set_data(uid, data) + return 0 + + @dbus_helpers.method(_DS_OBJECT_DBUS_INTERFACE, + in_signature="as", out_signature="a{sv}", dbus_message_keyword="dbus_message") + def get_properties(self, keys, dbus_message=None): + if not dbus_message: + raise RuntimeError("Need the dbus message.") + uid = _get_uid_from_op(dbus_message.get_path()) + return self._parent.get_properties(uid, keys) + + @dbus_helpers.method(_DS_OBJECT_DBUS_INTERFACE, + in_signature="a{sv}", out_signature="i", dbus_message_keyword="dbus_message") + def set_properties(self, prop_dict, dbus_message=None): + if not dbus_message: + raise RuntimeError("Need the dbus message.") + uid = _get_uid_from_op(dbus_message.get_path()) + self._parent.set_properties(uid, prop_dict) + return 0 + + @dbus_helpers.fallback_signal(_DS_OBJECT_DBUS_INTERFACE, + signature="ba{sv}b", ignore_args=["uid"]) + def Updated(self, data, prop_dict, deleted, uid=None): + # Return the object path so the signal decorator knows what + # object this signal should be fore + if not uid: + raise RuntimeError("Need a UID.") + op = _create_op(uid) + return op + +class DataStore(object): + def __init__(self): + self._session_bus = dbus.SessionBus() + self._bus_name = dbus.service.BusName(_DS_SERVICE, bus=self._session_bus) + self._dbus_helper = DataStoreDBusHelper(self, self._bus_name) + self._dbus_obj_helper = ObjectDBusHelper(self, self._bus_name) + + ppath = "/tmp" + if have_sugar: + ppath = env.get_profile_path() + self._dbfile = os.path.join(ppath, "ds", "data-store.db") + if not os.path.exists(os.path.dirname(self._dbfile)): + os.makedirs(os.path.dirname(self._dbfile), 0755) + + self._dbcx = sqlite.connect(self._dbfile, encoding="utf-8", timeout=3) + try: + self._ensure_table() + except StandardError, e: + logging.info("Could not access the data store. Reason: '%s'. Exiting..." % e) + os._exit(1) + + def __del__(self): + self._dbcx.close() + del self._dbcx + + def _ensure_table(self): + curs = self._dbcx.cursor() + try: + curs.execute('SELECT * FROM properties LIMIT 4') + self._dbcx.commit() + except Exception, e: + # If table wasn't created, try to create it + self._dbcx.commit() + curs.execute('CREATE TABLE objects (' \ + 'uid INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, ' \ + 'data BLOB' \ + ');') + curs.execute('CREATE TABLE properties (' \ + 'objid INTEGER NOT NULL, ' \ + 'key VARCHAR(100),' \ + 'value VARCHAR(200)' \ + ');') + curs.execute('CREATE INDEX objid_idx ON properties(objid);') + self._dbcx.commit() + del curs + + def get(self, uid): + curs = self._dbcx.cursor() + curs.execute('SELECT uid FROM objects WHERE uid=%d;' % uid) + res = curs.fetchall() + self._dbcx.commit() + del curs + if len(res) > 0: + return uid + raise NotFoundError("Object %d was not found." % uid) + + def create(self, data, prop_dict=None): + curs = self._dbcx.cursor() + curs.execute("INSERT INTO objects (uid) VALUES (NULL);") + curs.execute("SELECT last_insert_rowid();") + rows = curs.fetchall() + self._dbcx.commit() + last_row = rows[0] + uid = last_row[0] + for (key, value) in prop_dict.items(): + safe_key = key.replace("'", "''") + value = str(value) + curs.execute("INSERT INTO properties (objid, key, value) VALUES (%d, '%s', '%s');" % (uid, safe_key, sqlite.encode(value))) + self._dbcx.commit() + del curs + return uid + + def delete(self, uid): + curs = self._dbcx.cursor() + curs.execute("DELETE FROM objects WHERE (uid=%d);" % uid) + curs.execute("DELETE FROM properties WHERE (objid=%d);" % uid) + self._dbcx.commit() + del curs + self._dbus_obj_helper.Updated(False, {}, True, uid=uid) + return 0 + + def find(self, prop_dict): + query = "SELECT objid FROM properties WHERE (" + subquery = "" + for (key, value) in prop_dict.items(): + safe_key = key.replace("'", "''") + value = str(value) + substr = "key='%s' AND value='%s'" % (safe_key, sqlite.encode(value)) + if len(subquery) > 0: + subquery += " AND" + subquery += substr + query += subquery + ")" + curs = self._dbcx.cursor() + curs.execute(query) + rows = curs.fetchall() + self._dbcx.commit() + # FIXME: ensure that each properties.objid has a match in objects.uid + uids = [] + for row in rows: + uids.append(row['objid']) + del curs + return uids + + def set_data(self, uid, data): + curs = self._dbcx.cursor() + curs.execute('SELECT uid FROM objects WHERE uid=%d;' % uid) + res = curs.fetchall() + self._dbcx.commit() + if len(res) <= 0: + del curs + raise NotFoundError("Object %d was not found." % uid) + data = _get_data_as_string(data) + curs.execute("UPDATE objects SET data='%s' WHERE uid=%d;" % (sqlite.encode(data), uid)) + self._dbcx.commit() + del curs + self._dbus_obj_helper.Updated(True, {}, False, uid=uid) + + _reserved_keys = ["uid", "objid", "data", "created", "modified"] + def set_properties(self, uid, prop_dict): + curs = self._dbcx.cursor() + curs.execute('SELECT uid FROM objects WHERE uid=%d;' % uid) + res = curs.fetchall() + self._dbcx.commit() + if len(res) <= 0: + del curs + raise NotFoundError("Object %d was not found." % uid) + + for key in prop_dict.keys(): + if key in self._reserved_keys: + raise ValueError("key %s is a reserved key." % key) + + for (key, value) in prop_dict.items(): + safe_key = key.replace("'", "''") + enc_value = sqlite.encode(_get_data_as_string(value)) + curs.execute("SELECT objid FROM properties WHERE (objid=%d AND key='%s');" % (uid, safe_key)) + if len(curs.fetchall()) > 0: + curs.execute("UPDATE properties SET value='%s' WHERE (objid=%d AND key='%s');" % (enc_value, uid, safe_key)) + else: + curs.execute("INSERT INTO properties (objid, key, value) VALUES (%d, '%s', '%s');" % (uid, safe_key, enc_value)) + self._dbcx.commit() + del curs + self._dbus_obj_helper.Updated(False, {}, False, uid=uid) + + def get_data(self, uid): + curs = self._dbcx.cursor() + curs.execute('SELECT uid, data FROM objects WHERE uid=%d;' % uid) + res = curs.fetchall() + self._dbcx.commit() + del curs + if len(res) <= 0: + raise NotFoundError("Object %d was not found." % uid) + return sqlite.decode(res[0]['data']) + + def get_properties(self, uid, keys): + query = "SELECT objid, key, value FROM properties WHERE (objid=%d AND (" % uid + subquery = "" + for key in keys: + if len(subquery) > 0: + subquery += " OR " + safe_key = key.replace("'", "''") + subquery += "key='%s'" % safe_key + query += subquery + "));" + curs = self._dbcx.cursor() + curs.execute(query) + rows = curs.fetchall() + self._dbcx.commit() + prop_dict = {} + for row in rows: + conv_key = row['key'].replace("''", "'") + prop_dict[conv_key] = row['value'] + del curs + return prop_dict + + +def main(): + loop = gobject.MainLoop() + ds = DataStore() + try: + loop.run() + except KeyboardInterrupt: + print 'Ctrl+C pressed, exiting...' + +if __name__ == "__main__": + main() |