# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2004 Donald N. Allingham # # 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # # $Id$ """ Provides the common infrastructure for database formats that must hold all of their data in memory. """ from RelLib import * from GrampsDbBase import * import os import md5 import gtk class GrampsInMemCursor(GrampsCursor): """ Cursor class for in-memory database classes. Since the in-memory classes use python dictionaries, the python iter class is used to provide the cursor function. """ def __init__(self,src_map): self.src_map = src_map self.current = iter(src_map) def first(self): self.current = iter(self.src_map) return self.next() def next(self): try: index = self.current.next() return (index,self.src_map[index]) except StopIteration: return None def close(self): pass #------------------------------------------------------------------------- # # GrampsInMemDB # #------------------------------------------------------------------------- class GrampsInMemDB(GrampsDbBase): """GRAMPS database object. This object is a base class for other objects.""" def __init__(self): """creates a new GrampsDB""" GrampsDbBase.__init__(self) self.person_map = {} self.name_group = {} self.family_map = {} self.place_map = {} self.source_map = {} self.media_map = {} self.event_map = {} self.metadata = {} self.filename = "" self.id_trans = {} self.pid_trans = {} self.fid_trans = {} self.sid_trans = {} self.oid_trans = {} self.eventnames = {} self.undodb = [] def load(self,name,callback): pass def get_person_cursor(self): return GrampsInMemCursor(self.person_map) def get_family_cursor(self): return GrampsInMemCursor(self.family_map) def get_place_cursor(self): return GrampsInMemCursor(self.place_map) def get_source_cursor(self): return GrampsInMemCursor(self.source_map) def get_media_cursor(self): return GrampsInMemCursor(self.media_map) def close(self): pass def abort_changes(self): pass def set_name_group_mapping(self,name,group): if group == None and self.name_group.has_key(name): del self.name_group[name] else: self.name_group[name] = group def get_surname_list(self): a = {} for person_id in iter(self.person_map): p = self.get_person_from_handle(person_id) a[p.get_primary_name().get_group_as()] = 1 vals = a.keys() vals.sort() return vals def get_person_event_type_list(self): names = self.eventnames.keys() a = {} for name in names: a[unicode(name)] = 1 vals = a.keys() vals.sort() return vals def remove_person(self,handle,transaction): person = self.get_person_from_handle(handle) self.genderStats.uncount_person (person) if transaction != None: old_data = self.person_map.get(handle) transaction.add(PERSON_KEY,handle,old_data) del self.id_trans[self.person_map[handle].get_gramps_id()] del self.person_map[handle] def remove_source(self,handle,transaction): if transaction != None: old_data = self.source_map.get(str(handle)) transaction.add(SOURCE_KEY,handle,old_data) del self.sid_trans[self.source_map[handle].get_gramps_id()] del self.source_map[str(handle)] def remove_family(self,handle,transaction): if transaction != None: old_data = self.family_map.get(str(handle)) transaction.add(FAMILY_KEY,handle,old_data) del self.fid_trans[self.family_map[handle].get_gramps_id()] del self.family_map[str(handle)] def remove_event(self,handle,transaction): if transaction != None: old_data = self.event_map.get(str(handle)) transaction.add(EVENT_KEY,handle,old_data) del self.event_map[str(handle)] def commit_person(self,person,transaction,change_time=None): gid = person.get_gramps_id() self.id_trans[gid] = person.get_handle() GrampsDbBase.commit_person(self,person,transaction,change_time) def commit_place(self,place,transaction,change_time=None): gid = place.get_gramps_id() self.pid_trans[gid] = place.get_handle() GrampsDbBase.commit_place(self,place,transaction,change_time) def commit_family(self,family,transaction,change_time=None): gid = family.get_gramps_id() self.fid_trans[gid] = family.get_handle() GrampsDbBase.commit_family(self,family,transaction,change_time) def commit_media_object(self,obj,transaction,change_time=None): gid = obj.get_gramps_id() self.oid_trans[gid] = obj.get_handle() GrampsDbBase.commit_media_object(self,obj,transaction,change_time) def commit_source(self,source,transaction,change_time=None): gid = source.get_gramps_id() self.sid_trans[gid] = source.get_handle() GrampsDbBase.commit_source(self,source,transaction,change_time) def get_person_from_gramps_id(self,val): handle = self.id_trans.get(str(val)) if handle: return self.person_map[handle] else: return None def get_family_from_gramps_id(self,val): handle = self.fid_trans.get(str(val)) if handle: return self.family_map[handle] else: return None def get_place_from_gramps_id(self,val): handle = self.pid_trans.get(str(val)) if handle: return self.place_map[handle] else: return None def get_source_from_gramps_id(self,val): handle = self.sid_trans.get(str(val)) if handle: return self.source_map[handle] else: return None def get_object_from_gramps_id(self,val): handle = self.oid_trans.get(str(val)) if handle: return self.media_map[handle] else: return None def _build_thumb_path(self,path): base = os.path.expanduser('~/.gramps/thumb') m = md5.md5(path) return os.path.join(base,m.hexdigest()+'.jpg') def get_thumbnail_image(self,handle): data = self.media_map.get(handle) if data: filename = self._build_thumb_path(data[2]) if not os.path.isfile(filename): self.set_thumbnail_image(handle,filename) return gtk.gdk.pixbuf_new_from_file(filename) else: return None def set_thumbnail_image(self,handle,path): try: pixbuf = gtk.gdk.pixbuf_new_from_file(path) w = pixbuf.get_width() h = pixbuf.get_height() scale = const.thumbScale / (float(max(w,h))) pw = int(w*scale) ph = int(h*scale) pixbuf = pixbuf.scale_simple(pw,ph,gtk.gdk.INTERP_BILINEAR) pixbuf.save(self._build_thumb_path(path),"jpeg") except: print "Could not create thumbnail for",path