# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2005 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 bsddb import dbshelve, db #------------------------------------------------------------------------- # # GRAMPS modules # #------------------------------------------------------------------------- from RelLib import * from GrampsDbBase import * 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.repository_map = {} self.media_map = {} self.event_map = {} self.metadata = {} self.filename = "" self.id_trans = {} self.pid_trans = {} self.fid_trans = {} self.sid_trans = {} self.rid_trans = {} self.oid_trans = {} self.eventnames = {} self.repository_types = {} self.undodb = [] def load(self,name,callback,mode="w"): self.undolog = "%s.log" % name self.undodb = db.DB() self.undodb.open(self.undolog, db.DB_RECNO, db.DB_CREATE) self.filename = name self.readonly = mode == "r" 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_repository_cursor(self): return GrampsInMemCursor(self.repository_map) def get_media_cursor(self): return GrampsInMemCursor(self.media_map) def close(self): if not self.readonly: self.undodb.close() try: os.remove(self.undolog) except: 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 get_repository_type_list(self): repos_types = self.repository_types.keys() a = {} for repos_type in repos_types: a[unicode(repos_type)] = 1 vals = a.keys() vals.sort() return vals def _del_person(self,handle): del self.id_trans[person.get_gramps_id()] del self.person_map[handle] def _del_source(self,handle): del self.sid_trans[source.get_gramps_id()] del self.source_map[str(handle)] def _del_repository(self,handle): del self.rid_trans[repository.get_gramps_id()] del self.repository_map[str(handle)] def _del_place(self,handle): del self.pid_trans[place.get_gramps_id()] del self.place_map[str(handle)] def _del_media(self,handle): del self.oid_trans[obj.get_gramps_id()] del self.media_map[str(handle)] def _del_family(self,handle): del self.fid_trans[family.get_gramps_id()] del self.family_map[str(handle)] def _del_event(self,handle): del self.event_map[str(handle)] def commit_person(self,person,transaction,change_time=None): if self.readonly or not person.get_handle(): return 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): if self.readonly or not place.get_handle(): return 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): if self.readonly or not family.get_handle(): return 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): if self.readonly or not obj.get_handle(): return 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): if self.readonly or not source.get_handle(): return gid = source.get_gramps_id() self.sid_trans[gid] = source.get_handle() GrampsDbBase.commit_source(self,source,transaction,change_time) def commit_repository(self,repository,transaction,change_time=None): if self.readonly or not repository.get_handle(): return gid = repository.get_gramps_id() self.rid_trans[gid] = repository.get_handle() GrampsDbBase.commit_repository(self,repository,transaction,change_time) def get_person_from_gramps_id(self,val): handle = self.id_trans.get(str(val)) if handle: data = self.person_map[handle] if data: person = Person() person.unserialize(data) return person return None def get_family_from_gramps_id(self,val): handle = self.fid_trans.get(str(val)) if handle: data = self.family_map[handle] if data: family = Family() family.unserialize(data) return family return None def get_place_from_gramps_id(self,val): handle = self.pid_trans.get(str(val)) if handle: data = self.place_map[handle] if data: place = Place() place.unserialize(data) return place return None def get_source_from_gramps_id(self,val): handle = self.sid_trans.get(str(val)) if handle: data = self.source_map[handle] if data: source = Source() source.unserialize(data) return source return None def get_repository_from_gramps_id(self,val): handle = self.rid_trans.get(str(val)) if handle: data = self.repository_map[handle] if data: repository = Repository() repository.unserialize(data) return repository return None def get_object_from_gramps_id(self,val): handle = self.oid_trans.get(str(val)) if handle: data = self.media_map[handle] if data: obj = MediaObject() obj.unserialize(data) return obj return None