298 lines
9.3 KiB
Python
298 lines
9.3 KiB
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 logging
|
|
|
|
import dbus
|
|
import dbus.glib
|
|
import gobject
|
|
|
|
from sugar import util
|
|
|
|
DS_DBUS_SERVICE = "org.laptop.sugar.DataStore"
|
|
DS_DBUS_INTERFACE = "org.laptop.sugar.DataStore"
|
|
DS_DBUS_PATH = "/org/laptop/sugar/DataStore"
|
|
|
|
_bus = dbus.SessionBus()
|
|
_data_store = dbus.Interface(_bus.get_object(DS_DBUS_SERVICE, DS_DBUS_PATH),
|
|
DS_DBUS_INTERFACE)
|
|
|
|
class DataStoreObject:
|
|
def __init__(self, metadata, file_path=None, handle=None):
|
|
self._metadata = metadata
|
|
self._file_path = file_path
|
|
self._handle = handle
|
|
|
|
def get_metadata(self):
|
|
return self._metadata
|
|
|
|
def get_file_path(self):
|
|
return self._file_path
|
|
|
|
def set_file_path(self, file_path):
|
|
self._file_path = file_path
|
|
|
|
def get_handle(self):
|
|
return self._handle
|
|
|
|
def get_object_type(self):
|
|
raise NotImplementedError()
|
|
|
|
class Text(DataStoreObject):
|
|
def get_object_type(self):
|
|
return 'text'
|
|
|
|
class Picture(DataStoreObject):
|
|
def get_object_type(self):
|
|
return 'picture'
|
|
|
|
class Link(DataStoreObject):
|
|
def get_object_type(self):
|
|
return 'link'
|
|
|
|
def _read_from_object_path(object_path):
|
|
dbus_object = _bus.get_object(DS_DBUS_SERVICE, object_path)
|
|
metadata = dbus_object.get_properties(dbus.Dictionary({}, signature='sv'))
|
|
|
|
object_type = metadata['object-type']
|
|
file_path = metadata['file-path']
|
|
handle = metadata['handle']
|
|
|
|
del metadata['object-type']
|
|
del metadata['file-path']
|
|
del metadata['handle']
|
|
|
|
if object_type == 'text':
|
|
return Text(metadata, file_path, handle)
|
|
elif object_type == 'picture':
|
|
return Picture(metadata, file_path, handle)
|
|
elif object_type == 'link':
|
|
return Link(metadata, file_path, handle)
|
|
else:
|
|
raise NotImplementedError('Unknown object type.')
|
|
|
|
def read(handle):
|
|
object_path = _data_store.get(handle)
|
|
return _read_from_object_path(object_path)
|
|
|
|
def write(obj):
|
|
metadata = obj.get_metadata().copy()
|
|
metadata['file-path'] = obj.get_file_path()
|
|
metadata['object-type'] = obj.get_object_type()
|
|
|
|
if obj.get_handle():
|
|
_data_store.update(int(obj.get_handle()), dbus.Dictionary(metadata))
|
|
return obj.get_handle()
|
|
else:
|
|
object_path = _data_store.create(dbus.Dictionary(metadata))
|
|
dbus_object = _bus.get_object(DS_DBUS_SERVICE, object_path)
|
|
return dbus_object.get_properties(['handle'])['handle']
|
|
|
|
def find(query):
|
|
object_paths = _data_store.find(query)
|
|
objects = []
|
|
for object_path in object_paths:
|
|
objects.append(_read_from_object_path(object_path))
|
|
return objects
|
|
|
|
def delete(handle):
|
|
pass
|
|
|
|
################################################################################
|
|
|
|
class ObjectCache(object):
|
|
def __init__(self):
|
|
self._cache = {}
|
|
|
|
def get(self, object_path):
|
|
try:
|
|
return self._cache[object_path]
|
|
except KeyError:
|
|
return None
|
|
|
|
def add(self, obj):
|
|
op = obj.object_path()
|
|
if not self._cache.has_key(op):
|
|
self._cache[op] = obj
|
|
|
|
def remove(self, object_path):
|
|
try:
|
|
del self._cache[object_path]
|
|
except IndexError:
|
|
pass
|
|
|
|
|
|
|
|
class DSObject(gobject.GObject):
|
|
|
|
__gsignals__ = {
|
|
'updated': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
|
|
([gobject.TYPE_PYOBJECT, gobject.TYPE_PYOBJECT, gobject.TYPE_PYOBJECT]))
|
|
}
|
|
|
|
_DS_OBJECT_DBUS_INTERFACE = "org.laptop.sugar.DataStore.Object"
|
|
|
|
def __init__(self, bus, new_obj_cb, del_obj_cb, object_path):
|
|
gobject.GObject.__init__(self)
|
|
self._object_path = object_path
|
|
self._ps_new_object = new_obj_cb
|
|
self._ps_del_object = del_obj_cb
|
|
bobj = bus.get_object(DS_DBUS_SERVICE, object_path)
|
|
self._dsobj = dbus.Interface(bobj, self._DS_OBJECT_DBUS_INTERFACE)
|
|
self._dsobj.connect_to_signal('Updated', self._updated_cb)
|
|
self._data = None
|
|
self._data_needs_update = True
|
|
self._properties = None
|
|
self._deleted = False
|
|
|
|
def object_path(self):
|
|
return self._object_path
|
|
|
|
def uid(self):
|
|
if not self._properties:
|
|
self._properties = self._dsobj.get_properties([])
|
|
return self._properties['uid']
|
|
|
|
def _emit_updated_signal(self, data, prop_dict, deleted):
|
|
self.emit('updated', data, prop_dict, deleted)
|
|
return False
|
|
|
|
def _update_internal_properties(self, prop_dict):
|
|
did_update = False
|
|
for (key, value) in prop_dict.items():
|
|
if not len(value):
|
|
if self._properties.has_key(ley):
|
|
did_update = True
|
|
del self._properties[key]
|
|
else:
|
|
if self._properties.has_key(key):
|
|
if self._properties[key] != value:
|
|
did_update = True
|
|
self._properties[key] = value
|
|
else:
|
|
did_update = True
|
|
self._properties[key] = value
|
|
return did_update
|
|
|
|
def _updated_cb(self, data=False, prop_dict={}, deleted=False):
|
|
if self._update_internal_properties(prop_dict):
|
|
gobject.idle_add(self._emit_updated_signal, data, prop_dict, deleted)
|
|
self._deleted = deleted
|
|
|
|
def get_data(self):
|
|
if self._data_needs_update:
|
|
data = self._dsobj.get_data()
|
|
self._data = ""
|
|
for c in data:
|
|
self._data += chr(c)
|
|
return self._data
|
|
|
|
def set_data(self, data):
|
|
old_data = self._data
|
|
self._data = data
|
|
try:
|
|
self._dsobj.set_data(dbus.ByteArray(data))
|
|
del old_data
|
|
except dbus.DBusException, e:
|
|
self._data = old_data
|
|
raise e
|
|
|
|
def set_properties(self, prop_dict):
|
|
old_props = self._properties
|
|
self._update_internal_properties(prop_dict)
|
|
try:
|
|
self._dsobj.set_properties(prop_dict)
|
|
del old_props
|
|
except dbus.DBusException, e:
|
|
self._properties = old_props
|
|
raise e
|
|
|
|
def get_properties(self, prop_list=[]):
|
|
if not self._properties:
|
|
self._properties = self._dsobj.get_properties(prop_list)
|
|
return self._properties
|
|
|
|
class DataStore(gobject.GObject):
|
|
|
|
_DS_DBUS_OBJECT_PATH = DS_DBUS_PATH + "/Object/"
|
|
|
|
def __init__(self):
|
|
gobject.GObject.__init__(self)
|
|
self._objcache = ObjectCache()
|
|
self._bus = dbus.SessionBus()
|
|
self._ds = dbus.Interface(self._bus.get_object(DS_DBUS_SERVICE,
|
|
DS_DBUS_PATH), DS_DBUS_INTERFACE)
|
|
|
|
def _new_object(self, object_path):
|
|
obj = self._objcache.get(object_path)
|
|
if obj:
|
|
return obj
|
|
|
|
if object_path.startswith(self._DS_DBUS_OBJECT_PATH):
|
|
obj = DSObject(self._bus, self._new_object,
|
|
self._del_object, object_path)
|
|
else:
|
|
raise RuntimeError("Unknown object type")
|
|
self._objcache.add(obj)
|
|
return obj
|
|
|
|
def _del_object(self, object_path):
|
|
# FIXME
|
|
pass
|
|
|
|
def get(self, uid=None, activity_id=None):
|
|
if not activity_id and not uid:
|
|
raise ValueError("At least one of activity_id or uid must be specified")
|
|
if activity_id and uid:
|
|
raise ValueError("Only one of activity_id or uid can be specified")
|
|
if activity_id:
|
|
if not util.validate_activity_id(activity_id):
|
|
raise ValueError("activity_id must be valid")
|
|
return self._new_object(self._ds.getActivityObject(activity_id))
|
|
elif uid:
|
|
if not len(uid):
|
|
raise ValueError("uid must be valid")
|
|
return self._new_object(self._ds.get(int(uid)))
|
|
raise RuntimeError("At least one of activity_id or uid must be specified")
|
|
|
|
def create(self, data, prop_dict={}, activity_id=None):
|
|
if activity_id and not util.validate_activity_id(activity_id):
|
|
raise ValueError("activity_id must be valid")
|
|
if not activity_id:
|
|
activity_id = ""
|
|
op = self._ds.create(dbus.ByteArray(data), dbus.Dictionary(prop_dict), activity_id)
|
|
return self._new_object(op)
|
|
|
|
def delete(self, obj):
|
|
op = obj.object_path()
|
|
obj = self._objcache.get(op)
|
|
if not obj:
|
|
raise RuntimeError("Object not found.")
|
|
self._ds.delete(op)
|
|
|
|
def find(self, prop_dict):
|
|
ops = self._ds.find(dbus.Dictionary(prop_dict))
|
|
objs = []
|
|
for op in ops:
|
|
objs.append(self._new_object(op))
|
|
return objs
|
|
|
|
_ds = None
|
|
def get_instance():
|
|
global _ds
|
|
if not _ds:
|
|
_ds = DataStore()
|
|
return _ds
|