1146 lines
39 KiB
Python
1146 lines
39 KiB
Python
|
|
# 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$
|
|
|
|
"""
|
|
Base class for the GRAMPS databases. All database interfaces should inherit
|
|
from this class.
|
|
"""
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# libraries
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from RelLib import *
|
|
import cPickle
|
|
import time
|
|
import locale
|
|
from gettext import gettext as _
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# constants
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
_UNDO_SIZE = 1000
|
|
_id_reg = compile("%\d+d")
|
|
|
|
PERSON_KEY = 0
|
|
FAMILY_KEY = 1
|
|
SOURCE_KEY = 2
|
|
EVENT_KEY = 3
|
|
MEDIA_KEY = 4
|
|
PLACE_KEY = 5
|
|
|
|
class GrampsDbBase:
|
|
"""
|
|
GRAMPS database object. This object is a base class for all
|
|
database interfaces.
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""
|
|
Creates a new GrampsDbBase instance. A new GrampDbBase class should
|
|
never be directly created. Only classes derived from this class should
|
|
be created.
|
|
"""
|
|
|
|
self.smap_index = 0
|
|
self.emap_index = 0
|
|
self.pmap_index = 0
|
|
self.fmap_index = 0
|
|
self.lmap_index = 0
|
|
self.omap_index = 0
|
|
|
|
self.set_person_id_prefix(GrampsCfg.get_person_id_prefix())
|
|
self.set_object_id_prefix(GrampsCfg.get_object_id_prefix())
|
|
self.set_family_id_prefix(GrampsCfg.get_family_id_prefix())
|
|
self.set_source_id_prefix(GrampsCfg.get_source_id_prefix())
|
|
self.set_place_id_prefix(GrampsCfg.get_place_id_prefix())
|
|
self.set_event_id_prefix(GrampsCfg.get_event_id_prefix())
|
|
|
|
self.open = 0
|
|
self.genderStats = GenderStats()
|
|
|
|
self.id_trans = None
|
|
self.fid_trans = None
|
|
self.pid_trans = None
|
|
self.sid_trans = None
|
|
self.oid_trans = None
|
|
self.env = None
|
|
self.person_map = None
|
|
self.family_map = None
|
|
self.place_map = None
|
|
self.source_map = None
|
|
self.media_map = None
|
|
self.event_map = None
|
|
self.eventnames = None
|
|
self.metadata = None
|
|
self.undo_callback = None
|
|
self.redo_callback = None
|
|
self.modified = 0
|
|
|
|
self.undoindex = -1
|
|
self.translist = [None] * _UNDO_SIZE
|
|
self.default = None
|
|
self.owner = Researcher()
|
|
self.bookmarks = []
|
|
self.path = ""
|
|
self.place2title = {}
|
|
|
|
def load(self,name,callback):
|
|
"""
|
|
Opens the specified database. The method needs to be overridden
|
|
in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def close(self):
|
|
"""
|
|
Closes the specified database. The method needs to be overridden
|
|
in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def is_open(self):
|
|
"""
|
|
Returns 1 if the database has been opened.
|
|
"""
|
|
return self.person_map != None
|
|
|
|
def commit_person(self,person,transaction,change_time=None):
|
|
"""
|
|
Commits the specified Person to the database, storing the changes
|
|
as part of the transaction.
|
|
"""
|
|
|
|
if change_time:
|
|
person.change = int(change_time)
|
|
else:
|
|
person.change = int(time.time())
|
|
handle = str(person.get_handle())
|
|
if transaction != None:
|
|
old_data = self.person_map.get(handle)
|
|
transaction.add(PERSON_KEY,handle,old_data)
|
|
self.person_map[handle] = person.serialize()
|
|
|
|
def commit_media_object(self,obj,transaction,change_time=None):
|
|
"""
|
|
Commits the specified MediaObject to the database, storing the changes
|
|
as part of the transaction.
|
|
"""
|
|
if change_time:
|
|
obj.change = int(change_time)
|
|
else:
|
|
obj.change = int(time.time())
|
|
handle = str(obj.get_handle())
|
|
if transaction != None:
|
|
old_data = self.media_map.get(handle)
|
|
transaction.add(MEDIA_KEY,handle,old_data)
|
|
self.media_map[handle] = obj.serialize()
|
|
|
|
def commit_source(self,source,transaction,change_time=None):
|
|
"""
|
|
Commits the specified Source to the database, storing the changes
|
|
as part of the transaction.
|
|
"""
|
|
if change_time:
|
|
source.change = int(change_time)
|
|
else:
|
|
source.change = int(time.time())
|
|
handle = str(source.get_handle())
|
|
if transaction != None:
|
|
old_data = self.source_map.get(handle)
|
|
transaction.add(SOURCE_KEY,handle,old_data)
|
|
self.source_map[handle] = source.serialize()
|
|
|
|
def commit_place(self,place,transaction,change_time=None):
|
|
"""
|
|
Commits the specified Place to the database, storing the changes
|
|
as part of the transaction.
|
|
"""
|
|
if change_time:
|
|
place.change = int(change_time)
|
|
else:
|
|
place.change = int(time.time())
|
|
handle = str(place.get_handle())
|
|
if transaction != None:
|
|
old_data = self.place_map.get(handle)
|
|
transaction.add(PLACE_KEY,handle,old_data)
|
|
self.place_map[handle] = place.serialize()
|
|
|
|
def commit_event(self,event,transaction,change_time=None):
|
|
"""
|
|
Commits the specified Event to the database, storing the changes
|
|
as part of the transaction.
|
|
"""
|
|
if change_time:
|
|
event.change = int(change_time)
|
|
else:
|
|
event.change = int(time.time())
|
|
handle = str(event.get_handle())
|
|
if transaction != None:
|
|
old_data = self.event_map.get(handle)
|
|
transaction.add(EVENT_KEY,handle,old_data)
|
|
self.event_map[handle] = event.serialize()
|
|
|
|
def commit_family(self,family,transaction,change_time=None):
|
|
"""
|
|
Commits the specified Family to the database, storing the changes
|
|
as part of the transaction.
|
|
"""
|
|
if change_time:
|
|
family.change = int(change_time)
|
|
else:
|
|
family.change = int(time.time())
|
|
handle = str(family.get_handle())
|
|
if transaction != None:
|
|
old_data = self.family_map.get(handle)
|
|
transaction.add(FAMILY_KEY,handle,old_data)
|
|
self.family_map[handle] = family.serialize()
|
|
|
|
def find_next_person_gramps_id(self):
|
|
"""
|
|
Returns the next available GRAMPS' ID for a Person object based
|
|
off the person ID prefix.
|
|
"""
|
|
index = self.iprefix % self.pmap_index
|
|
while self.id_trans.get(str(index)):
|
|
self.pmap_index += 1
|
|
index = self.iprefix % self.pmap_index
|
|
self.pmap_index += 1
|
|
return index
|
|
|
|
def find_next_place_gramps_id(self):
|
|
"""
|
|
Returns the next available GRAMPS' ID for a Place object based
|
|
off the person ID prefix.
|
|
"""
|
|
index = self.pprefix % self.lmap_index
|
|
while self.pid_trans.get(str(index)):
|
|
self.lmap_index += 1
|
|
index = self.pprefix % self.lmap_index
|
|
self.lmap_index += 1
|
|
return index
|
|
|
|
def find_next_event_gramps_id(self):
|
|
"""
|
|
Returns the next available GRAMPS' ID for a Event object based
|
|
off the person ID prefix.
|
|
"""
|
|
index = self.eprefix % self.emap_index
|
|
self.emap_index += 1
|
|
return index
|
|
|
|
def find_next_object_gramps_id(self):
|
|
"""
|
|
Returns the next available GRAMPS' ID for a MediaObject object based
|
|
off the person ID prefix.
|
|
"""
|
|
index = self.oprefix % self.omap_index
|
|
while self.oid_trans.get(str(index)):
|
|
self.omap_index += 1
|
|
index = self.oprefix % self.omap_index
|
|
self.omap_index += 1
|
|
return index
|
|
|
|
def find_next_source_gramps_id(self):
|
|
"""
|
|
Returns the next available GRAMPS' ID for a Source object based
|
|
off the person ID prefix.
|
|
"""
|
|
index = self.sprefix % self.smap_index
|
|
while self.sid_trans.get(str(index)):
|
|
self.smap_index += 1
|
|
index = self.sprefix % self.smap_index
|
|
self.smap_index += 1
|
|
return index
|
|
|
|
def find_next_family_gramps_id(self):
|
|
"""
|
|
Returns the next available GRAMPS' ID for a Family object based
|
|
off the person ID prefix.
|
|
"""
|
|
index = self.fprefix % self.fmap_index
|
|
while self.fid_trans.get(str(index)):
|
|
self.fmap_index += 1
|
|
index = self.fprefix % self.fmap_index
|
|
self.fmap_index += 1
|
|
return index
|
|
|
|
def get_person_from_handle(self,val):
|
|
"""finds a Person in the database from the passed gramps' ID.
|
|
If no such Person exists, None is returned."""
|
|
|
|
data = self.person_map.get(str(val))
|
|
if data:
|
|
person = Person()
|
|
person.unserialize(data)
|
|
return person
|
|
return None
|
|
|
|
def get_source_from_handle(self,val):
|
|
"""finds a Source in the database from the passed gramps' ID.
|
|
If no such Source exists, None is returned."""
|
|
|
|
data = self.source_map.get(str(val))
|
|
if data:
|
|
source = Source()
|
|
source.unserialize(data)
|
|
return source
|
|
return None
|
|
|
|
def get_object_from_handle(self,handle):
|
|
"""finds an Object in the database from the passed gramps' ID.
|
|
If no such Object exists, None is returned."""
|
|
data = self.media_map.get(str(handle))
|
|
if data:
|
|
mobject = MediaObject()
|
|
mobject.unserialize(data)
|
|
return mobject
|
|
return None
|
|
|
|
def get_place_from_handle(self,handle):
|
|
"""finds a Place in the database from the passed gramps' ID.
|
|
If no such Place exists, None is returned."""
|
|
data = self.place_map.get(str(handle))
|
|
if data:
|
|
place = Place()
|
|
place.unserialize(data)
|
|
return place
|
|
return None
|
|
|
|
def get_event_from_handle(self,handle):
|
|
"""finds a Event in the database from the passed gramps' ID.
|
|
If no such Event exists, None is returned."""
|
|
|
|
data = self.event_map.get(str(handle))
|
|
if data:
|
|
event = Event()
|
|
event.unserialize(data)
|
|
return event
|
|
return None
|
|
|
|
def get_family_from_handle(self,handle):
|
|
"""finds a Family in the database from the passed gramps' ID.
|
|
If no such Family exists, None is returned."""
|
|
|
|
data = self.family_map.get(str(handle))
|
|
if data:
|
|
family = Family()
|
|
family.unserialize(data)
|
|
return family
|
|
return None
|
|
|
|
def find_person_from_handle(self,val,transaction):
|
|
"""
|
|
Finds a Person in the database from the passed GRAMPS ID.
|
|
If no such Person exists, a new Person is added to the database.
|
|
"""
|
|
person = Person()
|
|
data = self.person_map.get(str(val))
|
|
if data:
|
|
person.unserialize(data)
|
|
else:
|
|
person.set_handle(val)
|
|
if transaction != None:
|
|
transaction.add(PERSON_KEY, val, None)
|
|
self.person_map[str(val)] = person.serialize()
|
|
self.genderStats.count_person (person, self)
|
|
return person
|
|
|
|
def find_source_from_handle(self,val,transaction):
|
|
"""
|
|
Finds a Source in the database from the passed GRAMPS ID.
|
|
If no such Source exists, a new Source is added to the database.
|
|
"""
|
|
source = Source()
|
|
if self.source_map.get(str(val)):
|
|
source.unserialize(self.source_map.get(str(val)))
|
|
else:
|
|
source.set_handle(val)
|
|
self.add_source(source,transaction)
|
|
return source
|
|
|
|
def find_event_from_handle(self,val,transaction):
|
|
"""
|
|
Finds a Event in the database from the passed GRAMPS ID.
|
|
If no such Event exists, a new Event is added to the database.
|
|
"""
|
|
event = Event()
|
|
data = self.event_map.get(str(val))
|
|
if data:
|
|
event.unserialize(data)
|
|
else:
|
|
event.set_handle(val)
|
|
self.add_event(event,transaction)
|
|
return event
|
|
|
|
def find_object_from_handle(self,handle,transaction):
|
|
"""
|
|
Finds an MediaObject in the database from the passed GRAMPS ID.
|
|
If no such MediaObject exists, a new Object is added to the database."""
|
|
|
|
obj = MediaObject()
|
|
if self.media_map.get(str(handle)):
|
|
obj.unserialize(self.media_map.get(str(handle)))
|
|
else:
|
|
obj.set_handle(handle)
|
|
self.add_object(obj,transaction)
|
|
return obj
|
|
|
|
def find_place_from_handle(self,handle,transaction):
|
|
"""
|
|
Finds a Place in the database from the passed GRAMPS ID.
|
|
If no such Place exists, a new Place is added to the database.
|
|
"""
|
|
place = Place()
|
|
if self.place_map.get(str(handle)):
|
|
place.unserialize(self.place_map.get(str(handle)))
|
|
else:
|
|
place.set_handle(handle)
|
|
self.add_place(place,transaction)
|
|
return place
|
|
|
|
def find_family_from_handle(self,val,transaction):
|
|
"""finds a Family in the database from the passed gramps' ID.
|
|
If no such Family exists, a new Family is added to the database."""
|
|
|
|
family = Family()
|
|
if self.family_map.get(str(val)):
|
|
family.unserialize(self.family_map.get(str(val)))
|
|
else:
|
|
family.set_handle(val)
|
|
self.add_family(family,transaction)
|
|
return family
|
|
|
|
def get_person_from_gramps_id(self,val):
|
|
"""
|
|
Finds a Person in the database from the passed GRAMPS ID.
|
|
If no such Person exists, None is returned.
|
|
|
|
Needs to be overridden by the derrived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def get_family_from_gramps_id(self,val):
|
|
"""
|
|
Finds a Family in the database from the passed GRAMPS ID.
|
|
If no such Family exists, None is returned.
|
|
|
|
Needs to be overridden by the derrived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def get_place_from_gramps_id(self,val):
|
|
"""finds a Place in the database from the passed gramps' ID.
|
|
If no such Place exists, a new Person is added to the database.
|
|
|
|
Needs to be overridden by the derrived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def get_source_from_gramps_id(self,val):
|
|
"""finds a Source in the database from the passed gramps' ID.
|
|
If no such Source exists, a new Person is added to the database.
|
|
|
|
Needs to be overridden by the derrived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def get_object_from_gramps_id(self,val):
|
|
"""finds a MediaObject in the database from the passed gramps' ID.
|
|
If no such MediaObject exists, a new Person is added to the database.
|
|
|
|
Needs to be overridden by the derrived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def add_person(self,person,transaction):
|
|
"""
|
|
Adds a Person to the database, assigning internal IDs if they have
|
|
not already been defined.
|
|
"""
|
|
if person.get_gramps_id() == None:
|
|
person.set_gramps_id(self.find_next_person_gramps_id())
|
|
if person.get_handle() == None:
|
|
person.set_handle(Utils.create_id())
|
|
self.commit_person(person,transaction)
|
|
self.genderStats.count_person (person, self)
|
|
return person.get_handle()
|
|
|
|
def add_family(self,family,transaction):
|
|
"""
|
|
Adds a Family to the database, assigning internal IDs if they have
|
|
not already been defined.
|
|
"""
|
|
if family.get_gramps_id() == None:
|
|
family.set_gramps_id(self.find_next_family_gramps_id())
|
|
if family.get_handle() == None:
|
|
family.set_handle(Utils.create_id())
|
|
self.commit_family(family,transaction)
|
|
return family.get_handle()
|
|
|
|
def add_source(self,source,transaction):
|
|
"""
|
|
Adds a Source to the database, assigning internal IDs if they have
|
|
not already been defined.
|
|
"""
|
|
if source.get_handle() == None:
|
|
source.set_handle(Utils.create_id())
|
|
if source.get_gramps_id() == None:
|
|
source.set_gramps_id(self.find_next_source_gramps_id())
|
|
self.commit_source(source,transaction)
|
|
return source.get_handle()
|
|
|
|
def add_event(self,event,transaction):
|
|
"""
|
|
Adds an Event to the database, assigning internal IDs if they have
|
|
not already been defined.
|
|
"""
|
|
if event.get_handle() == None:
|
|
event.set_handle(Utils.create_id())
|
|
if event.get_gramps_id() == None:
|
|
event.set_gramps_id(self.find_next_event_gramps_id())
|
|
self.commit_event(event,transaction)
|
|
return event.get_handle()
|
|
|
|
def add_place(self,place,transaction):
|
|
"""
|
|
Adds a Place to the database, assigning internal IDs if they have
|
|
not already been defined.
|
|
"""
|
|
if place.get_handle() == None:
|
|
index = Utils.create_id()
|
|
place.set_handle(index)
|
|
if place.get_gramps_id() == None:
|
|
place.set_gramps_id(self.find_next_place_gramps_id())
|
|
self.commit_place(place,transaction)
|
|
return place.get_handle()
|
|
|
|
def add_object(self,obj,transaction):
|
|
"""
|
|
Adds a MediaObject to the database, assigning internal IDs if they have
|
|
not already been defined.
|
|
"""
|
|
index = obj.get_handle()
|
|
if index == None:
|
|
index = Utils.create_id()
|
|
obj.set_handle(index)
|
|
if obj.get_gramps_id() == None:
|
|
obj.set_gramps_id(self.find_next_object_gramps_id())
|
|
self.commit_media_object(obj,transaction)
|
|
return index
|
|
|
|
def get_people_view_maps(self):
|
|
"""
|
|
Allows the saving people display data into the database metadata.
|
|
This allows faster display of the treeview.
|
|
"""
|
|
if self.metadata:
|
|
return (self.metadata.get('tp_iter'),
|
|
self.metadata.get('tp_path'),
|
|
self.metadata.get('p_iter'),
|
|
self.metadata.get('p_path'),
|
|
self.metadata.get('sname'))
|
|
else:
|
|
return (None,None,None,None,None)
|
|
|
|
def set_people_view_maps(self,maps):
|
|
"""
|
|
Allows the retreiving people display data into the database metadata.
|
|
This allows faster display of the treeview.
|
|
"""
|
|
if self.metadata:
|
|
self.metadata['tp_iter'] = maps[0]
|
|
self.metadata['tp_path'] = maps[1]
|
|
self.metadata['p_iter'] = maps[2]
|
|
self.metadata['p_path'] = maps[3]
|
|
self.metadata['sname'] = maps[4]
|
|
|
|
def get_number_of_people(self):
|
|
"""
|
|
Returns the number of people currently in the databse.
|
|
"""
|
|
return len(self.person_map)
|
|
|
|
def get_person_handles(self,sort_handles=True):
|
|
"""
|
|
Returns a list of database handles, one handle for each Person in
|
|
the database. If sort_handles is True, the list is sorted by surnames
|
|
"""
|
|
if self.person_map:
|
|
handle_list = self.person_map.keys()
|
|
if sort_handles:
|
|
handle_list.sort(self._sortbyname)
|
|
return handle_list
|
|
return []
|
|
|
|
def get_place_handles(self,sort_handles=True):
|
|
"""
|
|
Returns a list of database handles, one handle for each Place in
|
|
the database. If sort_handles is True, the list is sorted by
|
|
Place title.
|
|
"""
|
|
if self.place_map:
|
|
handle_list = self.place_map.keys()
|
|
if sort_handles:
|
|
handle_list.sort(self._sortbyplace)
|
|
return handle_list
|
|
return []
|
|
|
|
def get_source_handles(self,sort_handles=True):
|
|
"""
|
|
Returns a list of database handles, one handle for each Source in
|
|
the database. If sort_handles is True, the list is sorted by
|
|
Source title.
|
|
"""
|
|
if self.source_map:
|
|
handle_list = self.source_map.keys()
|
|
if sort_handles:
|
|
handle_list.sort(self._sortbysource)
|
|
return handle_list
|
|
return []
|
|
|
|
def get_media_object_handles(self,sort_handles=True):
|
|
"""
|
|
Returns a list of database handles, one handle for each MediaObject in
|
|
the database. If sort_handles is True, the list is sorted by title.
|
|
"""
|
|
if self.media_map:
|
|
handle_list = self.media_map.keys()
|
|
if sort_handles:
|
|
handle_list.sort(self._sortbymedia)
|
|
return handle_list
|
|
return []
|
|
|
|
def get_event_handles(self):
|
|
"""
|
|
Returns a list of database handles, one handle for each Event in
|
|
the database.
|
|
"""
|
|
if self.event_map:
|
|
return self.event_map.keys()
|
|
return []
|
|
|
|
def get_family_handles(self):
|
|
"""
|
|
Returns a list of database handles, one handle for each Family in
|
|
the database.
|
|
"""
|
|
if self.family_map:
|
|
return self.family_map.keys()
|
|
return []
|
|
|
|
def set_person_id_prefix(self,val):
|
|
"""
|
|
Sets the naming template for GRAMPS Person ID values. The string is expected
|
|
to be in the form of a simple text string, or in a format that contains
|
|
a C/Python style format string using %d, such as I%d or I%04d.
|
|
"""
|
|
if val:
|
|
if _id_reg.search(val):
|
|
self.iprefix = val
|
|
else:
|
|
self.iprefix = val + "%d"
|
|
else:
|
|
self.iprefix = "I%04d"
|
|
|
|
def set_source_id_prefix(self,val):
|
|
"""
|
|
Sets the naming template for GRAMPS Source ID values. The string is expected
|
|
to be in the form of a simple text string, or in a format that contains
|
|
a C/Python style format string using %d, such as S%d or S%04d.
|
|
"""
|
|
if val:
|
|
if _id_reg.search(val):
|
|
self.sprefix = val
|
|
else:
|
|
self.sprefix = val + "%d"
|
|
else:
|
|
self.sprefix = "S%04d"
|
|
|
|
def set_object_id_prefix(self,val):
|
|
"""
|
|
Sets the naming template for GRAMPS MediaObject ID values. The string is expected
|
|
to be in the form of a simple text string, or in a format that contains
|
|
a C/Python style format string using %d, such as O%d or O%04d.
|
|
"""
|
|
if val:
|
|
if _id_reg.search(val):
|
|
self.oprefix = val
|
|
else:
|
|
self.oprefix = val + "%d"
|
|
else:
|
|
self.oprefix = "O%04d"
|
|
|
|
def set_place_id_prefix(self,val):
|
|
"""
|
|
Sets the naming template for GRAMPS Place ID values. The string is expected
|
|
to be in the form of a simple text string, or in a format that contains
|
|
a C/Python style format string using %d, such as P%d or P%04d.
|
|
"""
|
|
if val:
|
|
if _id_reg.search(val):
|
|
self.pprefix = val
|
|
else:
|
|
self.pprefix = val + "%d"
|
|
else:
|
|
self.pprefix = "P%04d"
|
|
|
|
def set_family_id_prefix(self,val):
|
|
"""
|
|
Sets the naming template for GRAMPS Family ID values. The string is expected
|
|
to be in the form of a simple text string, or in a format that contains
|
|
a C/Python style format string using %d, such as F%d or F%04d.
|
|
"""
|
|
if val:
|
|
if _id_reg.search(val):
|
|
self.fprefix = val
|
|
else:
|
|
self.fprefix = val + "%d"
|
|
else:
|
|
self.fprefix = "F%04d"
|
|
|
|
def set_event_id_prefix(self,val):
|
|
"""
|
|
Sets the naming template for GRAMPS Event ID values. The string is expected
|
|
to be in the form of a simple text string, or in a format that contains
|
|
a C/Python style format string using %d, such as E%d or E%04d.
|
|
"""
|
|
if val:
|
|
if _id_reg.search(val):
|
|
self.eprefix = val
|
|
else:
|
|
self.eprefix = val + "%d"
|
|
else:
|
|
self.eprefix = "E%04d"
|
|
|
|
def transaction_begin(self,msg=""):
|
|
"""
|
|
Creates a new Transaction tied to the current UNDO database. The transaction
|
|
has no effect until it is committed using the transaction_commit function of
|
|
the this database object.
|
|
"""
|
|
return Transaction(msg,self.undodb)
|
|
|
|
def transaction_commit(self,transaction,msg):
|
|
"""
|
|
Commits the transaction to the assocated UNDO database.
|
|
"""
|
|
if not len(transaction):
|
|
return
|
|
transaction.set_description(msg)
|
|
self.undoindex += 1
|
|
if self.undoindex == _UNDO_SIZE:
|
|
self.translist = transaction[0:-1] + [ transaction ]
|
|
else:
|
|
self.translist[self.undoindex] = transaction
|
|
|
|
if self.undo_callback:
|
|
self.undo_callback(_("_Undo %s") % transaction.get_description())
|
|
|
|
def undo(self):
|
|
"""
|
|
Accesses the last committed transaction, and reverts the data to
|
|
the state before the transaction was committed.
|
|
"""
|
|
if self.undoindex == -1:
|
|
return
|
|
transaction = self.translist[self.undoindex]
|
|
|
|
self.undoindex -= 1
|
|
subitems = transaction.get_recnos()
|
|
subitems.reverse()
|
|
for record_id in subitems:
|
|
(key, handle, data) = transaction.get_record(record_id)
|
|
if key == PERSON_KEY:
|
|
if data == None:
|
|
del self.person_map[str(handle)]
|
|
else:
|
|
self.person_map[str(handle)] = data
|
|
elif key == FAMILY_KEY:
|
|
if data == None:
|
|
del self.family_map[str(handle)]
|
|
else:
|
|
self.family_map[str(handle)] = data
|
|
elif key == SOURCE_KEY:
|
|
if data == None:
|
|
del self.source_map[str(handle)]
|
|
else:
|
|
self.source_map[str(handle)] = data
|
|
elif key == EVENT_KEY:
|
|
if data == None:
|
|
del self.event_map[str(handle)]
|
|
else:
|
|
self.event_map[str(handle)] = data
|
|
elif key == PLACE_KEY:
|
|
if data == None:
|
|
del self.place_map[str(handle)]
|
|
else:
|
|
self.place_map[str(handle)] = data
|
|
elif key == MEDIA_KEY:
|
|
if data == None:
|
|
del self.media_map[str(handle)]
|
|
else:
|
|
self.media_map[str(handle)] = data
|
|
|
|
if self.undo_callback:
|
|
if self.undoindex == -1:
|
|
self.undo_callback(None)
|
|
else:
|
|
transaction = self.translist[self.undoindex]
|
|
self.undo_callback(_("_Undo %s") % transaction.get_description())
|
|
|
|
def set_undo_callback(self,callback):
|
|
"""
|
|
Defines the callback function that is called whenever an undo operation
|
|
is executed. The callback function recieves a single argument that is a
|
|
text string that defines the operation.
|
|
"""
|
|
self.undo_callback = callback
|
|
|
|
def set_redo_callback(self,callback):
|
|
"""
|
|
Defines the callback function that is called whenever an redo operation
|
|
is executed. The callback function recieves a single argument that is a
|
|
text string that defines the operation.
|
|
"""
|
|
self.redo_callback = callback
|
|
|
|
def get_surname_list(self):
|
|
"""
|
|
Returns the list of surnames contained within the database.
|
|
The function must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def get_person_event_type_list(self):
|
|
"""
|
|
Returns the list of personal event types contained within the
|
|
database. The function must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def get_bookmarks(self):
|
|
"""returns the list of Person handles in the bookmarks"""
|
|
return self.bookmarks
|
|
|
|
def set_researcher(self,owner):
|
|
"""sets the information about the owner of the database"""
|
|
self.owner.set(owner.get_name(),owner.get_address(),owner.get_city(),
|
|
owner.get_state(),owner.get_country(),
|
|
owner.get_postal_code(),owner.get_phone(),owner.get_email())
|
|
|
|
def get_researcher(self):
|
|
"""returns the Researcher instance, providing information about
|
|
the owner of the database"""
|
|
return self.owner
|
|
|
|
def set_default_person_handle(self,handle):
|
|
"""sets the default Person to the passed instance"""
|
|
self.metadata['default'] = handle
|
|
|
|
def get_default_person(self):
|
|
"""returns the default Person of the database"""
|
|
if self.metadata and self.metadata.has_key('default'):
|
|
person = Person()
|
|
handle = self.metadata['default']
|
|
data = self.person_map.get(str(handle))
|
|
person.unserialize(data)
|
|
return person
|
|
return None
|
|
|
|
def get_save_path(self):
|
|
"""returns the save path of the file, or "" if one does not exist"""
|
|
return self.path
|
|
|
|
def set_save_path(self,path):
|
|
"""sets the save path for the database"""
|
|
self.path = path
|
|
|
|
def get_person_event_types(self):
|
|
"""returns a list of all Event types assocated with Person
|
|
instances in the database"""
|
|
return []
|
|
|
|
def get_person_attribute_types(self):
|
|
"""returns a list of all Attribute types assocated with Person
|
|
instances in the database"""
|
|
return []
|
|
|
|
def get_family_attribute_types(self):
|
|
"""returns a list of all Attribute types assocated with Family
|
|
instances in the database"""
|
|
return []
|
|
|
|
def get_family_event_types(self):
|
|
"""returns a list of all Event types assocated with Family
|
|
instances in the database"""
|
|
return []
|
|
|
|
def get_media_attribute_types(self):
|
|
"""returns a list of all Attribute types assocated with Media
|
|
instances in the database"""
|
|
return []
|
|
|
|
def get_family_relation_types(self):
|
|
"""returns a list of all relationship types assocated with Family
|
|
instances in the database"""
|
|
return []
|
|
|
|
def remove_person(self,handle,transaction):
|
|
"""
|
|
Removes the Person specified by the database handle from the
|
|
database, preserving the change in the passed transaction. This
|
|
method must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def remove_source(self,handle,transaction):
|
|
"""
|
|
Removes the Source specified by the database handle from the
|
|
database, preserving the change in the passed transaction. This
|
|
method must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def remove_event(self,handle,transaction):
|
|
"""
|
|
Removes the Event specified by the database handle from the
|
|
database, preserving the change in the passed transaction. This
|
|
method must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def remove_object(self,handle,transaction):
|
|
"""
|
|
Removes the MediaObjectPerson specified by the database handle from the
|
|
database, preserving the change in the passed transaction. This
|
|
method must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def remove_place(self,handle,transaction):
|
|
"""
|
|
Removes the Place specified by the database handle from the
|
|
database, preserving the change in the passed transaction. This
|
|
method must be overridden in the derived class.
|
|
"""
|
|
assert False, "Needs to be overridden in the derived class"
|
|
|
|
def has_person_handle(self,handle):
|
|
"""
|
|
returns True if the handle exists in the current Person database.
|
|
"""
|
|
return self.person_map.has_key(str(handle))
|
|
|
|
def has_family_handle(self,handle):
|
|
"""
|
|
returns True if the handle exists in the current Family database.
|
|
"""
|
|
return self.family_map.has_key(str(handle))
|
|
|
|
def has_object_handle(self,handle):
|
|
"""
|
|
returns True if the handle exists in the current MediaObjectdatabase.
|
|
"""
|
|
return self.media_map.has_key(str(handle)) != None
|
|
|
|
def _sortbyname(self,f,s):
|
|
n1 = self.person_map.get(str(f))[2].sname
|
|
n2 = self.person_map.get(str(s))[2].sname
|
|
return locale.strcoll(n1,n2)
|
|
|
|
def _sortbyplace(self,f,s):
|
|
return locale.strcoll(self.place_map.get(str(f))[2].upper(),
|
|
self.place_map.get(str(s))[2].upper())
|
|
|
|
def _sortbysource(self,f,s):
|
|
fp = self.source_map[str(f)][2].upper()
|
|
sp = self.source_map[str(s)][2].upper()
|
|
return locale.strcoll(fp,sp)
|
|
|
|
def _sortbymedia(self,f,s):
|
|
fp = self.media_map[str(f)][4].upper()
|
|
sp = self.media_map[str(s)][4].upper()
|
|
return locale.strcoll(fp,sp)
|
|
|
|
def set_person_column_order(self,list):
|
|
"""
|
|
Stores the Person display common information in the
|
|
database's metadata.
|
|
"""
|
|
if self.metadata != None:
|
|
self.metadata['columns'] = list
|
|
|
|
def set_place_column_order(self,list):
|
|
"""
|
|
Stores the Place display common information in the
|
|
database's metadata.
|
|
"""
|
|
if self.metadata != None:
|
|
self.metadata['place_columns'] = list
|
|
|
|
def set_source_column_order(self,list):
|
|
"""
|
|
Stores the Source display common information in the
|
|
database's metadata.
|
|
"""
|
|
if self.metadata != None:
|
|
self.metadata['source_columns'] = list
|
|
|
|
def set_media_column_order(self,list):
|
|
"""
|
|
Stores the Media display common information in the
|
|
database's metadata.
|
|
"""
|
|
if self.metadata != None:
|
|
self.metadata['media_columns'] = list
|
|
|
|
def get_person_column_order(self):
|
|
"""
|
|
Returns the Person display common information stored in the
|
|
database's metadata.
|
|
"""
|
|
default = [(1,1),(1,2),(1,3),(0,4),(1,5),(0,6),(0,7),(0,8)]
|
|
if self.metadata == None:
|
|
return default
|
|
else:
|
|
cols = self.metadata.get('columns',default)
|
|
if len(cols) != len(default):
|
|
return cols + default[len(cols):]
|
|
else:
|
|
return cols
|
|
|
|
def get_place_column_order(self):
|
|
"""
|
|
Returns the Place display common information stored in the
|
|
database's metadata.
|
|
"""
|
|
default = [(1,1),(1,2),(0,3),(1,4),(0,5),(1,6),(0,7),(0,8),(0,9)]
|
|
if self.metadata == None:
|
|
return default
|
|
else:
|
|
cols = self.metadata.get('place_columns',default)
|
|
if len(cols) != len(default):
|
|
return cols + default[len(cols):]
|
|
else:
|
|
return cols
|
|
|
|
def get_source_column_order(self):
|
|
"""
|
|
Returns the Source display common information stored in the
|
|
database's metadata.
|
|
"""
|
|
default = [(1,1),(1,2),(1,3),(0,4),(0,5)]
|
|
if self.metadata == None:
|
|
return default
|
|
else:
|
|
cols = self.metadata.get('source_columns',default)
|
|
if len(cols) != len(default):
|
|
return cols + default[len(cols):]
|
|
else:
|
|
return cols
|
|
|
|
def get_media_column_order(self):
|
|
"""
|
|
Returns the MediaObject display common information stored in the
|
|
database's metadata.
|
|
"""
|
|
default = [(1,1),(1,2),(1,3),(0,4)]
|
|
if self.metadata == None:
|
|
return default
|
|
else:
|
|
cols = self.metadata.get('media_columns',default)
|
|
if len(cols) != len(default):
|
|
return cols + default[len(cols):]
|
|
else:
|
|
return cols
|
|
|
|
class Transaction:
|
|
"""
|
|
Defines a group of database commits that define a single logical
|
|
operation.
|
|
"""
|
|
def __init__(self,msg,db):
|
|
"""
|
|
Creates a new transaction. A Transaction instance should not be created
|
|
directly, but by the GrampsDbBase class or classes derived from
|
|
GrampsDbBase. The db parameter is a list-like interface that stores
|
|
the commit data. This could be a simple list, or a RECNO-style database
|
|
object.
|
|
"""
|
|
self.db = db
|
|
self.first = None
|
|
self.last = None
|
|
|
|
def get_description(self):
|
|
"""
|
|
Returns the text string that describes the logical operation
|
|
performed by the Transaction.
|
|
"""
|
|
return self.msg
|
|
|
|
def set_description(self,msg):
|
|
"""
|
|
Sets the text string that describes the logical operation
|
|
performed by the Transaction.
|
|
"""
|
|
self.msg = msg
|
|
|
|
def add(self, type, handle, data):
|
|
"""
|
|
Adds a commit operation to the Transaction. The type is a constant
|
|
that indicates what type of PrimaryObject is being added. The handle
|
|
is the object's database handle, and the data is the tuple returned
|
|
by the object's serialize method.
|
|
"""
|
|
self.last = self.db.append(cPickle.dumps((type,handle,data),1))
|
|
if self.first == None:
|
|
self.first = self.last
|
|
|
|
def get_recnos(self):
|
|
"""
|
|
Returns a list of record numbers associated with the transaction.
|
|
While the list is an arbitrary index of integers, it can be used
|
|
to indicate record numbers for a database.
|
|
"""
|
|
return range (self.first, self.last+1)
|
|
|
|
def get_record(self,recno):
|
|
"""
|
|
Returns a tuple representing the PrimaryObject type, database handle
|
|
for the PrimaryObject, and a tuple representing the data created by
|
|
the object's serialize method.
|
|
"""
|
|
return cPickle.loads(self.db[recno])
|
|
|
|
def __len__(self):
|
|
"""
|
|
Returns the number of commits associated with the Transaction.
|
|
"""
|
|
if self.last and self.first:
|
|
return self.last - self.first + 1
|
|
return 0
|
|
|