_SimpleAccess.py from gramps at Krugle
Show _SimpleAccess.py syntax highlighted
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2007 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
#
"""
Provides a simplified database access interface to the GRAMPS database.
"""
from types import NoneType
import RelLib
import DateHandler
import Utils
from BasicUtils import name_displayer
from ReportBase import ReportUtils
from RelLib import EventType
class SimpleAccess:
"""
Provides a simplified database access system. This system has been designed to
ease the development of reports.
The user needs to take care when using this interface. Since it returns real
objects instead of database references, it can consume a significant amount
of memory if the user is not careful.
Example
=======
A typical system of usage would be::
sa = SimpleAccess(database)
print "Person : ", sa.name(person)
print "Gender : ", sa.gender(person)
print "Birth date : ", sa.birth_date(person)
print "Birth place : ", sa.birth_place(person)
print "Death date : ", sa.death_date(person)
print "Death place : ", sa.death_place(person)
print "Father : ", sa.name(sa.father(person))
print "Mother : ", sa.name(sa.mother(person))
print "Spouse : ", sa.name(sa.spouse(person))
print "Marriage Type : ", sa.marriage_type(person)
print "Marriage Date : ", sa.marriage_date(person)
print "Marriage Place: ", sa.marriage_place(person)
for child in sa.children(person):
print "Child : ", sa.name(child)
# Print out burial and baptism events
for event in sa.events( person , [ "Burial", "Baptism" ]):
print "Event : ", sa.event_type(event), sa.event_date(event),
print sa.event_place(event)
This would produce an output that looks like::
Person : Garner, Lewis Anderson
Gender : male
Birth date : 6/21/1855
Birth place : Great Falls, MT
Death date : 6/28/1911
Death place : Twin Falls, ID
Father : Garner, Robert W.
Mother : Zielinski, Phoebe Emily
Spouse : Martel, Luella Jacques
Marriage Type : Married
Marriage Date : 4/1/1875
Marriage Place: Paragould, AR
Child : Garner, Eugene Stanley
Child : Garner, Jesse V.
Child : Garner, Raymond E.
Child : Garner, Jennie S.
Child : Garner, Walter E.
Child : Garner, Daniel Webster
Child : Garner, Bertha P.
Child : Garner, Elizabeth
Event : Burial 7/1/1911 Twin Falls, ID
"""
def __init__(self, dbase):
"""
Initializes the SimpleAccess object with the database that will be used.
@param dbase: GRAMPS database object
@type dbase: GrampsDbBase
"""
self.dbase = dbase
def name(self, person):
"""
Returns the name of the person, or and empty string if the person is None
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns the name of the person based of the program preferences
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
return name_displayer.display(person)
else:
return u''
def surname(self, person):
"""
Returns the name of the person, or and empty string if the person is None
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns the name of the person based of the program preferences
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
return person.get_primary_name().get_surname()
else:
return u''
def first_name(self, person):
"""
Returns the first name of the person, or and empty string if the person is None
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns the first name of the person based of the program preferences
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
return person.get_primary_name().get_first_name()
else:
return u''
def gid(self, obj):
"""
Returns the GRAMPS ID of the person or family
@param obj: Person or Family object
@type obj: L{RelLib.Person} or L{RelLib.Family}
@return: Returns the GRAMPS Id value of the person or family
@rtype: unicode
"""
assert(isinstance(obj, (RelLib.Person, RelLib.Family, NoneType)))
if obj:
return obj.get_gramps_id()
else:
return u''
def gender(self, person):
"""
Returns a string representing the gender of the person
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string indentifying the person's gender
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
return Utils.gender[person.get_gender()]
return u''
def __parent(self, person, func):
"""
Returns a person associated as a parent of the person
@param person: Person object
@type person: L{RelLib.Person}
@param func: function used to extract the appropriate parent
@type func: function
@return: mother or father of the associated person
@rtype: L{RelLib.Person}
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
parent_handle_list = person.get_parent_family_handle_list()
if parent_handle_list:
family_id = parent_handle_list[0]
family = self.dbase.get_family_from_handle(family_id)
return self.__family_parent(family, func)
return None
def __family_parent(self, family, func):
"""
Returns a person associated as a parent of the family
@param family: Family object
@type family: L{RelLib.Family}
@param func: function used to extract the appropriate parent
@type func: function
@return: mother or father of the associated family
@rtype: L{RelLib.Family}
"""
assert(isinstance(family, (RelLib.Family, NoneType)))
if family:
handle = func(family)
if handle:
return self.dbase.get_person_from_handle(handle)
return None
def __event_date(self, person, func):
"""
Returns a string describing the date associated with the person
@param person: Person object
@type person: L{RelLib.Person}
@param func: function used to extract the associated date information
@type func: function
@return: Returns a string describing the date
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
ref = func(person)
if ref:
event_handle = ref.get_reference_handle()
if event_handle:
event = self.dbase.get_event_from_handle(event_handle)
date_obj = event.get_date_object()
if date_obj:
return DateHandler.displayer.display(date_obj)
return u''
def __event_place(self, person, func):
"""
Returns a string describing the place associated with the person
@param person: Person object
@type person: L{RelLib.Person}
@param func: function used to extract the associated place information
@type func: function
@return: Returns a string describing the place
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
ref = func(person)
if ref:
event_handle = ref.get_reference_handle()
if event_handle:
event = self.dbase.get_event_from_handle(event_handle)
place_handle = event.get_place_handle()
return ReportUtils.place_name(self.dbase, place_handle)
return u''
def spouse(self, person):
"""
Returns the primary spouse of the person
@param person: Person object
@type person: L{RelLib.Person}
@return: The spouse identified as the person's primary spouse
@rtype: L{RelLib.Person}
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
family_handle_list = person.get_family_handle_list()
if family_handle_list:
family_id = family_handle_list[0]
family = self.dbase.get_family_from_handle(family_id)
if family.get_father_handle() == person.get_handle():
person_handle = family.get_mother_handle()
else:
person_handle = family.get_father_handle()
if person_handle:
return self.dbase.get_person_from_handle(person_handle)
return None
def marriage_type(self, person):
"""
Returns a string describing the relationship between the person and
his/per primary spouse.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string describing the relationship between the person and
his/per primary spouse.
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
family_handle_list = person.get_family_handle_list()
if family_handle_list:
family_id = family_handle_list[0]
family = self.dbase.get_family_from_handle(family_id)
if family:
return str(family.get_relationship())
return u''
def marriage_place(self, person):
"""
Returns a string describing the place where the person and his/her spouse
where married.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string describing the place where the person and his/her spouse
where married.
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
family_handle_list = person.get_family_handle_list()
if family_handle_list:
family_id = family_handle_list[0]
family = self.dbase.get_family_from_handle(family_id)
if family:
reflist = family.get_event_ref_list()
if reflist:
elist = [ self.dbase.get_event_from_handle(ref.ref)
for ref in reflist ]
events = [ evnt for evnt in elist
if int(evnt.get_type()) == EventType.MARRIAGE ]
if events:
place_handle = events[0].get_place_handle()
return ReportUtils.place_name(self.dbase, place_handle)
return u''
def marriage_date(self, person):
"""
Returns a string indicating the date when the person and his/her spouse
where married.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string indicicating the date when the person and his/her spouse
where married.
@rtype: unicode
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
family_handle_list = person.get_family_handle_list()
if family_handle_list:
family_id = family_handle_list[0]
family = self.dbase.get_family_from_handle(family_id)
if family:
reflist = family.get_event_ref_list()
if reflist:
elist = [ self.dbase.get_event_from_handle(ref.ref)
for ref in reflist ]
events = [ evnt for evnt in elist
if int(evnt.get_type()) == EventType.MARRIAGE ]
if events:
date_obj = events[0].get_date_object()
if date_obj:
return DateHandler.displayer.display(date_obj)
return u''
def children(self, obj):
"""
Returns a list of the children as the children of the primary spouse.
@param obj: Person or Family object
@type obj: L{RelLib.Person} or L{RelLib.Family}
@return: Returns a list of L{RelLib.Person} objects representing the children
@rtype: list
"""
assert(isinstance(obj, (RelLib.Person, RelLib.Family, NoneType)))
if isinstance(obj, RelLib.Person):
family_handle_list = obj.get_family_handle_list()
if family_handle_list:
family_id = family_handle_list[0]
family = self.dbase.get_family_from_handle(family_id)
return [ self.dbase.get_person_from_handle(hndl.ref)
for hndl in family.get_child_ref_list() ]
elif isinstance(obj, RelLib.Family):
return [ self.dbase.get_person_from_handle(hndl.ref)
for hndl in obj.get_child_ref_list() ]
return []
def father(self, obj):
"""
Returns the primary father of the person or the father of the associated
family.
@param obj: Person or Family object
@type obj: L{RelLib.Person} or L{RelLib.Family}
@return: The father in the person's primary family or the father of the
family
@rtype: L{RelLib.Person}
"""
if isinstance(obj, RelLib.Person):
return self.__parent(obj, RelLib.Family.get_father_handle)
elif isinstance(obj, RelLib.Family):
return self.__family_parent(obj, RelLib.Family.get_father_handle)
else:
return None
def mother(self, obj):
"""
Returns the primary mother of the person or the mother of the associated
family.
@param obj: Person object
@type obj: L{RelLib.Person} or L{RelLib.Family}
@return: The mother in the person's primary family or the mother of the
family
@rtype: L{RelLib.Person}
"""
if isinstance(obj, RelLib.Person):
return self.__parent(obj, RelLib.Family.get_mother_handle)
elif isinstance(obj, RelLib.Family):
return self.__family_parent(obj, RelLib.Family.get_mother_handle)
else:
return None
def birth_date(self, person):
"""
Returns a string indicating the date when the person's birth.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string indicating the date when the person's birth.
@rtype: unicode
"""
return self.__event_date(person, RelLib.Person.get_birth_ref)
def birth_place(self, person):
"""
Returns a string indicating the place of the person's birth.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string indicating the place of the person's birth.
@rtype: unicode
"""
return self.__event_place(person, RelLib.Person.get_birth_ref)
def death_date(self, person):
"""
Returns a string indicating the date when the person's death.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string indicating the date when the person's death.
@rtype: unicode
"""
return self.__event_date(person, RelLib.Person.get_death_ref)
def death_place(self, person):
"""
Returns a string indicating the place of the person's death.
@param person: Person object
@type person: L{RelLib.Person}
@return: Returns a string indicating the place of the person's death.
@rtype: unicode
"""
return self.__event_place(person, RelLib.Person.get_death_ref)
def event_place(self, event):
"""
Returns a string indicating the place of the event
@param event: Event object
@type event: L{RelLib.Event}
@return: Returns a string indicating the place of the event
@rtype: unicode
"""
assert(isinstance(event, (RelLib.Event, NoneType)))
if event:
place_handle = event.get_place_handle()
return ReportUtils.place_name(self.dbase, place_handle)
else:
return u''
def event_date(self, event):
"""
Returns a string indicating the date of the event
@param event: Event object
@type event: L{RelLib.Event}
@return: Returns a string indicating the date of the event
@rtype: unicode
"""
assert(isinstance(event, (RelLib.Event, NoneType)))
date_obj = event.get_date_object()
if date_obj:
return DateHandler.displayer.display(date_obj)
else:
return u''
def event_type(self, event):
"""
Returns a string indicating the type of the event
@param event: Event object
@type event: L{RelLib.Event}
@return: Returns a string indicating the type of the event
@rtype: unicode
"""
assert(isinstance(event, (RelLib.Event, NoneType)))
if event:
return str(event.get_type())
else:
return u''
def events(self, obj, restrict=None):
"""
Returns a list of events associated with the object. This object
can be either a L{RelLib.Person} or L{RelLib.Family}.
@param obj: Person or Family
@type obj: L{RelLib.Person} or L{RelLib.Family}
@param restrict: Optional list of strings that will limit the types
of events to those of the specfied types.
@type restrict: list
@return: list of events assocated with the object
@rtype: list
"""
assert(isinstance(obj, (RelLib.Person, RelLib.Family, NoneType)))
assert(type(restrict) == type([]) or restrict == None)
if obj:
event_handles = [ ref.ref for ref in obj.get_event_ref_list() ]
events = [ self.dbase.get_event_from_handle(h)
for h in event_handles ]
if restrict:
restrict = [ r.lower() for r in restrict ]
events = [ event for event in events
if str(event.get_type()).lower() in restrict ]
return events
else:
return []
def sources(self, obj):
"""
Returns a list of events associated with the object. This object
can be either a L{RelLib.Person} or L{RelLib.Family}.
@param obj: Person or Family
@type obj: L{RelLib.Person} or L{RelLib.Family}
@return: list of events assocated with the object
@rtype: list
"""
assert(isinstance(obj, (RelLib.Person, RelLib.Family, RelLib.Event, NoneType)))
if obj:
handles = [ ref.ref for ref in obj.get_source_references() ]
return [ self.dbase.get_source_from_handle(h) for h in handles ]
else:
return []
def parent_in(self, person):
"""
Returns a list of families in which the person is listed as a parent.
@param person: Person object
@type person: L{RelLib.Person}
@return: list of L{RelLib.Family} objects in which the person is listed
as a parent.
@rtype: list
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
return [ self.dbase.get_family_from_handle(handle)
for handle in person.get_family_handle_list() ]
return []
def child_in(self, person):
"""
Returns a list of families in which the person is listed as a child.
@param person: Person object
@type person: L{RelLib.Person}
@return: list of L{RelLib.Family} objects in which the person is listed
as a child.
@rtype: list
"""
assert(isinstance(person, (RelLib.Person, NoneType)))
if person:
return [ self.dbase.get_family_from_handle(handle)
for handle in person.get_parent_family_handle_list() ]
return []
def __all_objects(self, gen_cursor, get_object):
"""
Returns a all the objects of a particular type in the database, one
at a time as an iterator. The user can treat this just like a list.
@return: list of objects of a particular type in the database
@rtype: list
"""
slist = []
cursor = gen_cursor()
data = cursor.first()
while data:
slist.append(data[0])
data = cursor.next()
cursor.close()
for info in slist:
obj = get_object(info)
yield obj
def all_people(self):
"""
Returns a all the people in the database, one at a time as an iterator.
The user can treat this just like a list. For example::
for person in sa.all_people():
sa.print(person)
@return: list of people in the database
@rtype: list
"""
slist = []
cursor = self.dbase.get_person_cursor()
data = cursor.first()
while data:
slist.append((data[1][3][3], data[0]))
data = cursor.next()
cursor.close()
slist.sort()
for info in slist:
obj = self.dbase.get_person_from_handle(info[1])
yield obj
def all_families(self):
"""
Returns all the families in the database, one at a time as an iterator.
The user can treat this just like a list. For example::
for person in sa.all_families():
sa.print(sa.father(person))
@return: list of families in the database
@rtype: list
"""
return self.__all_objects(self.dbase.get_family_cursor,
self.dbase.get_family_from_handle)
def all_events(self):
"""
Returns all the events in the database, one at a time as an iterator.
The user can treat this just like a list. For example::
for person in sa.all_events():
sa.print(sa.event_place(event))
@return: list of events in the database
@rtype: list
"""
return self.__all_objects(self.dbase.get_events_cursor,
self.dbase.get_event_from_handle)
def all_sources(self):
"""
Returns all the sources in the database, one at a time as an iterator.
The user can treat this just like a list. For example::
@return: list of sources in the database
@rtype: list
"""
return self.__all_objects(self.dbase.get_source_cursor,
self.dbase.get_source_from_handle)
def title(self, source):
"""
Returns the title of the source.
@param source: Source object
@type source: L{RelLib.Source}
@return: title of the source
@rtype: unicode
"""
assert(isinstance(source, (RelLib.Source, NoneType)))
if source:
return source.get_title()
return u''
def author(self, source):
"""
Returns the author of the source.
@param source: Source object
@type source: L{RelLib.Source}
@return: author of the source
@rtype: unicode
"""
assert(isinstance(source, (RelLib.Source, NoneType)))
if source:
return source.get_author()
return u''
def by_date(event1, event2):
"""
Sort function that will compare two events by their dates.
@param event1: first event
@type event1: L{Event}
@param event2: second event
@type event2: L{Event}
@return: Returns -1 if event1 < event2, 0 if they are equal, and
1 if they are the same.
@rtype: int
"""
return cmp(event1.get_date_object() , event2.get_date_object())
See more files for this project here