# # 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$ """The core library of the GRAMPS database""" __author__ = "Donald N. Allingham" __version__ = "$Revision$" #------------------------------------------------------------------------- # # standard python modules # #------------------------------------------------------------------------- from gettext import gettext as _ import os import os.path import time #------------------------------------------------------------------------- # # GRAMPS modules # #------------------------------------------------------------------------- import Date import DateHandler import GrampsCfg import const #------------------------------------------------------------------------- # # Confidence levels # #------------------------------------------------------------------------- CONF_VERY_HIGH = 4 CONF_HIGH = 3 CONF_NORMAL = 2 CONF_LOW = 1 CONF_VERY_LOW = 0 #------------------------------------------------------------------------- # # Class definitions # #------------------------------------------------------------------------- display = DateHandler.create_display() parser = DateHandler.create_parser() #------------------------------------------------------------------------- # # Class definitions # #------------------------------------------------------------------------- class PrimaryObject: """ The PrimaryObject is the base class for all primary objects in the database. Primary objects are the core objects in the database. Each object has a database handle and a GRAMPS ID value. The database handle is used as the record number for the database, and the GRAMPS ID is the user visible version. """ def __init__(self,source=None): """ Initialize a PrimaryObject. If source is None, both the ID and handle are assigned as empty strings. If source is not None, then object is initialized from values of the source object. @param source: Object used to initialize the new object @type source: PrimaryObject """ if source: self.gramps_id = source.gramps_id self.handle = source.handle self.change = source.change else: self.gramps_id = None self.handle = None self.change = 0 def get_change_time(self): """ Returns the time that the data was last changed. The value in the format returned by the time.time() command. @returns: Time that the data was last changed. The value in the format returned by the time.time() command. @rtype: int """ return self.change def get_change_display(self): """ Returns the string representation of the last change time. @returns: string representation of the last change time. @rtype: str """ if self.change: return time.asctime(time.localtime(self.change)) else: return '' def set_handle(self,handle): """ Sets the database handle for the primary object @param handle: object database handle @type handle: str """ self.handle = handle def get_handle(self): """ Returns the database handle for the primary object @returns: database handle associated with the object @rtype: str """ return self.handle def set_gramps_id(self,gramps_id): """ Sets the GRAMPS ID for the primary object @param gramps_id: GRAMPS ID @type gramps_id: str """ self.gramps_id = gramps_id def get_gramps_id(self): """ Returns the GRAMPS ID for the primary object @returns: GRAMPS ID associated with the object @rtype: str """ return self.gramps_id class SourceNote: """ Base class for storing source references and notes """ def __init__(self,source=None): """ Create a new SourceNote, copying from source if not None @param source: Object used to initialize the new object @type source: SourceNote """ self.source_list = [] self.note = None if source: for sref in source.source_list: self.source_list.append(SourceRef(sref)) if source.note: self.note = Note(source.note.get()) def add_source_reference(self,src_ref) : """ Adds a source reference to this object. @param src_ref: The source reference to be added to the SourceNote's list of source references. @type src_ref: L{SourceRef} """ self.source_list.append(src_ref) def get_source_references(self) : """ Returns the list of source references associated with the object. @return: Returns the list of L{SourceRef} objects assocated with the object. @rtype: list """ return self.source_list def set_source_reference_list(self,src_ref_list) : """ Assigns the passed list to the object's list of source references. @param src_ref_list: List of source references to ba associated with the object @type src_ref_list: list of L{SourceRef} instances """ self.source_list = src_ref_list def set_note(self,text): """ Assigns the specified text to the associated note. @param text: Text of the note @type text: str """ if self.note == None: self.note = Note() self.note.set(text) def get_note(self): """ Returns the text of the current note. @returns: the text of the current note @rtype: str """ if self.note == None: return "" else: return self.note.get() def set_note_format(self,val): """ Sets the note's format to the given value. The format indicates whether the text is flowed (wrapped) or preformatted. @param val: True indicates the text is flowed @type val: bool """ if self.note: self.note.set_format(val) def get_note_format(self): """ Returns the current note's format @returns: True indicates that the note should be flowed (wrapped) @rtype: bool """ if self.note == None: return False else: return self.note.get_format() def set_note_object(self,note_obj): """ Replaces the current Note object associated with the object @param note_obj: New Note object to be assigned @type note_obj: Note """ self.note = note_obj def get_note_object(self): """ Returns the Note instance associated with the object. @returns: Note object assocated with the object @rtype: Note """ return self.note def unique_note(self): """Creates a unique instance of the current note""" self.note = Note(self.note.get()) class DataObj(SourceNote): """ Base class for data elements, providing source, note, and privacy data """ def __init__(self,source=None): """ Create a new DataObj, copying data from a source object if provided """ SourceNote.__init__(self,source) if source: self.private = source.private else: self.private = False def set_privacy(self,val): """ Sets or clears the privacy flag of the data """ self.private = val def get_privacy(self): """ Returns the privacy level of the data """ return self.private class Person(PrimaryObject,SourceNote): """ GRAMPS Person record. This object represents an individual person. It contains all the information about the person, including names, events, attributes, and other information. """ unknown = 2 male = 1 female = 0 def __init__(self): """ Creates a new Person instance. After initialization, most data items have empty or null values, including the database handle. """ PrimaryObject.__init__(self) SourceNote.__init__(self) self.primary_name = Name() self.event_list = [] self.family_list = [] self.parent_family_list = [] self.media_list = [] self.nickname = "" self.alternate_names = [] self.gender = Person.unknown self.death_handle = None self.birth_handle = None self.address_list = [] self.attribute_list = [] self.urls = [] self.lds_bapt = None self.lds_endow = None self.lds_seal = None self.complete = False # We hold a reference to the GrampsDB so that we can maintain # its genderStats. It doesn't get set here, but from # GenderStats.count_person. self.db = None def serialize(self): """ Converts the data held in the Person to a Python tuple that represents all the data elements. This method is used to convert the object into a form that can easily be saved to a database. These elements may be primative Python types (string, integers), complex Python types (lists or tuples, or Python objects. If the target database cannot handle complex types (such as objectes or lists), the database is responsible for converting the data into a form that it can use. @returns: Returns a python tuple containing the data that should be considered persistent. @rtype: tuple """ return (self.handle, self.gramps_id, self.gender, self.primary_name, self.alternate_names, unicode(self.nickname), self.death_handle, self.birth_handle, self.event_list, self.family_list, self.parent_family_list, self.media_list, self.address_list, self.attribute_list, self.urls, self.lds_bapt, self.lds_endow, self.lds_seal, self.complete, self.source_list, self.note, self.change) def unserialize(self,data): """ Converts the data held in a tuple created by the serialize method back into the data in an Person structure. @param data: tuple containing the persistent data associated the Person object @type data: tuple """ (self.handle, self.gramps_id, self.gender, self.primary_name, self.alternate_names, self.nickname, self.death_handle, self.birth_handle, self.event_list, self.family_list, self.parent_family_list, self.media_list, self.address_list, self.attribute_list, self.urls, self.lds_bapt, self.lds_endow, self.lds_seal, self.complete, self.source_list, self.note, self.change) = data def set_complete_flag(self,val): """ Sets or clears the complete flag, which is used to indicate that the Person's data is considered to be complete. @param val: True indicates the Person object is considered to be complete @type val: bool """ self.complete = val def get_complete_flag(self): """ Returns the complete flag, which is used to indicate that the Person's data is considered to be complete. @return: True indicates that the Person's record is considered to be complete. @rtype: bool """ return self.complete def get_display_info(self): """ Returns a list consisting of the information typically used for a display. The data consists of: Display Name, ID, Gender, Date of Birth, Date of Death, sort name, etc. """ if self.gender == Person.male: gender = const.male elif self.gender == Person.female: gender = const.female else: gender = const.unknown bday = self.birth_handle dday = self.death_handle return [ GrampsCfg.get_display_name()(self), self.gramps_id, gender, bday, dday, self.get_primary_name().get_sort_name(), GrampsCfg.get_display_surname()(self.primary_name)] def set_primary_name(self,name): """ Sets the primary name of the Person to the specified L{Name} instance @param name: L{Name} to be assigned to the person @type name: L{Name} """ db = self.db if db: db.genderStats.uncount_person (self) self.primary_name = name if db: db.genderStats.count_person (self, db) def get_primary_name(self): """ Returns the L{Name} instance marked as the Person's primary name @return: Returns the primary name @rtype: L{Name} """ return self.primary_name def get_alternate_names(self): """ Returns the list of alternate L{Name} instances @return: List of L{Name} instances @rtype: list """ return self.alternate_names def set_alternate_names(self,alt_name_list): """ Changes the list of alternate names to the passed list. @param alt_name_list: List of L{Name} instances @type alt_name_list: list """ self.alternate_names = alt_name_list def add_alternate_name(self,name): """ Adds a L{Name} instance to the list of alternative names @param name: L{Name} to add to the list @type name: L{Name} """ self.alternate_names.append(name) def get_url_list(self): """ Returns the list of L{Url} instances associated with the Person @returns: List of L{Url} instances @rtype: list """ return self.urls def set_url_list(self,url_list): """ Sets the list of L{Url} instances to passed the list. @param url_list: List of L{Url} instances @type url_list: list """ self.urls = url_list def add_url(self,url): """ Adds a L{Url} instance to the Person's list of L{Url} instances @param url: L{Url} instance to be added to the Person's list of related web sites. @type url: L{Url} """ self.urls.append(url) def set_nick_name(self,name): """ Sets the nickname field for the Person @param name: Nickname to be assigned @type name: str """ self.nickname = name def get_nick_name(self) : """ Returns the nickname for the Person @returns: Returns the nickname associated with the Person @rtype str """ return self.nickname def set_gender(self,gender) : """ Sets the gender of the Person. @param gender: Assigns the Person's gender to one of the following constants:: Person.male Person.female Person.unknown @type gender: int """ # if the db object has been assigned, update the # genderStats of the database if self.db: self.db.genderStats.uncount_person (self) self.gender = gender if self.db: self.db.genderStats.count_person (self, self.db) def get_gender(self) : """ Returns the gender of the Person @returns: Returns one of the following constants:: Person.male Person.female Person.unknown @rtype: int """ return self.gender def set_birth_handle(self,event_handle): """ Assigns the birth event to the Person object. This is accomplished by assigning the handle of a valid L{Event} in the current database. @param event_handle: handle of the L{Event} to be associated with the Person's birth. @type event_handle: str """ self.birth_handle = event_handle def set_death_handle(self,event_handle): """ Assigns the death event to the Person object. This is accomplished by assigning the handle of a valid L{Event} in the current database. @param event_handle: handle of the L{Event} to be associated with the Person's death. @type event_handle: str """ self.death_handle = event_handle def get_birth_handle(self): """ Returns the database handle for the Person's birth event. This should correspond to an L{Event} in the database's L{Event} list. @returns: Returns the birth L{Event} handle or None if no birth L{Event} has been assigned. @rtype: str """ return self.birth_handle def get_death_handle(self): """ Returns the database handle for the Person's death event. This should correspond to an L{Event} in the database's L{Event} list. @returns: Returns the death L{Event} handle or None if no death L{Event} has been assigned. @rtype: str """ return self.death_handle def add_media_reference(self,media_ref): """ Adds a L{MediaRef} instance to the Person's media list. @param media_ref: L{MediaRef} instance to be added to the Person's media list. @type media_ref: L{MediaRef} """ self.media_list.append(media_ref) def get_media_list(self): """ Returns the list of L{MediaRef} instances associated with the Person @returns: list of L{MediaRef} instances associated with the Person @rtype: list """ return self.media_list def set_media_list(self,media_ref_list): """ Sets the list of L{MediaRef} instances associated with the Person. It replaces the previous list. @param media_ref_list: list of L{MediaRef} instances to be assigned to the Person. @type media_ref_list: list """ self.media_list = media_ref_list def add_event_handle(self,event_handle): """ Adds the L{Event} to the Person instance's L{Event} list. This is accomplished by assigning the handle of a valid L{Event} in the current database. @param event_handle: handle of the L{Event} to be added to the Person's L{Event} list. @type event_handle: str """ self.event_list.append(event_handle) def get_event_list(self): """ Returns the list of handles associated with L{Event} instances. @returns: Returns the list of L{Event} handles associated with the Person instance. @rtype: list """ return self.event_list def set_event_list(self,event_list): """ Sets the Person instance's L{Event} list to the passed list. @param event_list: List of valid L{Event} handles @type event_list: list """ self.event_list = event_list def add_family_handle(self,family_handle): """ Adds the L{Family} handle to the Person instance's L{Family} list. This is accomplished by assigning the handle of a valid L{Family} in the current database. Adding a L{Family} handle to a Person does not automatically update the corresponding L{Family}. The developer is responsible to make sure that when a L{Family} is added to Person, that the Person is assigned to either the father or mother role in the L{Family}. @param family_handle: handle of the L{Family} to be added to the Person's L{Family} list. @type family_handle: str """ self.family_list.append(family_handle) def set_preferred_family_handle(self,family_handle): """ Sets the family_handle specified to be the preferred L{Family}. The preferred L{Family} is determined by the first L{Family} in the L{Family} list, and is typically used to indicate the preferred L{Family} for navigation or reporting. The family_handle must already be in the list, or the function call has no effect. @param family_handle: Handle of the L{Family} to make the preferred L{Family}. @type family_handle: str @returns: True if the call succeeded, False if the family_handle was not already in the L{Family} list @rtype: bool """ if family_handle in self.family_list: self.family_list.remove(family_handle) self.family_list = [family_handle] + self.family_list return True else: return False def get_family_handle_list(self) : """ Returns the list of L{Family} handles in which the person is a parent or spouse. @return: Returns the list of handles corresponding to the L{Family} records with which the person is associated. @rtype: list """ return self.family_list def set_family_handle_list(self,family_list) : """ Assigns the passed list to the Person's list of families in which it is a parent or spouse. @param family_list: List of L{Family} handles to ba associated with the Person @type family_list: list """ self.family_list = family_list def clear_family_handle_list(self): """ Removes all L{Family} handles from the L{Family} list. """ self.family_list = [] def remove_family_handle(self,family_handle): """ Removes the specified L{Family} handle from the list of marriages/partnerships. If the handle does not exist in the list, the operation has no effect. @param family_handle: L{Family} handle to remove from the list @type family_handle: str @return: True if the handle was removed, False if it was not in the list. @rtype: bool """ if family_handle in self.family_list: self.family_list.remove(family_handle) return True else: return False def add_address(self,address): """ Adds the L{Address} instance to the Person's list of addresses @param address: L{Address} instance to add to the Person's address list @type address: list """ self.address_list.append(address) def remove_address(self,address): """ Removes the specified L{Address} instance from the address list If the instance does not exist in the list, the operation has no effect. @param address: L{Address} instance to remove from the list @type address: L{Address} @return: True if the address was removed, False if it was not in the list. @rtype: bool """ if address in self.address_list: self.address_list.remove(address) return True else: return False def get_address_list(self): """ Returns the list of L{Address} instances associated with the Person @return: Returns the list of L{Address} instances @rtype: list """ return self.address_list def set_address_list(self,address_list): """ Assigns the passed list to the Person's list of L{Address} instances. @param address_list: List of L{Address} instances to ba associated with the Person @type address_list: list """ self.address_list = address_list def add_attribute(self,attribute): """ Adds the L{Attribute} instance to the Person's list of attributes @param attribute: L{Attribute} instance to add to the Person's address list @type attribute: list """ self.attribute_list.append(attribute) def remove_attribute(self,attribute): """ Removes the specified L{Attribute} instance from the attribute list If the instance does not exist in the list, the operation has no effect. @param attribute: L{Attribute} instance to remove from the list @type attribute: L{Attribute} @return: True if the attribute was removed, False if it was not in the list. @rtype: bool """ if attribute in self.attribute_list: self.attribute_list.remove(attribute) return True else: return False def get_attribute_list(self): """ Returns the list of L{Attribute} instances associated with the Person @return: Returns the list of L{Attribute} instances @rtype: list """ return self.attribute_list def set_attribute_list(self,attribute_list): """ Assigns the passed list to the Person's list of L{Attribute} instances. @param attribute_list: List of L{Attribute} instances to ba associated with the Person @type attribute_list: list """ self.attribute_list = attribute_list def get_parent_family_handle_list(self): """ Returns the list of L{Family} handles in which the person is a child. @return: Returns the list of handles corresponding to the L{Family} records with which the person is a child. @rtype: list """ return self.parent_family_list def add_parent_family_handle(self,family_handle,mrel,frel): """ Adds the L{Family} handle to the Person instance's list of families in which it is a child. This is accomplished by assigning the handle of a valid L{Family} in the current database. Adding a L{Family} handle to a Person does not automatically update the corresponding L{Family}. The developer is responsible to make sure that when a L{Family} is added to Person, that the Person is added to the L{Family} instance's child list. @param family_handle: handle of the L{Family} to be added to the Person's L{Family} list. @type family_handle: str @param mrel: relationship between the Person and its mother @type mrel: str @param frel: relationship between the Person and its father @type frel: str """ self.parent_family_list.append((family_handle,mrel,frel)) def clear_parent_family_handle_list(self): """ Removes all L{Family} handles from the parent L{Family} list. """ self.parent_family_list = [] def remove_parent_family_handle(self,family_handle): """ Removes the specified L{Family} handle from the list of parent families (families in which the parent is a child). If the handle does not exist in the list, the operation has no effect. @param family_handle: L{Family} handle to remove from the list @type family_handle: str @return: Returns a tuple of three strings, consisting of the removed handle, relationship to mother, and relationship to father. None is returned if the handle is not in the list. @rtype: tuple """ for f in self.parent_family_list[:]: if f[0] == family_handle: self.parent_family_list.remove(f) return f else: return None def change_parent_family_handle(self,family_handle,mrel,frel): """ Changes the relationships of the L{Family} handle in the Person instance's list of families in which it is a child. The handle is assumed to already be in the list. @param family_handle: handle of the L{Family} to be added to the Person's L{Family} list. @type family_handle: str @param mrel: relationship between the Person and its mother @type mrel: str @param frel: relationship between the Person and its father @type frel: str """ index=0 for f in self.parent_family_list[:]: if f[0] == family_handle: self.parent_family_list[index] = (family_handle,mrel,frel) return True index += 1 return False def get_parent_family(self,family_handle): """ Finds the L{Family} and relationships associated with passed family_handle. @param family_handle: L{Family} handle used to search the parent family list. @type family_handle: str @return: Returns a tuple consisting of the L{Family} handle, the mother relationship, and father relationship @rtype: tuple """ for f in self.parent_family_list: if f[0] == family_handle: return f else: return None def set_main_parent_family_handle(self,family_handle): """ Sets the main L{Family} in which the Person is a child. The main L{Family} is the L{Family} typically used for reports and navigation. This is accomplished by moving the L{Family} to the beginning of the list. The family_handle must be in the list for this to have any effect. @param family_handle: handle of the L{Family} to be marked as the main L{Family} @type family_handle: str @return: Returns True if the assignment has successful @rtype: bool """ f = self.remove_parent_family_handle(family_handle) if f: self.parent_family_list = [f] + self.parent_family_list return True else: return False def get_main_parents_family_handle(self): """ Returns the handle of the L{Family} considered to be the main L{Family} in which the Person is a child. @return: Returns the family_handle if a family_handle exists, If no L{Family} is assigned, None is returned @rtype: str """ if len(self.parent_family_list) == 0: return None else: return self.parent_family_list[0][0] def set_lds_baptism(self,lds_ord): """ Sets the LDS Baptism ordinance. An ordinance can be removed by assigning to None. @param lds_ord: L{LdsOrd} to assign as the LDS Baptism ordinance. @type lds_ord: L{LdsOrd} """ self.lds_bapt = lds_ord def get_lds_baptism(self): """ Returns the LDS Baptism ordinance. @returns: returns the L{LdsOrd} instance assigned as the LDS Baptism ordinance, or None if no ordinance has been assigned. @rtype: L{LdsOrd} """ return self.lds_bapt def set_lds_endowment(self,lds_ord): """ Sets the LDS Endowment ordinance. An ordinance can be removed by assigning to None. @param lds_ord: L{LdsOrd} to assign as the LDS Endowment ordinance. @type lds_ord: L{LdsOrd} """ self.lds_endow = lds_ord def get_lds_endowment(self): """ Returns the LDS Endowment ordinance. @returns: returns the L{LdsOrd} instance assigned as the LDS Endowment ordinance, or None if no ordinance has been assigned. @rtype: L{LdsOrd} """ return self.lds_endow def set_lds_sealing(self,lds_ord): """ Sets the LDS Sealing ordinance. An ordinance can be removed by assigning to None. @param lds_ord: L{LdsOrd} to assign as the LDS Sealing ordinance. @type lds_ord: L{LdsOrd} """ self.lds_seal = lds_ord def get_lds_sealing(self): """ Returns the LDS Sealing ordinance. @returns: returns the L{LdsOrd} instance assigned as the LDS Sealing ordinance, or None if no ordinance has been assigned. @rtype: L{LdsOrd} """ return self.lds_seal class Family(PrimaryObject,SourceNote): """ GRAMPS Family record. Represents a family unit, which defines the relationship between people. This can consist of a single person and a set of children, or two people with a defined relationship and an optional set of children. The relationship between people may be either opposite sex or same sex. """ MARRIED = 0 UNMARRIED = 1 CIVIL_UNION = 2 UNKNOWN = 3 OTHER = 4 def __init__(self): """creates a new Family instance""" PrimaryObject.__init__(self) SourceNote.__init__(self) self.father_handle = None self.mother_handle = None self.child_list = [] self.type = Family.MARRIED self.event_list = [] self.media_list = [] self.attribute_list = [] self.lds_seal = None self.complete = 0 def serialize(self): """ Converts the data held in the event to a Python tuple that represents all the data elements. This method is used to convert the object into a form that can easily be saved to a database. These elements may be primative Python types (string, integers), complex Python types (lists or tuples, or Python objects. If the target database cannot handle complex types (such as objectes or lists), the database is responsible for converting the data into a form that it can use. """ return (self.handle, self.gramps_id, self.father_handle, self.mother_handle, self.child_list, self.type, self.event_list, self.media_list, self.attribute_list, self.lds_seal, self.complete, self.source_list, self.note, self.change) def unserialize(self, data): """ Converts the data held in a tuple created by the serialize method back into the data in an Event structure. """ (self.handle, self.gramps_id, self.father_handle, self.mother_handle, self.child_list, self.type, self.event_list, self.media_list, self.attribute_list, self.lds_seal, self.complete, self.source_list, self.note, self.change) = data def set_complete_flag(self,val): self.complete = val def get_complete_flag(self): return self.complete def set_lds_sealing(self,ord): self.lds_seal = ord def get_lds_sealing(self): return self.lds_seal def add_attribute(self,attribute) : """adds an Attribute instance to the attribute list""" self.attribute_list.append(attribute) def remove_attribute(self,attribute): """removes the specified Attribute instance from the attribute list""" if attribute in self.attribute_list: self.attribute_list.remove(attribute) def get_attribute_list(self) : """returns the attribute list""" return self.attribute_list def set_attribute_list(self,list) : """sets the attribute list to the specified list""" self.attribute_list = list def set_relationship(self,type): """assigns a string indicating the relationship between the father and the mother""" self.type = type def get_relationship(self): """returns a string indicating the relationship between the father and the mother""" return self.type def set_father_handle(self,person_handle): """sets the father of the Family to the specfied Person""" self.father_handle = person_handle def get_father_handle(self): """returns the father of the Family""" return self.father_handle def set_mother_handle(self,person): """sets the mother of the Family to the specfied Person""" self.mother_handle = person def get_mother_handle(self): """returns the mother of the Family""" return self.mother_handle def add_child_handle(self,person): """adds the specfied Person as a child of the Family, adding it to the child list""" if person not in self.child_list: self.child_list.append(person) def remove_child_handle(self,person): """removes the specified Person from the child list""" if person in self.child_list: self.child_list.remove(person) def get_child_handle_list(self): """returns the list of children""" return self.child_list def set_child_handle_list(self, list): """sets the list of children""" self.child_list = list[:] def add_event_handle(self,event_handle): """adds an L{Event} to the event list""" self.event_list.append(event_handle) def get_event_list(self) : """returns the list of L{Event} instances""" return self.event_list def set_event_list(self,list) : """sets the event list to the passed list""" self.event_list = list def add_media_reference(self,media_id): """Adds a MediaObject object to the Family instance's image list""" self.media_list.append(media_id) def get_media_list(self): """Returns the list of MediaObject objects""" return self.media_list def set_media_list(self,list): """Sets the list of MediaObject objects""" self.media_list = list class Event(PrimaryObject,DataObj): """ The Event record is used to store information about some type of action that occurred at a particular place at a particular time, such as a birth, death, or marriage. """ NAME = 0 ID = 1 def __init__(self,source=None): """ creates a new Event instance, copying from the source if present @param source: An event used to initialize the new event @type source: Event """ PrimaryObject.__init__(self,source) DataObj.__init__(self,source) if source: self.place = source.place self.date = Date.Date(source.date) self.description = source.description self.name = source.name self.cause = source.cause self.media_list = [MediaRef(media_id) for media_id in source.media_list] if source.witness != None: self.witness = source.witness[:] else: self.witness = None else: self.place = u'' self.date = None self.description = "" self.name = "" self.cause = "" self.witness = None self.media_list = [] def serialize(self): """ Converts the data held in the event to a Python tuple that represents all the data elements. This method is used to convert the object into a form that can easily be saved to a database. These elements may be primative Python types (string, integers), complex Python types (lists or tuples, or Python objects. If the target database cannot handle complex types (such as objectes or lists), the database is responsible for converting the data into a form that it can use. """ return (self.handle, self.gramps_id, self.name, self.date, self.description, self.place, self.cause, self.private, self.source_list, self.note, self.witness, self.media_list, self.change) def unserialize(self,data): """ Converts the data held in a tuple created by the serialize method back into the data in an Event structure. """ (self.handle, self.gramps_id, self.name, self.date, self.description, self.place, self.cause, self.private, self.source_list, self.note, self.witness, self.media_list, self.change) = data def add_media_reference(self,media_id): """Adds a MediaObject object to the Event object's image list""" self.media_list.append(media_id) def get_media_list(self): """Returns the list of MediaObject objects""" return self.media_list def set_media_list(self,mlist): """Sets the list of MediaObject objects""" self.media_list = mlist def get_witness_list(self): """Returns the list of Witness objects associated with the Event""" return self.witness def set_witness_list(self,list): """ Sets the Witness list to a copy of the list passed to the method. """ if list: self.witness = list[:] else: self.witness = None def add_witness(self,value): """ Adds a Witness object to the current witness list. """ if self.witness: self.witness.append(value) else: self.witness = [value] def is_empty(self): date = self.get_date_object() place = self.get_place_handle() description = self.description cause = self.cause name = self.name return (not name or name == "Birth" or name == "Death") and \ date.is_empty() and not place and not description and not cause def set(self,name,date,place,description): """sets the name, date, place, and description of an Event instance""" self.name = name self.place = place self.description = description self.set_date(date) def are_equal(self,other): """returns 1 if the specified event is the same as the instance""" if other == None: other = Event (None) if (self.name != other.name or self.place != other.place or self.description != other.description or self.cause != other.cause or self.private != other.private or (not self.get_date_object().is_equal(other.get_date_object())) or len(self.get_source_references()) != len(other.get_source_references())): return 0 index = 0 olist = other.get_source_references() for a in self.get_source_references(): if not a.are_equal(olist[index]): return 0 index = index + 1 witness_list = self.get_witness_list() other_list = other.get_witness_list() if (not witness_list) and (not other_list): return 1 elif not (witness_list and other_list): return 0 for a in witness_list: if a in other_list: other_list.remove(a) else: return 0 if other_list: return 0 return 1 def set_name(self,name): """sets the name of the Event""" self.name = name def get_name(self): """returns the name of the Event""" return self.name def set_place_handle(self,place): """sets the Place instance of the Event""" self.place = place def get_place_handle(self): """returns the Place instance of the Event""" return self.place def set_cause(self,cause): """sets the cause of the Event""" self.cause = cause def get_cause(self): """returns the cause of the Event""" return self.cause def set_description(self,description): """sets the description of the Event instance""" self.description = description def get_description(self) : """returns the description of the Event instance""" return self.description def set_date(self, date) : """attempts to sets the date of the Event instance""" self.date = parser.parse(date) def get_date(self) : """returns a string representation of the date of the Event instance""" if self.date: return display.display(self.date) return u"" def get_preferred_date(self) : """returns a string representation of the date of the Event instance""" return self.get_date() def get_quote_date(self) : """returns a string representation of the date of the Event instance, enclosing the results in quotes if it is not a valid date""" if self.date: return display.quote_display(self.date) return u"" def get_date_object(self): """returns the Date object associated with the Event""" if not self.date: self.date = Date.Date() return self.date def set_date_object(self,date): """sets the Date object associated with the Event""" self.date = date class Place(PrimaryObject,SourceNote): """Contains information related to a place, including multiple address information (since place names can change with time), longitude, latitude, a collection of images and URLs, a note and a source""" def __init__(self,source=None): """Creates a new Place object. source - Object to copy. If none supplied, create an empty place object""" PrimaryObject.__init__(self,source) SourceNote.__init__(self,source) if source: self.long = source.long self.lat = source.lat self.title = source.title self.main_loc = Location(source.main_loc) self.alt_loc = [] for loc in source.alt_loc: self.alt_loc = Location(loc) self.urls = [] for u in source.urls: self.urls.append(Url(u)) self.media_list = [] for media_id in source.media_list: self.media_list.append(MediaRef(media_id)) else: self.long = "" self.lat = "" self.title = "" self.main_loc = None self.alt_loc = [] self.urls = [] self.media_list = [] def serialize(self): """ Converts the data held in the event to a Python tuple that represents all the data elements. This method is used to convert the object into a form that can easily be saved to a database. These elements may be primative Python types (string, integers), complex Python types (lists or tuples, or Python objects. If the target database cannot handle complex types (such as objectes or lists), the database is responsible for converting the data into a form that it can use. """ return (self.handle, self.gramps_id, self.title, self.long, self.lat, self.main_loc, self.alt_loc, self.urls, self.media_list, self.source_list, self.note, self.change) def unserialize(self,data): """ Converts the data held in a tuple created by the serialize method back into the data in an Event structure. """ (self.handle, self.gramps_id, self.title, self.long, self.lat, self.main_loc, self.alt_loc, self.urls, self.media_list, self.source_list, self.note, self.change) = data def get_url_list(self): """Return the list of URLs""" return self.urls def set_url_list(self,list): """Replace the current URL list with the new one""" self.urls = list def add_url(self,url): """Add a URL to the URL list""" self.urls.append(url) def set_title(self,name): """Sets the title of the place object""" self.title = name def get_title(self): """Returns the title of the place object""" return self.title def set_longitude(self,long): """Sets the longitude of the place""" self.long = long def get_longitude(self): """Returns the longitude of the place""" return self.long def set_latitude(self,long): """Sets the latitude of the place""" self.lat = long def get_latitude(self): """Returns the latitude of the place""" return self.lat def get_main_location(self): """Returns the Location object representing the primary information""" if not self.main_loc: self.main_loc = Location() return self.main_loc def set_main_location(self,loc): """Assigns the main location to the Location object passed""" self.main_loc = loc def get_alternate_locations(self): """Returns a list of alternate location information objects""" return self.alt_loc def set_alternate_locations(self,list): """Replaces the current alternate location list with the new one""" self.alt_loc = list def add_alternate_locations(self,loc): """Adds a Location to the alternate location list""" if loc not in self.alt_loc: self.alt_loc.append(loc) def add_media_reference(self,media_id): """Adds a MediaObject object to the place object's image list""" self.media_list.append(media_id) def get_media_list(self): """Returns the list of MediaObject objects""" return self.media_list def set_media_list(self,list): """Sets the list of MediaObject objects""" self.media_list = list def get_display_info(self): """Gets the display information associated with the object. This includes the information that is used for display and for sorting. Returns a list consisting of 13 strings. These are: Place Title, Place ID, Main Location Parish, Main Location County, Main Location City, Main Location State/Province, Main Location Country, upper case Place Title, upper case Parish, upper case city, upper case county, upper case state, upper case country""" if self.main_loc: return [self.title,self.gramps_id,self.main_loc.parish,self.main_loc.city, self.main_loc.county,self.main_loc.state,self.main_loc.country, self.title.upper(), self.main_loc.parish.upper(), self.main_loc.city.upper(), self.main_loc.county.upper(), self.main_loc.state.upper(), self.main_loc.country.upper()] else: return [self.title,self.gramps_id,'','','','','', self.title.upper(), '','','','',''] class MediaObject(PrimaryObject,SourceNote): """Containter for information about an image file, including location, description and privacy""" def __init__(self,source=None): """Create a new MediaObject object, copying from the source if provided""" PrimaryObject.__init__(self,source) SourceNote.__init__(self,source) self.attrlist = [] if source: self.path = source.path self.mime = source.mime self.desc = source.desc self.thumb = source.thumb self.date = Date.Date(source.date) self.place = source.place for attr in source.attrlist: self.attrlist.append(Attribute(attr)) else: self.path = "" self.mime = "" self.desc = "" self.date = None self.place = u"" self.thumb = None def serialize(self): """ Converts the data held in the event to a Python tuple that represents all the data elements. This method is used to convert the object into a form that can easily be saved to a database. These elements may be primative Python types (string, integers), complex Python types (lists or tuples, or Python objects. If the target database cannot handle complex types (such as objectes or lists), the database is responsible for converting the data into a form that it can use. """ return (self.handle, self.gramps_id, self.path, self.mime, self.desc, self.attrlist, self.source_list, self.note, self.change, self.date, self.place) def unserialize(self,data): """ Converts the data held in a tuple created by the serialize method back into the data in an Event structure. """ (self.handle, self.gramps_id, self.path, self.mime, self.desc, self.attrlist, self.source_list, self.note, self.change, self.date, self.place) = data def set_place_handle(self,place): """sets the Place instance of the Event""" self.place = place def get_place_handle(self): """returns the Place instance of the Event""" return self.place def get_date(self) : """returns a string representation of the date of the Event instance""" if self.date: return display.display(self.date) return u"" def get_date_object(self): """returns the Date object associated with the Event""" if not self.date: self.date = Date.Date() return self.date def set_date(self, date) : """attempts to sets the date of the Event instance""" self.date = parser.parse(date) def set_date_object(self,date): """sets the Date object associated with the Event""" self.date = date def set_mime_type(self,type): self.mime = type def get_mime_type(self): return self.mime def set_path(self,path): """set the file path to the passed path""" self.path = os.path.normpath(path) def get_path(self): """return the file path""" return self.path def set_description(self,text): """sets the description of the image""" self.desc = text def get_description(self): """returns the description of the image""" return self.desc def add_attribute(self,attr): """Adds a propery to the MediaObject object. This is not used by gramps, but provides a means for XML users to attach other properties to the image""" self.attrlist.append(attr) def get_attribute_list(self): """returns the property list associated with the image""" return self.attrlist def set_attribute_list(self,list): self.attrlist = list class Source(PrimaryObject): """A record of a source of information""" def __init__(self): """creates a new Source instance""" PrimaryObject.__init__(self) self.title = "" self.author = "" self.pubinfo = "" self.note = Note() self.media_list = [] self.datamap = {} self.abbrev = "" def serialize(self): return (self.handle, self.gramps_id, unicode(self.title), unicode(self.author), unicode(self.pubinfo), self.note, self.media_list, unicode(self.abbrev), self.change,self.datamap) def unserialize(self,data): """ Converts the data held in a tuple created by the serialize method back into the data in an Event structure. """ (self.handle, self.gramps_id, self.title, self.author, self.pubinfo, self.note, self.media_list, self.abbrev, self.change, self.datamap) = data def get_display_info(self): return [self.title,self.gramps_id,self.author, self.title.upper(),self.author.upper()] def add_media_reference(self,media_id): """Adds a MediaObject object to the Source instance's image list""" self.media_list.append(media_id) def get_media_list(self): """Returns the list of MediaObject objects""" return self.media_list def set_media_list(self,list): """Sets the list of MediaObject objects""" self.media_list = list def get_data_map(self): """Returns the data map of attributes for the source""" return self.datamap def set_data_map(self,datamap): """Sets the data map of attributes for the source""" self.datamap = datamap def set_data_item(self,key,value): """Sets the particular data item in the attribute data map""" self.datamap[key] = value def set_title(self,title): """sets the title of the Source""" self.title = title def get_title(self): """returns the title of the Source""" return self.title def set_note(self,text): """sets the text of the note attached to the Source""" self.note.set(text) def get_note(self): """returns the text of the note attached to the Source""" return self.note.get() def set_note_format(self,val): """Set the note's format to the given value""" self.note.set_format(val) def get_note_format(self): """Return the current note's format""" return self.note.get_format() def set_note_object(self,obj): """sets the Note instance attached to the Source""" self.note = obj def get_note_object(self): """returns the Note instance attached to the Source""" return self.note def unique_note(self): """Creates a unique instance of the current note""" self.note = Note(self.note.get()) def set_author(self,author): """sets the author of the Source""" self.author = author def get_author(self): """returns the author of the Source""" return self.author def set_publication_info(self,text): """sets the publication information of the Source""" self.pubinfo = text def get_publication_info(self): """returns the publication information of the Source""" return self.pubinfo def set_abbreviation(self,abbrev): """sets the title abbreviation of the Source""" self.abbrev = abbrev def get_abbreviation(self): """returns the title abbreviation of the Source""" return self.abbrev class LdsOrd(SourceNote): """ Class that contains information about LDS Ordinances. LDS ordinances are similar to events, but have very specific additional information related to data collected by the Church of Jesus Christ of Latter Day Saints (Morman church). The LDS church is the largest source of genealogical information in the United States. """ def __init__(self,source=None): """Creates a LDS Ordinance instance""" SourceNote.__init__(self,source) if source: self.famc = source.famc self.date = Date.Date(source.date) self.temple = source.temple self.status = source.status self.place = source.place else: self.famc = None self.date = None self.temple = "" self.status = 0 self.place = None def set_place_handle(self,place): """sets the Place database handle of the ordinance""" self.place = place def get_place_handle(self): """returns the Place handle of the ordinance""" return self.place def set_family_handle(self,family): """Sets the Family database handle associated with the LDS ordinance""" self.famc = family def get_family_handle(self): """Gets the Family database handle associated with the LDS ordinance""" return self.famc def set_status(self,val): """ Sets the status of the LDS ordinance. The status is a text string that matches a predefined set of strings.""" self.status = val def get_status(self): """Gets the status of the LDS ordinance""" return self.status def set_date(self, date) : """attempts to sets the date of the ordinance""" if not self.date: self.date = Date.Date() parser.set_date(self.date,date) def get_date(self) : """returns a string representation of the date of the ordinance""" if self.date: return display.display(self.date) return u"" def get_date_object(self): """returns the Date object associated with the ordinance""" if not self.date: self.date = Date.Date() return self.date def set_date_object(self,date): """sets the Date object associated with the ordinance""" self.date = date def set_temple(self,temple): """Sets the temple assocated with the ordinance""" self.temple = temple def get_temple(self): """Gets the temple assocated with the ordinance""" return self.temple def is_empty(self): """Returns 1 if the ordidance is actually empty""" if (self.famc or (self.date and not self.date.is_empty()) or self.temple or self.status or self.place): return 0 else: return 1 def are_equal(self,other): """returns 1 if the specified ordinance is the same as the instance""" if other == None: return self.is_empty() if (self.famc != other.famc or self.place != other.place or self.status != other.status or self.temple != other.temple or not self.get_date_object().is_equal(other.get_date_object()) or len(self.get_source_references()) != len(other.get_source_references())): return 0 index = 0 olist = other.get_source_references() for a in self.get_source_references(): if not a.are_equal(olist[index]): return 0 index = index + 1 return 1 class Researcher: """Contains the information about the owner of the database""" def __init__(self): """Initializes the Researcher object""" self.name = "" self.addr = "" self.city = "" self.state = "" self.country = "" self.postal = "" self.phone = "" self.email = "" def get_name(self): """returns the database owner's name""" return self.name def get_address(self): """returns the database owner's address""" return self.addr def get_city(self): """returns the database owner's city""" return self.city def get_state(self): """returns the database owner's state""" return self.state def get_country(self): """returns the database owner's country""" return self.country def get_postal_code(self): """returns the database owner's postal code""" return self.postal def get_phone(self): """returns the database owner's phone number""" return self.phone def get_email(self): """returns the database owner's email""" return self.email def set(self,name,addr,city,state,country,postal,phone,email): """sets the information about the database owner""" if name: self.name = name.strip() if addr: self.addr = addr.strip() if city: self.city = city.strip() if state: self.state = state.strip() if country: self.country = country.strip() if postal: self.postal = postal.strip() if phone: self.phone = phone.strip() if email: self.email = email.strip() class Location: """Provides information about a place, including city, county, state, and country. Multiple Location objects can represent the same place, since names of citys, countys, states, and even countries can change with time""" def __init__(self,source=None): """creates a Location object, copying from the source object if it exists""" if source: self.city = source.city self.parish = source.parish self.county = source.county self.state = source.state self.country = source.country self.postal = source.postal self.phone = source.phone else: self.city = "" self.parish = "" self.county = "" self.state = "" self.country = "" self.postal = "" self.phone = "" def is_empty(self): return not self.city and not self.county and not self.state and \ not self.country and not self.postal and not self.phone def set_city(self,data): """sets the city name of the Location object""" self.city = data def get_postal_code(self): """returns the postal code of the Location object""" return self.postal def set_postal_code(self,data): """sets the postal code of the Location object""" self.postal = data def get_phone(self): """returns the phone number of the Location object""" return self.phone def set_phone(self,data): """sets the phone number of the Location object""" self.phone = data def get_city(self): """returns the city name of the Location object""" return self.city def set_parish(self,data): """sets the religious parish name""" self.parish = data def get_parish(self): """gets the religious parish name""" return self.parish def set_county(self,data): """sets the county name of the Location object""" self.county = data def get_county(self): """returns the county name of the Location object""" return self.county def set_state(self,data): """sets the state name of the Location object""" self.state = data def get_state(self): """returns the state name of the Location object""" return self.state def set_country(self,data): """sets the country name of the Location object""" self.country = data def get_country(self): """returns the country name of the Location object""" return self.country class Note: """Provides general text information""" def __init__(self,text = ""): """create a new Note object from the passed string""" self.text = text self.format = 0 def set(self,text): """set the note contents to the passed string""" self.text = text def get(self): """return the note contents""" return self.text def append(self,text): """adds the text to the note's contents""" self.text = self.text + text def set_format(self,format): """set the format to the passed value""" self.format = format def get_format(self): """return the note's format""" return self.format class MediaRef(SourceNote): """Media reference class""" def __init__(self,source=None): SourceNote.__init__(self,source) self.attrlist = [] if source: self.private = source.private self.ref = source.ref self.note = Note(source.note) for attr in source.attrlist: self.attrlist.append(Attribute(attr)) self.rect = source.rect else: self.private = 0 self.ref = None self.note = None self.rect = None def set_privacy(self,val): """Sets or clears the privacy flag of the data""" self.private = val def get_privacy(self): """Returns the privacy level of the data""" return self.private def set_rectangle(self,coord): """Sets subection of an image""" self.rect = coord def get_rectangle(self): """Returns the subsection of an image""" return self.rect def set_reference_handle(self,obj_id): self.ref = obj_id def get_reference_handle(self): return self.ref def add_attribute(self,attr): """Adds a propery to the MediaObject object. This is not used by gramps, but provides a means for XML users to attach other properties to the image""" self.attrlist.append(attr) def get_attribute_list(self): """returns the property list associated with the image""" return self.attrlist def set_attribute_list(self,list): """sets the property list associated with the image""" self.attrlist = list class Attribute(DataObj): """Provides a simple key/value pair for describing properties. Used by the Person and Family objects to store descriptive information.""" def __init__(self,source=None): """creates a new Attribute object, copying from the source if provided""" DataObj.__init__(self,source) if source: self.type = source.type self.value = source.value else: self.type = "" self.value = "" def set_type(self,val): """sets the type (or key) of the Attribute instance""" self.type = val def get_type(self): """returns the type (or key) or the Attribute instance""" return self.type def set_value(self,val): """sets the value of the Attribute instance""" self.value = val def get_value(self): """returns the value of the Attribute instance""" return self.value class Address(DataObj): """Provides address information for a person""" def __init__(self,source=None): """Creates a new Address instance, copying from the source if provided""" DataObj.__init__(self,source) if source: self.street = source.street self.city = source.city self.state = source.state self.country = source.country self.postal = source.postal self.date = Date.Date(source.date) self.phone = source.phone else: self.street = "" self.city = "" self.state = "" self.country = "" self.postal = "" self.date = Date.Date() self.phone = "" def set_date(self,text): """attempts to sets the date that the person lived at the address from the passed string""" self.date = parser.parse(text) def get_date(self): """returns a string representation of the date that the person lived at the address""" if self.date: return display.display(self.date) return u"" def get_preferred_date(self): """returns a string representation of the date that the person lived at the address""" return self.date.get_preferred_date() def get_date_object(self): """returns the Date object associated with the Address""" return self.date def set_date_object(self,obj): """sets the Date object associated with the Address""" self.date = obj def set_street(self,val): """sets the street portion of the Address""" self.street = val def get_street(self): """returns the street portion of the Address""" return self.street def set_phone(self,val): """sets the phone number portion of the Address""" self.phone = val def get_phone(self): """returns the phone number portion of the Address""" return self.phone def set_city(self,val): """sets the city portion of the Address""" self.city = val def get_city(self): """returns the city portion of the Address""" return self.city def set_state(self,val): """sets the state portion of the Address""" self.state = val def get_state(self): """returns the state portion of the Address""" return self.state def set_country(self,val): """sets the country portion of the Address""" self.country = val def get_country(self): """returns the country portion of the Address""" return self.country def set_postal_code(self,val): """sets the postal code of the Address""" self.postal = val def get_postal_code(self): """returns the postal code of the Address""" return self.postal class Name(DataObj): """Provides name information about a person. A person may have more that one name throughout his or her life.""" DEF = 0 # locale default LNFN = 1 # last name, first name FNLN = 2 # first name, last name def __init__(self,source=None): """creates a new Name instance, copying from the source if provided""" DataObj.__init__(self,source) if source: self.first_name = source.first_name self.surname = source.surname self.suffix = source.suffix self.title = source.title self.type = source.type self.prefix = source.prefix self.patronymic = source.patronymic self.sname = source.sname self.group_as = source.group_as self.sort_as = source.sort_as self.display_as = source.display_as else: self.first_name = "" self.surname = "" self.suffix = "" self.title = "" self.type = "Birth Name" self.prefix = "" self.patronymic = "" self.sname = '@' self.group_as = "" self.sort_as = self.DEF self.display_as = self.DEF def set_group_as(self,name): """ Sets the grouping name for a person. Normally, this is the person's surname. However, some locales group equivalent names (e.g. Ivanova and Ivanov in Russian are usually considered equivalent. """ if name == self.surname: self.group_as = "" else: self.group_as = name def get_group_as(self): """ Returns the grouping name, which is used to group equivalent surnames. """ return self.group_as def get_group_name(self): """ Returns the grouping name, which is used to group equivalent surnames. """ if self.group_as: return self.group_as else: return self.surname def set_sort_as(self,value): """ Specifies the sorting method for the specified name. Typically the locale's default should be used. However, there may be names where a specific sorting structure is desired for a name. """ self.sort_as = value def get_sort_as(self): """ Returns the selected sorting method for the name. The options are DEF (default for the current locale), LNFN (last name, first name), or FNLN (first name, last name). """ return self.sort_as def set_display_as(self,value): """ Specifies the display format for the specified name. Typically the locale's default should be used. However, there may be names where a specific display format is desired for a name. """ self.display_as = value def get_display_as(self): """ Returns the selected display format for the name. The options are DEF (default for the current locale), LNFN (last name, first name), or FNLN (first name, last name). """ return self.display_as def get_surname_prefix(self): """ Returns the prefix (or article) of a surname. The prefix is not used for sorting or grouping. """ return self.prefix def set_surname_prefix(self,val): """ Sets the prefix (or article) of a surname. Examples of articles would be 'de' or 'van'. """ self.prefix = val def set_type(self,type): """sets the type of the Name instance""" self.type = type def get_type(self): """returns the type of the Name instance""" return self.type def build_sort_name(self): if self.surname: self.sname = "%-25s%-30s%s" % (self.surname,self.first_name,self.suffix) else: self.sname = "@" def set_first_name(self,name): """sets the given name for the Name instance""" self.first_name = name self.build_sort_name() def set_patronymic(self,name): """sets the patronymic name for the Name instance""" self.patronymic = name self.build_sort_name() def set_surname(self,name): """sets the surname (or last name) for the Name instance""" self.surname = name self.build_sort_name() def set_suffix(self,name): """sets the suffix (such as Jr., III, etc.) for the Name instance""" self.suffix = name self.build_sort_name() def get_sort_name(self): return self.sname def get_first_name(self): """returns the given name for the Name instance""" return self.first_name def get_patronymic(self): """returns the patronymic name for the Name instance""" return self.patronymic def get_surname(self): """returns the surname (or last name) for the Name instance""" return self.surname def get_upper_surname(self): """returns the surname (or last name) for the Name instance""" return self.surname.upper() def get_suffix(self): """returns the suffix for the Name instance""" return self.suffix def set_title(self,title): """sets the title (Dr., Reverand, Captain) for the Name instance""" self.title = title def get_title(self): """returns the title for the Name instance""" return self.title def get_name(self): """returns a name string built from the components of the Name instance, in the form of surname, Firstname""" if self.patronymic: first = "%s %s" % (self.first_name, self.patronymic) else: first = self.first_name if self.suffix: if self.prefix: return "%s %s, %s %s" % (self.prefix, self.surname, first, self.suffix) else: return "%s, %s %s" % (self.surname, first, self.suffix) else: if self.prefix: return "%s %s, %s" % (self.prefix,self.surname, first) else: return "%s, %s" % (self.surname, first) def get_upper_name(self): """returns a name string built from the components of the Name instance, in the form of surname, Firstname""" if self.patronymic: first = "%s %s" % (self.first_name, self.patronymic) else: first = self.first_name if self.suffix: if self.prefix: return "%s %s, %s %s" % (self.prefix.upper(), self.surname.upper(), first, self.suffix) else: return "%s, %s %s" % (self.surname.upper(), first, self.suffix) else: if self.prefix: return "%s %s, %s" % (self.prefix.upper(), self.surname.upper(), first) else: return "%s, %s" % (self.surname.upper(), first) def get_regular_name(self): """returns a name string built from the components of the Name instance, in the form of Firstname surname""" if self.patronymic: first = "%s %s" % (self.first_name, self.patronymic) else: first = self.first_name if (self.suffix == ""): if self.prefix: return "%s %s %s" % (first, self.prefix, self.surname) else: return "%s %s" % (first, self.surname) else: if self.prefix: return "%s %s %s, %s" % (first, self.prefix, self.surname, self.suffix) else: return "%s %s, %s" % (first, self.surname, self.suffix) def get_regular_upper_name(self): """returns a name string built from the components of the Name instance, in the form of Firstname surname""" if self.patronymic: first = "%s %s" % (self.first_name, self.patronymic) else: first = self.first_name if (self.suffix == ""): if self.prefix: return "%s %s %s" % (first, self.prefix.upper(), self.surname.upper()) else: return "%s %s" % (first, self.surname.upper()) else: if self.prefix: return "%s %s %s, %s" % (first, self.prefix.upper(), self.surname.upper(), self.suffix) else: return "%s %s, %s" % (first, self.surname.upper(), self.suffix) def is_equal(self,other): """ compares to names to see if they are equal, return 0 if they are not """ if self.first_name != other.first_name: return False if self.surname != other.surname: return False if self.patronymic != other.patronymic: return False if self.prefix != other.prefix: return False if self.suffix != other.suffix: return False if self.title != other.title: return False if self.type != other.type: return False if self.private != other.private: return False if self.get_note() != other.get_note(): return False if len(self.get_source_references()) != len(other.get_source_references()): return False index = 0 olist = other.get_source_references() for a in self.get_source_references(): if not a.are_equal(olist[index]): return True index += 1 return True class Url: """Contains information related to internet Uniform Resource Locators, allowing gramps to store information about internet resources""" def __init__(self,source=None): """creates a new URL instance, copying from the source if present""" if source: self.path = source.path self.desc = source.desc self.private = source.private else: self.path = "" self.desc = "" self.private = 0 def set_privacy(self,val): """sets the privacy flag for the URL instance""" self.private = val def get_privacy(self): """returns the privacy flag for the URL instance""" return self.private def set_path(self,path): """sets the URL path""" self.path = path def get_path(self): """returns the URL path""" return self.path def set_description(self,description): """sets the description of the URL""" self.desc = description def get_description(self): """returns the description of the URL""" return self.desc def are_equal(self,other): """returns 1 if the specified URL is the same as the instance""" if other == None: return 0 if self.path != other.path: return 0 if self.desc != other.desc: return 0 return 1 class Witness: """ The Witness class is used to represent a person who may or may not be in the database. If the person is in the database, the type will be Event.ID, and the value with be the database handle for the person. If the person is not in the database, the type will be Event.NAME, and the value will be a string representing the person's name. """ def __init__(self,type=Event.NAME,val="",comment=""): self.set_type(type) self.set_value(val) self.set_comment(comment) self.private = False def set_privacy(self,val): """Sets or clears the privacy flag of the data""" self.private = val def get_privacy(self): """Returns the privacy level of the data""" return self.private def set_type(self,type): self.type = type def get_type(self): return self.type def set_value(self,val): self.val = val def get_value(self): return self.val def set_comment(self,comment): self.comment = comment def get_comment(self): return self.comment class SourceRef: """Source reference, containing detailed information about how a referenced source relates to it""" def __init__(self,source=None): """creates a new SourceRef, copying from the source if present""" if source: self.confidence = source.confidence self.ref = source.ref self.page = source.page self.date = Date.Date(source.date) self.comments = Note(source.comments.get()) self.text = source.text self.private = source.private else: self.confidence = CONF_NORMAL self.ref = None self.page = "" self.date = Date.Date() self.comments = Note() self.text = "" self.private = False def set_privacy(self,val): """Sets or clears the privacy flag of the data""" self.private = val def get_privacy(self): """Returns the privacy level of the data""" return self.private def set_confidence_level(self,val): """Sets the confidence level""" self.confidence = val def get_confidence_level(self): """Returns the confidence level""" return self.confidence def set_base_handle(self,ref): """sets the Source instance to which the SourceRef refers""" self.ref = ref def get_base_handle(self): """returns the Source instance to which the SourceRef refers""" return self.ref def set_date(self,date): """sets the Date instance of the SourceRef""" self.date = date def get_date(self): """returns the Date instance of the SourceRef""" return self.date def set_page(self,page): """sets the page indicator of the SourceRef""" self.page = page def get_page(self): """gets the page indicator of the SourceRef""" return self.page def set_text(self,text): """sets the text related to the SourceRef""" self.text = text def get_text(self): """returns the text related to the SourceRef""" return self.text def set_note_object(self,note): """Change the Note instance to obj""" self.comments = note def set_comments(self,comments): """sets the comments about the SourceRef""" self.comments.set(comments) def get_comments(self): """returns the comments about the SourceRef""" return self.comments.get() def are_equal(self,other): """returns 1 if the passed SourceRef is equal to the current""" if self.ref and other.ref: if self.page != other.page: return 0 if self.date != other.date: return 0 if self.get_text() != other.get_text(): return 0 if self.get_comments() != other.get_comments(): return 0 if self.confidence != other.confidence: return 0 return 1 elif not self.ref and not other.ref: return 1 else: return 0 def unique_note(self): """Creates a unique instance of the current note""" self.comments = Note(self.comments.get()) class GenderStats: """ Class for keeping track of statistics related to Given Name vs. Gender. This allows the tracking of the liklihood of a person's given name indicating the gender of the person. """ def __init__ (self,stats={}): if stats == None: self.stats = {} else: self.stats = stats def save_stats(self): return self.stats def _get_key (self, person): name = person.get_primary_name().get_first_name() return self._get_key_from_name (name) def _get_key_from_name (self, name): return name.split (' ')[0].replace ('?', '') def name_stats (self, name): if self.stats.has_key (name): return self.stats[name] return (0, 0, 0) def count_person (self, person, db, undo = 0): # Let the Person do their own counting later person.db = db name = self._get_key (person) if not name: return gender = person.get_gender() (male, female, unknown) = self.name_stats (name) if not undo: increment = 1 else: increment = -1 if gender == Person.male: male += increment elif gender == Person.female: female += increment elif gender == Person.unknown: unknown += increment self.stats[name] = (male, female, unknown) return def uncount_person (self, person): return self.count_person (person, None, undo = 1) def guess_gender (self, name): name = self._get_key_from_name (name) if not name or not self.stats.has_key (name): return Person.unknown (male, female, unknown) = self.stats[name] if unknown == 0: if male and not female: return Person.male if female and not male: return Person.female if male > (2 * female): return Person.male if female > (2 * male): return Person.female return Person.unknown