# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 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: _ReadGedcom.py 8032 2007-02-03 17:11:05Z hippy $ """ Import from GEDCOM The GEDCOM file format is defined by the GEDCOM 5.5 Specfication, found at http://www.familysearch.org/GEDCOM/GEDCOM55.EXE The basic structure is a line with three attributes: Because of this structure, it does not lend itself to more traditional parsing techniques, such as LALR. The LEVEL token implies too much to be useful in this context. While this makes parsing more difficult, it does provide one very beneficial feature for GEDCOM: Error recoverability. GEDCOM is a poorly implemented standard, primarily because it is a poor standard to begin with. Most commercial applications that implement GEDCOM output add custom extensions, and feel free to violate the existing structure. If one were cynical, one might believe that the commercial programs were trying to make it difficult to transfer your data to another application. This parser takes a different approach to parsing a GEDCOM file. The first state, GedcomLex, reads lines from the file, and does some basic lexical analysis on each line (actually several lines, since it automatically combines CONT and CONC tagged lines). Each logical line returned to this parser contains: Level, Token, Token text, Data, and line number. The Data field is typically text, but in some cases, it may be a integer value representing an enumerated type or a GRAMPS object (in the case of dates). The parser works on the current level. Each context and level has a an associated table (dictionary) of functions indexed by the corresponding TOKEN. When a token is found, we index into the table to find the function associated with the token. If no token is found, a function that skips the line and all subordinate (lines with a higher number). If a function is found, then we call that function, which in turn processes the line, and all tokens at the lower level. For example: 1 BIRT 2 DATE 1 JAN 2000 2 UKNOWN TAG 3 NOTE DATA The function parsing the individual at level 1, would encounter the BIRT tag. It would look up the BIRT token in the table to see if a function as defined for this TOKEN, and pass control to this function. This function would then start parsing level 2. It would encounter the DATE tag, look up the corresponding function the in level 2 table, and pass control to its associated function. This function would terminate, and return control back to the level 2 parser, which would then encounter the "UKNOWN" tag. Since this is not a valid token, it would not be in the table, and a function that would skip all lines until the next level 2 token is found (in this case, skipping the "3 NOTE DATA" line. """ __revision__ = "$Revision: $" __author__ = "Don Allingham" #------------------------------------------------------------------------- # # standard python modules # #------------------------------------------------------------------------- import os import re import string import time from gettext import gettext as _ #------------------------------------------------------------------------ # # Set up logging # #------------------------------------------------------------------------ import logging log = logging.getLogger(".GedcomImport") #------------------------------------------------------------------------- # # GRAMPS modules # #------------------------------------------------------------------------- import Errors import RelLib import NameDisplay import Utils import Mime import LdsUtils from ansel_utf8 import ansel_to_utf8 from _GedcomInfo import * from _GedcomTokens import * from _GedcomLex import Reader import _GedcomUtils as GedcomUtils from GrampsDb._GrampsDbConst import EVENT_KEY from BasicUtils import UpdateCallback try: import Config DEFAULT_SOURCE = Config.get(Config.DEFAULT_SOURCE) except: log.warn("No Config module available using defaults.") DEFAULT_SOURCE = False #------------------------------------------------------------------------- # # Address/Place constants # #------------------------------------------------------------------------- ADDR_RE = re.compile('(.+)([\n\r]+)(.+)\s*,(.+)\s+(\d+)\s*(.*)') ADDR2_RE = re.compile('(.+)([\n\r]+)(.+)\s*,(.+)\s+(\d+)') ADDR3_RE = re.compile('(.+)([\n\r]+)(.+)\s*,(.+)') TRUNC_MSG = _("Your GEDCOM file is corrupted. " "It appears to have been truncated.") #------------------------------------------------------------------------- # # latin/utf8 conversions # #------------------------------------------------------------------------- def latin_to_utf8(msg): """ Converts a string from iso-8859-1 to unicode. If the string is already unicode, we do nothing. @param msg: string to convert @type level: str @return: Returns the string, converted to a unicode object @rtype: unicode """ if type(msg) == unicode: return msg else: return unicode(msg, 'iso-8859-1') def nocnv(msg): """ Null operation that makes sure that a unicode string remains a unicode string @param msg: unicode to convert @type level: unicode @return: Returns the string, converted to a unicode object @rtype: unicode """ return unicode(msg) #------------------------------------------------------------------------- # # constants # #------------------------------------------------------------------------- ANSEL = 1 UNICODE = 2 UPDATE = 25 TYPE_BIRTH = RelLib.ChildRefType() TYPE_ADOPT = RelLib.ChildRefType(RelLib.ChildRefType.ADOPTED) TYPE_FOSTER = RelLib.ChildRefType(RelLib.ChildRefType.FOSTER) RELATION_TYPES = (RelLib.ChildRefType.BIRTH, RelLib.ChildRefType.UNKNOWN, RelLib.ChildRefType.NONE, ) PEDIGREE_TYPES = { 'birth' : RelLib.ChildRefType(), 'natural': RelLib.ChildRefType(), 'step' : TYPE_ADOPT, 'adopted': TYPE_ADOPT, 'foster' : TYPE_FOSTER, } MIME_MAP = { 'jpeg' : 'image/jpeg', 'jpg' : 'image/jpeg', 'rtf' : 'text/rtf', 'pdf' : 'application/pdf', 'mpeg' : 'video/mpeg', 'mpg' : 'video/mpeg', 'gif' : 'image/gif', 'bmp' : 'image/x-ms-bmp', 'tiff' : 'image/tiff', 'aif' : 'audio/x-aiff', 'text' : 'text/plain', 'w8bn' : 'application/msword', 'wav' : 'audio/x-wav', 'mov' : 'video/quicktime', } EVENT_FAMILY_STR = _("%(event_name)s of %(family)s") EVENT_PERSON_STR = _("%(event_name)s of %(person)s") TRANS_TABLE = string.maketrans('', '') DEL_CHARS = TRANS_TABLE[0:8] + TRANS_TABLE[10:31] TRANS_TABLE2 = TRANS_TABLE[0:128] + ('?' * 128) #------------------------------------------------------------------------- # # GEDCOM events to GRAMPS events conversion # #------------------------------------------------------------------------- GED_2_GRAMPS = {} for _val in personalConstantEvents.keys(): _key = personalConstantEvents[_val] if _key != "": GED_2_GRAMPS[_key] = _val GED_2_FAMILY = {} for _val in familyConstantEvents.keys(): _key = familyConstantEvents[_val] if _key != "": GED_2_FAMILY[_key] = _val GED_2_FAMILY_CUSTOM = {} #------------------------------------------------------------------------- # # regular expressions # #------------------------------------------------------------------------- INT_RE = re.compile(r"\s*(\d+)\s*$") NAME_RE = re.compile(r"/?([^/]*)(/([^/]*)(/([^/]*))?)?") SURNAME_RE = re.compile(r"/([^/]*)/([^/]*)") NOTE_RE = re.compile(r"\s*\d+\s+\@(\S+)\@\s+NOTE(.*)$") CONT_RE = re.compile(r"\s*\d+\s+CONT\s?(.*)$") CONC_RE = re.compile(r"\s*\d+\s+CONC\s?(.*)$") PERSON_RE = re.compile(r"\s*\d+\s+\@(\S+)\@\s+INDI(.*)$") #------------------------------------------------------------------------- # # CurrentState # #------------------------------------------------------------------------- class CurrentState: """ Keeps track of the current state variables """ def __init__(self, person=None, level=0, event=None, event_ref=None): """ Initializes the object """ self.note = "" self.name_cnt = 0 self.person = person self.level = level self.event = event self.event_ref = event_ref self.source_ref = None def __getattr__(self, name): """ Returns the value associated with the specified attribute. """ return self.__dict__.get(name) def __setattr__(self, name, value): """ Sets the value associated with the specified attribute. """ self.__dict__[name] = value def add_to_note(self, text): self.note += text def get_text(self): return self.note #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- class NoteParser: def __init__(self, ifile, broken, override): if override: if override == 1: self.cnv = ansel_to_utf8 elif override == 2: self.cnv = latin_to_utf8 else: self.cnv = nocnv else: for index in range(50): line = ifile.readline().split() if len(line) > 2 and line[1] == 'CHAR': if line[2] == "ANSEL": self.cnv = ansel_to_utf8 elif line[2] in ["UNICODE","UTF-8","UTF8"]: self.cnv = nocnv else: self.cnv = latin_to_utf8 self.name_map = {} self.count = 0 self.person_count = 0 self.trans = None self.groups = None ifile.seek(0) innote = False for line in ifile: try: text = line.translate(TRANS_TABLE, DEL_CHARS) except: text = line try: text = self.cnv(text) except: text = text.translate(TRANS_TABLE2) self.count += 1 if innote: match = CONT_RE.match(text) if match: noteobj.append("\n" + match.groups()[0]) continue match = CONC_RE.match(text) if match: if broken: noteobj.append(" " + match.groups()[0]) else: noteobj.append(match.groups()[0]) continue # Here we have finished parsing CONT/CONC tags for the NOTE # and ignored the rest of the tags (SOUR,CHAN,REFN,RIN). innote = False match = NOTE_RE.match(text) if match: data = match.groups()[0] noteobj = RelLib.Note() self.name_map["@%s@" % data] = noteobj noteobj.append(match.groups()[1]) innote = True elif PERSON_RE.match(line): self.person_count += 1 def get_map(self): return self.name_map def get_lines(self): return self.count def get_persons(self): return self.person_count #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- class GedcomParser(UpdateCallback): SyntaxError = "Syntax Error" BadFile = "Not a GEDCOM file" def __init__(self, dbase, ifile, filename, callback, code_set, note_map, lines, people): UpdateCallback.__init__(self, callback) self.set_total(lines) self.repo2id = {} self.maxpeople = people self.db = dbase self.emapper = GedcomUtils.IdFinder(dbase.get_gramps_ids(EVENT_KEY), dbase.eprefix) self.fam_count = 0 self.indi_count = 0 self.repo_count = 0 self.source_count = 0 self.place_parser = GedcomUtils.PlaceParser() self.debug = False self.person = None self.inline_srcs = {} self.media_map = {} self.fmap = {} self.smap = {} self.note_map = note_map self.refn = {} self.added = set() self.gedmap = GedcomInfoDB() self.gedsource = self.gedmap.get_from_source_tag('GEDCOM 5.5') self.use_def_src = DEFAULT_SOURCE if self.use_def_src: self.def_src = RelLib.Source() fname = os.path.basename(filename).split('\\')[-1] self.def_src.set_title(_("Import from GEDCOM") % unicode(fname)) self.dir_path = os.path.dirname(filename) self.localref = 0 self.placemap = {} self.broken_conc_list = [ 'FamilyOrigins', 'FTW' ] self.is_ftw = 0 self.idswap = {} self.gid2id = {} self.oid2id = {} self.sid2id = {} self.lid2id = {} self.fid2id = {} self.rid2id = {} # # Parse table for INDI tag # self.indi_parse_tbl = { # +1 RESN {0:1} TOKEN_RESN : self.func_person_resn, # +1 <> {0:M} TOKEN_NAME : self.func_person_name, # +1 SEX {0:1} TOKEN_SEX : self.func_person_sex, # +1 <> {0:M} TOKEN_EVEN : self.func_person_even, TOKEN_GEVENT: self.func_person_std_event, TOKEN_BIRT : self.func_person_birt, TOKEN_RELI : self.func_person_reli, TOKEN_ADOP : self.func_person_adop, TOKEN_DEAT : self.func_person_deat, TOKEN_RESI : self.func_person_resi, # +1 <> {0:M} # +1 AFN {0:1} TOKEN_ATTR : self.func_person_std_attr, #+1 <> {0:M} TOKEN_BAPL : self.func_person_bapl, TOKEN_CONL : self.func_person_conl, TOKEN_ENDL : self.func_person_endl, TOKEN_SLGC : self.func_person_slgc, #+1 <> {0:M} TOKEN_FAMC : self.func_person_famc, # +1 <> {0:M} TOKEN_FAMS : self.func_person_fams, # +1 SUBM @@ {0:M} TOKEN_SUBM : self.skip_record, # +1 <> {0:M} TOKEN_ASSO : self.func_person_asso, # +1 ALIA @@ {0:M} TOKEN_ALIA : self.func_person_alt_name, # +1 ANCI @@ {0:M} TOKEN_ANCI : self.skip_record, # +1 DESI @@ {0:M} TOKEN_DESI : self.skip_record, # +1 <> {0:M} TOKEN_SOUR : self.func_person_sour, # +1 <> {0:M} TOKEN_OBJE : self.func_person_object, # +1 <> {0:M} TOKEN_NOTE : self.func_person_note, TOKEN_RNOTE : self.func_person_rnote, TOKEN__COMM : self.func_person_note, # +1 RFN {0:1} TOKEN_RFN : self.func_person_attr, # +1 REFN {0:M} # +2 TYPE {0:1} TOKEN_REFN : self.func_person_refn, # +1 RIN {0:1} TOKEN_RIN : self.skip_record, # +1 <> {0:1} TOKEN_CHAN : self.func_person_chan, TOKEN_ADDR : self.func_person_addr, TOKEN_PHON : self.func_person_phon, TOKEN__TODO : self.skip_record, TOKEN_TITL : self.func_person_titl, } # # Parse table for INDI.NAME # self.name_parse_tbl = { # +1 NPFX {0:1} TOKEN_NPFX : self.func_name_npfx, # +1 GIVN {0:1} TOKEN_GIVN : self.func_name_givn, # NICK {0:1} TOKEN_NICK : self.func_name_nick, # +1 SPFX {0:1} TOKEN_SURN : self.func_name_surn, # +1 NSFX {0:1} TOKEN_NSFX : self.func_name_nsfx, # +1 <> {0:M} TOKEN_SOUR : self.func_name_sour, # +1 <> {0:M} TOKEN_NOTE : self.func_name_note, # Extensions TOKEN_ALIA : self.func_name_alia, TOKEN__MARNM : self.func_name_marnm, TOKEN__AKA : self.func_name_aka, } self.repo_parse_tbl = { TOKEN_NAME : self.func_repo_name, TOKEN_ADDR : self.func_repo_addr, TOKEN_RIN : self.func_ignore, } self.event_parse_tbl = { # n TYPE {0:1} TOKEN_TYPE : self.func_event_type, # n DATE {0:1} p.*/* TOKEN_DATE : self.func_event_date, # n <> {0:1} p.* TOKEN_PLAC : self.func_event_place, # n <> {0:1} p.* TOKEN_ADDR : self.func_event_addr, # n AGE {0:1} p.* TOKEN_AGE : self.func_event_age, # n AGNC {0:1} p.* TOKEN_AGNC : self.func_event_agnc, # n CAUS {0:1} p.* TOKEN_CAUS : self.func_event_cause, # n <> {0:M} p.* TOKEN_SOUR : self.func_event_source, # n <> {0:M} p.*,* TOKEN_OBJE : self.func_event_object, # n <> {0:M} p. TOKEN_NOTE : self.func_event_note, # Other TOKEN__PRIV : self.func_event_privacy, TOKEN_OFFI : self.func_event_note, TOKEN_PHON : self.func_ignore, TOKEN__GODP : self.func_ignore, TOKEN__WITN : self.func_ignore, TOKEN__WTN : self.func_ignore, TOKEN_RELI : self.func_ignore, TOKEN_TIME : self.func_ignore, TOKEN_ASSO : self.func_ignore, TOKEN_IGNORE : self.func_ignore, TOKEN_STAT : self.func_ignore, TOKEN_TEMP : self.func_ignore, TOKEN_HUSB : self.func_event_husb, TOKEN_WIFE : self.func_event_wife, TOKEN_FAMC : self.func_person_birth_famc, } self.adopt_parse_tbl = { TOKEN_TYPE : self.func_event_type, TOKEN__PRIV : self.func_event_privacy, TOKEN_DATE : self.func_event_date, TOKEN_SOUR : self.func_event_source, TOKEN_PLAC : self.func_event_place, TOKEN_ADDR : self.func_event_addr, TOKEN_CAUS : self.func_event_cause, TOKEN_AGNC : self.func_event_agnc, TOKEN_AGE : self.func_event_age, TOKEN_NOTE : self.func_event_note, TOKEN_OFFI : self.func_event_note, TOKEN__GODP : self.func_event_ignore, TOKEN__WITN : self.func_event_ignore, TOKEN__WTN : self.func_event_ignore, TOKEN_RELI : self.func_event_ignore, TOKEN_TIME : self.func_event_ignore, TOKEN_ASSO : self.func_event_ignore, TOKEN_IGNORE : self.func_event_ignore, TOKEN_STAT : self.func_event_ignore, TOKEN_TEMP : self.func_event_ignore, TOKEN_OBJE : self.func_event_object, TOKEN_FAMC : self.func_person_adopt_famc, } self.famc_parse_tbl = { # n FAMC @@ {1:1} # +1 PEDI {0:M} p.* TOKEN_PEDI : self.func_person_famc_pedi, # +1 <> {0:M} p.* TOKEN_NOTE : self.func_person_famc_note, # Extras TOKEN__PRIMARY: self.func_person_famc_primary, TOKEN_SOUR : self.func_person_famc_sour, } self.resi_parse_tbl = { TOKEN_DATE : self.func_person_resi_date, TOKEN_ADDR : self.func_person_resi_addr, TOKEN_SOUR : self.func_person_resi_sour, TOKEN_PLAC : self.func_person_resi_plac, TOKEN_PHON : self.func_person_resi_phon, TOKEN_NOTE : self.func_person_resi_note, TOKEN_IGNORE : self.func_ignore, TOKEN_CAUS : self.func_ignore, TOKEN_STAT : self.func_ignore, TOKEN_TEMP : self.func_ignore, TOKEN_OBJE : self.func_ignore, TOKEN_TYPE : self.func_ignore, } self.person_attr_parse_tbl = { TOKEN_TYPE : self.func_person_attr_type, TOKEN_CAUS : self.func_ignore, TOKEN_DATE : self.func_ignore, TOKEN_TIME : self.func_ignore, TOKEN_ADDR : self.func_ignore, TOKEN_IGNORE: self.func_ignore, TOKEN_STAT : self.func_ignore, TOKEN_TEMP : self.func_ignore, TOKEN_OBJE : self.func_ignore, TOKEN_SOUR : self.func_person_attr_source, TOKEN_PLAC : self.func_person_attr_place, TOKEN_NOTE : self.func_person_attr_note, } self.lds_parse_tbl = { TOKEN_TEMP : self.func_lds_temple, TOKEN_DATE : self.func_lds_date, TOKEN_FAMC : self.func_lds_famc, TOKEN_FORM : self.func_lds_form, TOKEN_PLAC : self.func_lds_plac, TOKEN_SOUR : self.func_lds_sour, TOKEN_NOTE : self.func_lds_note, TOKEN_STAT : self.func_lds_stat, } self.asso_parse_tbl = { TOKEN_TYPE : self.func_person_asso_type, TOKEN_RELA : self.func_person_asso_rela, TOKEN_SOUR : self.func_person_asso_sour, TOKEN_NOTE : self.func_person_asso_note, } self.srcref_parse_tbl = { TOKEN_PAGE : self.func_srcref_page, TOKEN_DATE : self.func_srcref_date, TOKEN_DATA : self.func_srcref_data, TOKEN_OBJE : self.func_srcref_obje, TOKEN_REFN : self.func_srcref_refn, TOKEN_EVEN : self.func_ignore, TOKEN_IGNORE : self.func_ignore, TOKEN__LKD : self.func_ignore, TOKEN_QUAY : self.func_srcref_quay, TOKEN_NOTE : self.func_srcref_note, TOKEN_TEXT : self.func_srcref_text, } self.object_parse_tbl = { TOKEN_FORM : self.func_object_ref_form, TOKEN_TITL : self.func_object_ref_titl, TOKEN_FILE : self.func_object_ref_file, TOKEN_NOTE : self.func_object_ref_note, TOKEN_IGNORE : self.func_ignore, } # # FAM # self.family_func = { # +1 <> {0:M} TOKEN_GEVENT: self.func_family_std_event, TOKEN_EVEN : self.func_family_even, # +1 HUSB @@ {0:1} TOKEN_HUSB : self.func_family_husb, # +1 WIFE @@ {0:1} TOKEN_WIFE : self.func_family_wife, # +1 CHIL @@ {0:M} TOKEN_CHIL : self.func_family_chil, # +1 NCHI {0:1} # +1 SUBM @@ {0:M} # +1 <> {0:M} TOKEN_SLGS : self.func_family_slgs, # +1 <> {0:M} TOKEN_SOUR : self.func_family_source, # +1 <> {0:M} TOKEN_OBJE : self.func_family_object, # +1 <> {0:M} TOKEN__COMM : self.func_family_comm, TOKEN_NOTE : self.func_family_note, # +1 REFN {0:M} TOKEN_REFN : self.func_ignore, # +1 RIN {0:1} # +1 <> {0:1} TOKEN_CHAN : self.func_family_chan, TOKEN_ADDR : self.func_family_addr, TOKEN_RIN : self.func_ignore, TOKEN_SUBM : self.func_ignore, TOKEN_ATTR : self.func_family_attr, } self.source_func = { TOKEN_TITL : self.func_source_title, TOKEN_TAXT : self.func_source_taxt_peri, TOKEN_PERI : self.func_source_taxt_peri, TOKEN_AUTH : self.func_source_auth, TOKEN_PUBL : self.func_source_publ, TOKEN_NOTE : self.func_source_note, TOKEN_TEXT : self.func_source_text, TOKEN_ABBR : self.func_source_abbr, TOKEN_REFN : self.func_source_ignore, TOKEN_RIN : self.func_source_ignore, TOKEN_REPO : self.func_source_repo, TOKEN_OBJE : self.func_source_object, TOKEN_CHAN : self.func_source_chan, TOKEN_DATA : self.func_source_ignore, TOKEN_IGNORE: self.func_source_ignore, } self.obje_func = { TOKEN_FORM : self.func_obje_form, TOKEN_TITL : self.func_obje_title, TOKEN_FILE : self.func_obje_file, TOKEN_NOTE : self.func_obje_note, TOKEN_BLOB : self.func_obje_blob, TOKEN_REFN : self.func_obje_refn, TOKEN_TYPE : self.func_obje_type, TOKEN_RIN : self.func_obje_rin, TOKEN_CHAN : self.func_obje_chan, } self.parse_addr_tbl = { TOKEN_DATE : self.func_address_date, TOKEN_CITY : self.func_address_city, TOKEN_STAE : self.func_address_state, TOKEN_POST : self.func_address_post, TOKEN_CTRY : self.func_address_country, TOKEN_PHON : self.func_address_phone, TOKEN_SOUR : self.func_address_sour, TOKEN_NOTE : self.func_address_note, TOKEN__LOC : self.func_ignore, TOKEN__NAME : self.func_ignore, TOKEN_IGNORE: self.func_ignore, TOKEN_TYPE : self.func_ignore, TOKEN_CAUS : self.func_ignore, } # look for existing place titles, build a map self.place_names = {} cursor = dbase.get_place_cursor() data = cursor.next() while data: (handle, val) = data self.place_names[val[2]] = handle data = cursor.next() cursor.close() self.lexer = Reader(ifile) self.filename = filename self.backoff = False self.override = code_set if self.db.get_number_of_people() == 0: self.map_gid = self.map_gid_empty else: self.map_gid = self.map_gid_not_empty if self.override != 0: if self.override == 1: self.lexer.set_charset_fn(ansel_to_utf8) elif self.override == 2: self.lexer.set_charset_fn(latin_to_utf8) fullpath = os.path.normpath(os.path.abspath(filename)) self.geddir = os.path.dirname(fullpath) self.error_count = 0 amap = personalConstantAttributes self.attrs = amap.values() self.gedattr = {} for val in amap.keys(): self.gedattr[amap[val]] = val self.search_paths = [] def find_file(self, fullname, altpath): tries = [] fullname = fullname.replace('\\', os.path.sep) tries.append(fullname) if os.path.isfile(fullname): return (1, fullname) other = os.path.join(altpath, fullname) tries.append(other) if os.path.isfile(other): return (1, other) other = os.path.join(altpath, os.path.basename(fullname)) tries.append(other) if os.path.isfile(other): return (1, other) if len(fullname) > 3: if fullname[1] == ':': fullname = fullname[2:] for path in self.search_paths: other = os.path.normpath("%s/%s" % (path, fullname)) tries.append(other) if os.path.isfile(other): return (1, other) return (0, tries) else: return (0, tries) def level_is_finished(self, text, level): """ Check to see if the level has been completed, indicated by finding a level indiciated by the passed level value. If the level is finished, then make sure to call self.backup to reset the text pointer. """ done = text.level < level if done: self.backup() return done def get_next(self): if not self.backoff: self.groups = self.lexer.readline() self.update() # EOF ? if not self.groups: self.text = ""; self.backoff = False self.warn(TRUNC_MSG) self.error_count += 1 self.groups = None raise Errors.GedcomError(TRUNC_MSG) self.backoff = False return self.groups def not_recognized(self, level): """ Prints a message when an undefined token is found. All subordinate items to the current item are ignored. @param level: Current level in the file @type level: int """ text = self.groups.line msg = _("Line %d was not understood, so it was ignored.") % text self.warn(msg) self.error_count += 1 self.skip_subordinate_levels(level) def warn(self, msg): log.warning(msg) self.error_count += 1 def backup(self): self.backoff = True def parse_gedcom_file(self, use_trans=False): no_magic = self.maxpeople < 1000 self.trans = self.db.transaction_begin("", not use_trans, no_magic) self.db.disable_signals() self.parse_header() self.parse_submitter() if self.use_def_src: self.db.add_source(self.def_src, self.trans) self.parse_record() self.parse_trailer() for title in self.inline_srcs.keys(): handle = self.inline_srcs[title] src = RelLib.Source() src.set_handle(handle) src.set_title(title) self.db.add_source(src, self.trans) self.db.transaction_commit(self.trans, _("GEDCOM import")) self.db.enable_signals() self.db.request_rebuild() def parse_trailer(self): """ Looks for the expected TRLR token """ try: line = self.get_next() if line and line.token != TOKEN_TRLR: self.not_recognized(0) except TypeError: pass def parse_header(self): self.parse_header_head() self.parse_header_source() def parse_submitter(self): line = self.get_next() if line.data != "SUBM": self.backup() return else: self.parse_submitter_data(1) def parse_submitter_data(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_NAME: if self.use_def_src: self.def_src.set_author(line.data) elif line.token == TOKEN_ADDR: self.skip_subordinate_levels(level+1) def parse_record(self): """ Parses the top level (0 level) instances. """ while True: line = self.get_next() key = line.data if not line or line.token == TOKEN_TRLR: self.backup() break if key in ("FAM", "FAMILY"): self.parse_fam(line) elif key in ("INDI", "INDIVIDUAL"): self.parse_indi(line) elif key in ("OBJE", "OBJECT"): self.parse_obje(line) elif key in ("REPO", "REPOSITORY"): self.parse_REPO(line) elif key in ("SUBM", "SUBN", "SUBMITTER"): print line self.skip_subordinate_levels(1) elif line.token in (TOKEN_SUBM, TOKEN_SUBN, TOKEN_IGNORE): self.skip_subordinate_levels(1) elif key in ("SOUR","SOURCE"): self.parse_source(line.token_text, 1) elif line.data.startswith("SOUR ") or \ line.data.startswith("SOURCE "): print line # A source formatted in a single line, for example: # 0 @S62@ SOUR This is the title of the source source = self.find_or_create_source(line[3][1:-1]) source.set_title(line.data[5:]) self.db.commit_source(source, self.trans) elif key[0:4] == "NOTE": self.skip_subordinate_levels(1) elif key in ("_LOC") : print line self.skip_subordinate_levels(1) elif key in ("_EVENT_DEFN") : print line self.skip_subordinate_levels(1) else: self.not_recognized(1) def parse_level(self, state, func_map, default): """ Loops trough the current GEDCOM level level, calling the appropriate functions associated with the TOKEN. If no matching function for the token is found, the default function is called instead. """ while True: line = self.get_next() if self.level_is_finished(line, state.level): return else: func = func_map.get(line.token, default) func(line, state) #---------------------------------------------------------------------- # # INDI parsing # #---------------------------------------------------------------------- def parse_indi(self, line): """ Handling of the GEDCOM INDI tag and all lines subordinate to the current line. n @XREF:INDI@ INDI {1:1} +1 RESN {0:1} +1 <> {0:M} +1 SEX {0:1} +1 <> {0:M} +1 <> {0:M} +1 <> {0:M} +1 <> {0:M} +1 <> {0:M} +1 SUBM @@ {0:M} +1 <> {0:M} +1 ALIA @@ {0:M} +1 ANCI @@ {0:M} +1 DESI @@ {0:M} +1 <> {0:M} +1 <> {0:M} +1 <> {0:M} +1 RFN {0:1} +1 AFN {0:1} +1 REFN {0:M} +2 TYPE {0:1} +1 RIN {0:1} +1 <> {0:1} """ # find the person self.indi_count += 1 self.person = self.find_or_create_person(self.map_gid(line.token_text)) self.added.add(self.person.handle) # set up the state for the parsing state = CurrentState(person=self.person, level=1) # do the actual parsing self.parse_level(state, self.indi_parse_tbl, self.func_person_event) # Add the default reference if no source has found self.add_default_source(self.person) # commit the person to the database if self.person.change: self.db.commit_person(self.person, self.trans, change_time=state.person.change) else: self.db.commit_person(self.person, self.trans) del self.person def func_person_resn(self, line, state): """ Parses the RESN tag, adding it as an attribute. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ attr = RelLib.Attribute() attr.set_type((RelLib.AttributeType.CUSTOM, 'RESN')) state.person.add_attribute(attr) def func_person_name(self, line, state): """ Parsers the NAME token in a GEDCOM file. The text is in the format of (according to the GEDCOM Spec): > |//|//|//|// We have encountered some variations that use: > / The basic Name structure is: n NAME {1:1} +1 NPFX {0:1} +1 GIVN {0:1} +1 NICK {0:1} +1 SPFX {0:1} +1 NSFX {0:1} +1 <> {0:M} +1 <> {0:M} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ # build a RelLib.Name structure from the text name = GedcomUtils.parse_name_personal(line.data) # Add the name as the primary name if this is the first one that # we have encountered for this person. Assume that if this is the # first name, that it is a birth name. Otherwise, label it as an # "Also Known As (AKA)". GEDCOM does not seem to have the concept # of different name types if state.name_cnt == 0: name.set_type(RelLib.NameType.BIRTH) state.person.set_primary_name(name) else: name.set_type(RelLib.NameType.AKA) state.person.add_alternate_name(name) state.name_cnt += 1 # Create a new state, and parse the remainder of the NAME level sub_state = CurrentState() sub_state.person = state.person sub_state.name = name sub_state.level = state.level+1 self.parse_level(sub_state, self.name_parse_tbl, self.func_name_undefined) def func_person_sex(self, line, state): """ Parses the SEX line of a GEDCOM file. It has the format of: +1 SEX {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.person.set_gender(line.data) def func_person_even(self, line, state): """ Parses the custom EVEN tag, which has the format of: n <> {1:1} +1 <> {0:1} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event_ref = self._build_event_pair(self, RelLib.EventType.CUSTOM, self.event_parse_tbl, line.data) state.person.add_event_ref(event_ref) def func_person_std_event(self, line, state): """ Parses GEDCOM event types that map to a GRAMPS standard type. Additional parsing required is for the event detail: +1 <> {0:1} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event = RelLib.Event() event_ref = RelLib.EventRef() event.set_gramps_id(self.emapper.find_next()) event.set_type(line.data) self.parse_event_detail(event_ref, event, self.event_parse_tbl, 2) person_event_name(event, state.person) self.db.add_event(event, self.trans) event_ref.ref = event.handle state.person.add_event_ref(event_ref) def func_person_reli(self, line, state): """ Parses the RELI tag. n RELI [Y|] {1:1} +1 <> {0:1} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event_ref = self._build_event_pair(state, RelLib.EventType.RELIGION, self.event_parse_tbl, line.data) state.person.add_event_ref(event_ref) def func_person_birt(self, line, state): """ Parses GEDCOM BIRT tag into a GRAMPS birth event. Additional work must be done, since additional handling must be done by GRAMPS to set this up as a birth reference event. n BIRT [Y|] {1:1} +1 <> {0:1} p.* +1 FAMC @@ {0:1} p.* I'm not sure what value the FAMC actually offers here, since the FAMC record should handle this. Why it is a valid sub value is beyond me. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event_ref = self._build_event_pair(state, RelLib.EventType.BIRTH, self.event_parse_tbl, line.data) if state.person.get_birth_ref(): state.person.add_event_ref(event_ref) else: state.person.set_birth_ref(event_ref) def func_person_adop(self, line, state): """ Parses GEDCOM ADOP tag, subordinate to the INDI tag. Additinal tags are needed by the tag, so we pass a different function map. n ADOP [Y|] {1:1} +1 <> {0:1} p.* +1 FAMC @@ {0:1} p.* +2 ADOP {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event_ref = self._build_event_pair(self, RelLib.EventType.ADOPT, self.adopt_parse_tbl, line.data) state.person.add_event_ref(event_ref) def func_person_deat(self, line, state): """ Parses GEDCOM DEAT tag into a GRAMPS birth event. Additional work must be done, since additional handling must be done by GRAMPS to set this up as a death reference event. n DEAT [Y|] {1:1} +1 <> {0:1} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event_ref = self._build_event_pair(self, RelLib.EventType.DEATH, self.event_parse_tbl, line.data) if state.person.get_death_ref(): state.person.add_event_ref(event_ref) else: state.person.set_death_ref(event_ref) def func_person_resi(self, line, state): """ The RESI tag follows the EVENT_DETAIL structure, which is: n TYPE {0:1} n DATE {0:1} n <> {0:1} n <> {0:1} n AGE {0:1} n AGNC {0:1} n CAUS {0:1} n <> {0:M} n <> {0:M} n <> {0:M} Currently, the TYPE, AGE, CAUSE, STAT, and other tags which do not apply to an address are ignored. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ addr = RelLib.Address() sub_state = CurrentState() sub_state.person = state.person sub_state.level = state.level+1 sub_state.addr = addr sub_state.person.add_address(addr) self.parse_level(sub_state, self.resi_parse_tbl, self.func_person_unknown) def func_person_resi_date(self, line, state): """ Sets the date on the address associated with and Address. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_date_object(line.data) def func_person_resi_addr(self, line, state): """ Parses the ADDR line of a RESI tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_street(line.data) sub_state = CurrentState() sub_state.addr = state.addr sub_state.level = state.level + 1 self.parse_level(sub_state, self.parse_addr_tbl, self.func_ignore) def func_person_resi_sour(self, line, state): """ Parses the source connected to a RESI tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.add_source_reference(self.handle_source(line, state.level+1)) def func_person_resi_plac(self, line, state): """ Parses the PLAC tag connected to a RESI tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_street(line.data) self.parse_level(state, self.parse_addr_tbl, self.func_ignore) #self.parse_address(state.addr, state.level+1) def func_person_resi_phon(self, line, state): """ Parses the source connected to a PHON tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ if state.addr.get_street() == "": state.addr.set_street("Unknown") state.addr.set_phone(line.data) def func_person_resi_note(self, line, state): """ Parses the NOTE connected to a RESI tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ note = self.parse_note(line, state.addr, state.level+1, '') def func_ignore(self, line, state): """ Ignores an unsupported tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.skip_subordinate_levels(state.level+1) def func_person_unknown(self, line, state): """ Dumps out a message when an unknown tag is found @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.not_recognized(state.level) def func_person_std_attr(self, line, state): """ Parses an TOKEN that GRAMPS recognizes as an Attribute @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ sub_state = CurrentState() sub_state.person = state.person sub_state.attr = line.data sub_state.level = state.level+1 state.person.add_attribute(sub_state.attr) self.parse_level(sub_state, self.person_attr_parse_tbl, self.func_ignore) def func_person_bapl(self, line, state): """ Parses an BAPL TOKEN, producing a GRAMPS LdsOrd instance @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.build_lds_ord(state, RelLib.LdsOrd.BAPTISM) def func_person_conl(self, line, state): """ Parses an CONL TOKEN, producing a GRAMPS LdsOrd instance @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.build_lds_ord(state, RelLib.LdsOrd.CONFIRMATION) def func_person_endl(self, line, state): """ Parses an ENDL TOKEN, producing a GRAMPS LdsOrd instance @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.build_lds_ord(state, RelLib.LdsOrd.ENDOWMENT) def func_person_slgc(self, line, state): """ Parses an SLGC TOKEN, producing a GRAMPS LdsOrd instance @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.build_lds_ord(state, RelLib.LdsOrd.SEAL_TO_PARENTS) def build_lds_ord(self, state, lds_type): """ Parses an LDS ordinance, using the type passed to the routine @param state: The current state @type state: CurrentState @param lds_type: The type of the LDS ordinance @type line: LdsOrd type """ sub_state = CurrentState() sub_state.level = state.level + 1 sub_state.lds_ord = RelLib.LdsOrd() sub_state.lds_ord.set_type(lds_type) sub_state.place = None sub_state.place_fields = GedcomUtils.PlaceParser() state.person.lds_ord_list.append(sub_state.lds_ord) self.parse_level(sub_state, self.lds_parse_tbl, self.func_ignore) if sub_state.place: sub_state.place_fields.load_place(sub_state.place, sub_state.place.get_title()) def func_lds_temple(self, line, state): """ Parses the TEMP tag, looking up the code for a match. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ value = self.extract_temple(line) if value: state.lds_ord.set_temple(value) def func_lds_date(self, line, state): """ Parses the DATE tag for the LdsOrd @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.lds_ord.set_date_object(line.data) def func_lds_famc(self, line, state): """ Parses the FAMC tag attached to the LdsOrd @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ gid = GedcomUtils.extract_id(line.data) state.lds_ord.set_family_handle(self.find_family_handle(gid)) def func_lds_form(self, line, state): """ Parses the FORM tag thate defines the place structure for a place. This tag, if found, will override any global place structure. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.pf = GedcomUtils.PlaceParser(line) def func_lds_plac(self, line, state): """ Parses the PLAC tag attached to the LdsOrd. Create a new place if needed and set the title. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ try: state.place = self.find_or_create_place(line.data) state.place.set_title(line.data) state.lds_ord.set_place_handle(state.place.handle) except NameError: return def func_lds_sour(self, line, state): """ Parses the SOUR tag attached to the LdsOrd. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ srcref = self.handle_source(line, state.level+1) state.lds_ord.add_source_reference(srcref) def func_lds_note(self, line, state): """ Parses the NOTE tag attached to the LdsOrd. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.parse_note(line, state.lds_ord, state.level+1, '') def func_lds_stat(self, line, state): """ Parses the STAT (status) tag attached to the LdsOrd. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ status = lds_status.get(line.data, RelLib.LdsOrd.STATUS_NONE) state.lds_ord.set_status(status) def func_person_famc(self, line, state): """ Handles the parsing of the FAMC line, which indicates which family the person is a child of. n FAMC @@ {1:1} +1 PEDI {0:M} p.* +1 <> {0:M} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ sub_state = CurrentState() sub_state.person = state.person sub_state.level = state.level + 1 sub_state.ftype = TYPE_BIRTH sub_state.primary = False notelist = [] gid = GedcomUtils.extract_id(line.data) handle = self.find_family_handle(gid) self.parse_level(sub_state, self.famc_parse_tbl, self.func_person_unknown) # if the handle is not already in the person's parent family list, we # need to add it to thie list. flist = [fam[0] for fam in state.person.get_parent_family_handle_list()] if not handle in flist: if int(sub_state.ftype) in RELATION_TYPES: state.person.add_parent_family_handle(handle) else: if state.person.get_main_parents_family_handle() == handle: state.person.set_main_parent_family_handle(None) state.person.add_parent_family_handle(handle) # search childrefs family = self.db.find_family_from_handle(handle, self.trans) family.set_gramps_id(gid) for ref in family.get_child_ref_list(): if ref.ref == state.person.handle: ref.set_mother_relation(sub_state.ftype) ref.set_father_relation(sub_state.ftype) break else: ref = RelLib.ChildRef() ref.ref = state.person.handle ref.set_mother_relation(sub_state.ftype) ref.set_father_relation(sub_state.ftype) family.add_child_ref(ref) self.db.commit_family(family, self.trans) def func_person_famc_pedi(self, line, state): """ Parses the PEDI tag attached to a INDI.FAMC record. No values are set at this point, because we have to do some post processing. Instead, we assign the ftype field of the state variable. We convert the text from the line to an index into the PEDIGREE_TYPES dictionary, which will map to the correct ChildTypeRef. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.ftype = PEDIGREE_TYPES.get(line.data.lower(), RelLib.ChildRefType.UNKNOWN) def func_person_famc_note(self, line, state): """ Parses the INDI.FAMC.NOTE tag . @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ if not line.data.strip() or line.data and line.data[0] != "@": self.parse_note_data(state.level+1) else: self.skip_subordinate_levels(state.level+1) def func_person_famc_primary(self, line, state): """ Parses the _PRIM tag on an INDI.FAMC tag. This value is stored in the state record to be used later. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.primary = True def func_person_famc_sour(self, line, state): """ Parses the SOUR tag on an INDI.FAMC tag. GRAMPS has no corresponding record on its family relationship, so we add the source to the Person record. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ source_ref = self.handle_source(line, state.level) state.person.add_source_reference(source_ref) def func_person_fams(self, line, state): """ Parses the INDI.FAMS record, which indicates the family in which the person is a spouse. n FAMS @@ {1:1} p.* +1 <> {0:M} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ handle = self.find_family_handle(GedcomUtils.extract_id(line.data)) state.person.add_family_handle(handle) state.add_to_note(self.parse_optional_note(2)) def func_person_asso(self, line, state): """ Parse the ASSO tag, add the the referenced person to the person we are currently parsing. The GEDCOM spec indicates that valid ASSO tag is: n ASSO @@ {0:M} And the the sub tags are: ASSOCIATION_STRUCTURE:= +1 TYPE {1:1} +1 RELA {1:1} +1 <> {0:M} +1 <> {0:M} GRAMPS only supports ASSO records to people, so if the TYPE is something other than INDI, the record is ignored. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ # find the id and person that we are referencing gid = GedcomUtils.extract_id(line.data) handle = self.find_person_handle(self.map_gid(gid)) # create a new PersonRef, and assign the handle, add the # PersonRef to the active person sub_state = CurrentState() sub_state.person = state.person sub_state.level = state.level + 1 sub_state.ref = RelLib.PersonRef() sub_state.ref.ref = handle sub_state.ignore = False self.parse_level(sub_state, self.asso_parse_tbl, self.func_ignore) if not sub_state.ignore: state.person.add_person_ref(sub_state.ref) def func_person_asso_type(self, line, state): """ Parses the INDI.ASSO.TYPE tag. GRAMPS only supports the ASSO tag when the tag represents an INDI. So if the data is not INDI, we set the ignore flag, so that we ignore the record. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ if line.data != "INDI": state.ignore = True def func_person_asso_rela(self, line, state): """ Parses the INDI.ASSO.RELA tag. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.ref.rel = line.data def func_person_asso_sour(self, line, state): """ Parses the INDI.ASSO.SOUR tag. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.ref.add_source_reference(self.handle_source(line, state.level)) def func_person_asso_note(self, line, state): """ Parses the INDI.ASSO.NOTE tag. @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ note = self.parse_note(line, state.ref, state.level, "") state.ref.set_note(note) #------------------------------------------------------------------- # # Family parsing # #------------------------------------------------------------------- def parse_fam(self, line): """ n @@ FAM {1:1} +1 <> {0:M} +1 HUSB @@ {0:1} +1 WIFE @@ {0:1} +1 CHIL @@ {0:M} +1 NCHI {0:1} +1 SUBM @@ {0:M} +1 <> {0:M} +1 <> {0:M} +1 <> {0:M} +1 <> {0:M} +1 REFN {0:M} +1 RIN {0:1} +1 <> {0:1} """ # create a family self.fam_count += 1 family = self.find_or_create_family(line.token_text) # parse the family state = CurrentState() state.level = 1 state.family = family while True: line = self.get_next() if self.level_is_finished(line, 1): break if line.token not in (TOKEN_ENDL, TOKEN_BAPL, TOKEN_CONL): func = self.family_func.get(line.token, self.func_family_even) func(line, state) # handle addresses attached to families if state.addr != None: father_handle = family.get_father_handle() father = self.db.get_person_from_handle(father_handle) if father: father.add_address(state.addr) self.db.commit_person(father, self.trans) mother_handle = family.get_mother_handle() mother = self.db.get_person_from_handle(mother_handle) if mother: mother.add_address(state.addr) self.db.commit_person(mother, self.trans) for child_ref in family.get_child_ref_list(): child_handle = child_ref.ref child = self.db.get_person_from_handle(child_handle) if child: child.add_address(state.addr) self.db.commit_person(child, self.trans) # add default reference if no reference exists self.add_default_source(family) # commit family to database if family.change: self.db.commit_family(family, self.trans, change_time=family.change) else: self.db.commit_family(family, self.trans) def func_family_husb(self, line, state): """ Parses the husband line of a family n HUSB @@ {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ gid = GedcomUtils.extract_id(line.data) handle = self.find_person_handle(self.map_gid(gid)) state.family.set_father_handle(handle) def func_family_wife(self, line, state): """ Parses the wife line of a family n WIFE @@ {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ gid = GedcomUtils.extract_id(line.data) handle = self.find_person_handle(self.map_gid(gid)) state.family.set_mother_handle(handle) def func_family_std_event(self, line, state): """ Parses GEDCOM event types that map to a GRAMPS standard type. Additional parsing required is for the event detail: +1 <> {0:1} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event = RelLib.Event() event_ref = RelLib.EventRef() event_ref.set_role(RelLib.EventRoleType.FAMILY) event.set_gramps_id(self.emapper.find_next()) event.set_type(line.data) self.parse_event_detail(event_ref, event, self.event_parse_tbl, 2) family_event_name(event, state.family) self.db.add_event(event, self.trans) event_ref.ref = event.handle state.family.add_event_ref(event_ref) def func_family_even(self, line, state): """ Parses GEDCOM event types that map to a GRAMPS standard type. Additional parsing required is for the event detail: +1 <> {0:1} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ event = RelLib.Event() event_ref = RelLib.EventRef() event_ref.set_role(RelLib.EventRoleType.FAMILY) event.set_gramps_id(self.emapper.find_next()) event.set_type(line.data) self.parse_event_detail(event_ref, event, self.event_parse_tbl, 2) if int(event.get_type()) == RelLib.EventType.MARRIAGE: descr = event.get_description() if descr == "Civil Union": state.family.type.set(RelLib.FamilyRelType.CIVIL_UNION) event.set_description('') elif descr == "Unmarried": state.family.type.set(RelLib.FamilyRelType.UNMARRIED) event.set_description('') else: state.family.type.set(RelLib.FamilyRelType.MARRIED) family_event_name(event, state.family) self.db.add_event(event, self.trans) event_ref.ref = event.handle state.family.add_event_ref(event_ref) def func_family_chil(self, line, state): """ Parses the child line of a family n CHIL @@ {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ mrel, frel = self.parse_ftw_relations(state.level+1) gid = GedcomUtils.extract_id(line.data) child = self.find_or_create_person(self.map_gid(gid)) reflist = [ ref for ref in state.family.get_child_ref_list() \ if ref.ref == child.handle ] if reflist: ref = reflist[0] if mrel != RelLib.ChildRefType.BIRTH or \ frel != RelLib.ChildRefType.BIRTH: ref.set_father_relation(frel) ref.set_mother_relation(mrel) else: ref = RelLib.ChildRef() ref.ref = child.handle ref.set_father_relation(frel) ref.set_mother_relation(mrel) state.family.add_child_ref(ref) def func_family_slgs(self, state, line): """ n SLGS {1:1} +1 STAT {0:1} +1 DATE {0:1} +1 TEMP {0:1} +1 PLAC {0:1} +1 <> {0:M} +1 <> {0:M} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ sub_state = CurrentState() sub_state.level = state.level + 1 sub_state.lds_ord = RelLib.LdsOrd() sub_state.lds_ord.set_type(RelLib.LdsOrd.SEAL_TO_SPOUSE) sub_state.place = None sub_state.place_fields = GedcomUtils.PlaceParser() state.family.lds_ord_list.append(sub_state.lds_ord) self.parse_level(sub_state, self.lds_parse_tbl, self.func_ignore) if sub_state.place: sub_state.place_fields.load_place(sub_state.place, sub_state.place.get_title()) def func_family_source(self, line, state): """ n SOUR @@ /* pointer to source record */ {1:1} p.* +1 PAGE {0:1} p.* +1 EVEN {0:1} p.* +1 DATA {0:1} +1 QUAY {0:1} p.* +1 <> {0:M} p.*,* +1 <> {0:M} p.* | /* Systems not using source records */ n SOUR {1:1} p.* +1 [ CONC | CONT ] {0:M} +1 TEXT {0:M} p.* +1 <> {0:M} p.* @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ source_ref = self.handle_source(line, state.level+1) state.family.add_source_reference(source_ref) def func_family_object(self, line, state): """ +1 <> {0:M} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ if line.data and line.data[0] == '@': self.not_recognized(state.level) else: (form, filename, title, note) = self.func_obje(state.level) self.build_media_object(state.family, form, filename, title, note) def func_family_comm(self, line, state): """ @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ note = line.data state.family.set_note(note) self.skip_subordinate_levels(state.level+1) def func_family_note(self, line, state): """ +1 <> {0:M} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.parse_note(line, state.family, state.level, '') def func_family_chan(self, line, state): """ +1 <> {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ self.parse_change(line, state.family, state.level) def func_family_addr(self, line, state): """ @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr = RelLib.Address() state.addr.set_street(line.data) self.parse_level(state, self.parse_addr_tbl, self.func_ignore) #self.parse_address(state.addr, state.level) def func_family_attr(self, line, state): """ @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.family.add_attribute(line.data) def func_obje(self, level): """ n OBJE {1:1} +1 FORM {1:1} +1 TITL {0:1} +1 FILE {1:1} +1 <> {0:M} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ sub_state = CurrentState() sub_state.form = "" sub_state.filename = "" sub_state.title = "" sub_state.note = "" sub_state.level = level self.parse_level(sub_state, self.object_parse_tbl, self.func_ignore) return (sub_state.form, sub_state.filename, sub_state.title, sub_state.note) def func_object_ref_form(self, line, state): """ +1 FORM {1:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.form = line.data def func_object_ref_titl(self, line, state): """ +1 TITL {0:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.title = line.data def func_object_ref_file(self, line, state): """ +1 FILE {1:1} @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.filename = line.data def func_object_ref_note(self, line, state): """ +1 <> {0:M} TODO: Fix this for full reference @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.note = line.data ############################################################################### def map_gid_empty(self, gid): return gid def map_gid_not_empty(self, gid): if self.idswap.get(gid): return self.idswap[gid] else: if self.db.id_trans.get(str(gid)): self.idswap[gid] = self.db.find_next_person_gramps_id() else: self.idswap[gid] = gid return self.idswap[gid] def find_or_create_person(self, gramps_id): person = RelLib.Person() intid = self.gid2id.get(gramps_id) if self.db.has_person_handle(intid): person.unserialize(self.db.get_raw_person_data(intid)) else: intid = self.find_person_handle(gramps_id) person.set_handle(intid) person.set_gramps_id(gramps_id) return person def find_person_handle(self, gramps_id): intid = self.gid2id.get(gramps_id) if not intid: intid = create_id() self.gid2id[gramps_id] = intid return intid def find_object_handle(self, gramps_id): intid = self.oid2id.get(gramps_id) if not intid: intid = create_id() self.oid2id[gramps_id] = intid return intid def find_or_create_family(self, gramps_id): family = RelLib.Family() intid = self.fid2id.get(gramps_id) if self.db.has_family_handle(intid): family.unserialize(self.db.get_raw_family_data(intid)) else: intid = self.find_family_handle(gramps_id) family.set_handle(intid) family.set_gramps_id(gramps_id) return family def find_or_create_repository(self, gramps_id): repository = RelLib.Repository() if not gramps_id: need_commit = True gramps_id = self.db.find_next_repository_gramps_id() else: need_commit = False intid = self.rid2id.get(gramps_id) if self.db.has_repository_handle(intid): repository.unserialize(self.db.get_raw_repository_data(intid)) else: intid = self.find_repository_handle(gramps_id) repository.set_handle(intid) repository.set_gramps_id(gramps_id) if need_commit: self.db.commit_repository(repository, self.trans) return repository def find_or_create_object(self, gramps_id): obj = RelLib.MediaObject() intid = self.oid2id.get(gramps_id) if self.db.has_object_handle(intid): obj.unserialize(self.db.get_raw_object_data(intid)) else: intid = self.find_object_handle(gramps_id) obj.set_handle(intid) obj.set_gramps_id(gramps_id) return obj def find_repository_handle(self, gramps_id): intid = self.rid2id.get(gramps_id) if not intid: intid = create_id() self.rid2id[gramps_id] = intid return intid def find_family_handle(self, gramps_id): intid = self.fid2id.get(gramps_id) if not intid: intid = create_id() self.fid2id[gramps_id] = intid return intid def find_or_create_source(self, gramps_id): source = RelLib.Source() intid = self.sid2id.get(gramps_id) if self.db.has_source_handle(intid): source.unserialize(self.db.get_raw_source_data(intid)) else: intid = create_id() source.set_handle(intid) source.set_gramps_id(gramps_id) self.db.add_source(source, self.trans) self.sid2id[gramps_id] = intid return source def find_or_create_place(self, title): place = RelLib.Place() # check to see if we've encountered this name before # if we haven't we need to get a new GRAMPS ID intid = self.place_names.get(title) if intid == None: intid = self.lid2id.get(title) if intid == None: new_id = self.db.find_next_place_gramps_id() else: new_id = None else: new_id = None # check to see if the name already existed in the database # if it does, create a new name by appending the GRAMPS ID. # generate a GRAMPS ID if needed if self.db.has_place_handle(intid): place.unserialize(self.db.get_raw_place_data(intid)) else: intid = create_id() place.set_handle(intid) place.set_title(title) place.set_gramps_id(new_id) self.db.add_place(place, self.trans) self.lid2id[title] = intid return place def parse_cause(self, event, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_SOUR: event.add_source_reference(self.handle_source(line, level+1)) else: self.not_recognized(1) def parse_repo_caln(self, line, repo, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_CALN: repo.set_call_number(line.data) #self.parse_repo_caln(line, repo. level+1) elif line.token == TOKEN_NOTE: repo.set_note(line.data) else: self.not_recognized(1) def parse_repo_ref(self, line, repo_ref, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_CALN: repo_ref.set_call_number(line.data) self.parse_repo_ref_caln(repo_ref, level+1) elif line.token == TOKEN_NOTE: note = self.parse_note(line, repo_ref, level+1, "") repo_ref.set_note(note) else: self.not_recognized(1) def parse_repo_ref_caln(self, reporef, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_MEDI: reporef.media_type.set(line.data) else: self.not_recognized(1) def parse_note_data(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token in (TOKEN_SOUR, TOKEN_CHAN, TOKEN_REFN, TOKEN_IGNORE): self.skip_subordinate_levels(level+1) elif line.token == TOKEN_RIN: continue else: self.not_recognized(level+1) def parse_ftw_relations(self, level): mrel = RelLib.ChildRefType() frel = RelLib.ChildRefType() while True: line = self.get_next() if self.level_is_finished(line, level): break # FTW elif line.token == TOKEN__FREL: frel = PEDIGREE_TYPES.get(line.data.lower(), TYPE_BIRTH) # FTW elif line.token == TOKEN__MREL: mrel = PEDIGREE_TYPES.get(line.data.lower(), TYPE_BIRTH) elif line.token == TOKEN_ADOP: mrel = TYPE_ADOPT frel = TYPE_ADOPT # Legacy elif line.token == TOKEN__STAT: mrel = TYPE_BIRTH frel = TYPE_BIRTH # Legacy _PREF elif line.token == TOKEN__PRIMARY: continue else: self.not_recognized(level+1) return (mrel, frel) def func_event_object(self, line, event_ref, event, level): if line.data and line.data[0] == '@': self.not_recognized(level) else: (form, filename, title, note) = self.func_obje(level) self.build_media_object(event, form, filename, title, note) def func_place_object(self, line, place, level): if line.data and line.data[0] == '@': self.not_recognized(level) else: (form, filename, title, note) = self.func_obje(level) self.build_media_object(place, form, filename, title, note) def func_source_object(self, line, source, level): if line.data and line.data[0] == '@': self.not_recognized(level) else: (form, filename, title, note) = self.func_obje(level+1) self.build_media_object(source, form, filename, title, note) def parse_note_base(self, line, obj, level, old_note, task): # reference to a named note defined elsewhere if line.token == TOKEN_RNOTE: note_obj = self.note_map.get(line.data) if note_obj: new_note = note_obj.get() else: new_note = u"" else: new_note = line.data self.skip_subordinate_levels(level+1) if old_note: note = u"%s\n%s" % (old_note, line.data) else: note = new_note task(note) return note def parse_note_simple(self, line, level): # reference to a named note defined elsewhere if line.data and line.data[0] == "@": note_obj = self.note_map.get(line.data) note = note_obj.get() else: note = line.data self.skip_subordinate_levels(level+1) return note def parse_note(self, line, obj, level, old_note): return self.parse_note_base(line, obj, level, old_note, obj.set_note) def parse_comment(self, line, obj, level, old_note): return self.parse_note_base(line, obj, level, old_note, obj.set_note) def parse_obje(self, line): """ n @XREF:OBJE@ OBJE {1:1} +1 FORM {1:1} p.* +1 TITL {0:1} p.* +1 <> {0:M} p.* +1 BLOB {1:1} +2 CONT {1:M} p.* +1 OBJE @@ /* chain to continued object */ {0:1} p.* +1 REFN {0:M} p.* +2 TYPE {0:1} p.* +1 RIN {0:1} p.* +1 <> {0:1} p.* """ gid = GedcomUtils.extract_id(line.data) self.media = self.find_or_create_object(self.map_gid(gid)) while True: line = self.get_next() if self.level_is_finished(line, 1): break else: func = self.obje_func.get(line.token, self.func_obje_ignore) func(line, self.media, 1) # Add the default reference if no source has found if self.use_def_src and len(self.media.get_source_references()) == 0: sref = RelLib.SourceRef() sref.set_reference_handle(self.def_src.handle) self.media.add_source_reference(sref) # commit the person to the database if self.media.change: self.db.commit_media_object(self.media, self.trans, change_time=self.media.change) else: self.db.commit_media_object(self.media, self.trans) del self.media #---------------------------------------------------------------------- # # REPO parsing # #---------------------------------------------------------------------- def parse_REPO(self, line): self.repo_count += 1 self.repo = self.find_or_create_repository(line.token_text) self.added.add(self.repo.handle) self.parse_repository(self.repo) self.db.commit_repository(self.repo, self.trans) del self.repo def parse_optional_note(self, level): note = "" while True: line = self.get_next() if self.level_is_finished(line, level): return note elif line.token == TOKEN_NOTE: if not line.data.strip() or line.data and line.data[0] != "@": note = line.data self.parse_note_data(level+1) else: self.skip_subordinate_levels(level+1) else: self.not_recognized(level+1) return None def parse_address(self, address, level): first = 0 note = "" while True: line = self.get_next() if line.level < level: if line.token == TOKEN_PHON: address.set_phone(line.data) else: self.backup() break elif line.token in (TOKEN_ADDR, TOKEN_ADR1, TOKEN_ADR2): val = address.get_street() if first == 0: val = line.data first = 1 else: val = "%s, %s" % (val, line.data) address.set_street(val) elif line.token == TOKEN_DATE: address.set_date_object(line.data) elif line.token == TOKEN_CITY: address.set_city(line.data) elif line.token == TOKEN_STAE: address.set_state(line.data) elif line.token == TOKEN_POST: address.set_postal_code(line.data) elif line.token == TOKEN_CTRY: address.set_country(line.data) elif line.token == TOKEN_PHON: address.set_phone(line.data) elif line.token == TOKEN_SOUR: address.add_source_reference(self.handle_source(line, level+1)) elif line.token == TOKEN_NOTE: note = self.parse_note(line, address, level+1, '') elif line.token in (TOKEN__LOC, TOKEN__NAME): continue # ignore unsupported extended location syntax elif line.token in (TOKEN_IGNORE, TOKEN_TYPE, TOKEN_CAUS): self.skip_subordinate_levels(level+1) else: self.not_recognized(level+1) def func_address_date(self, line, state): """ Parses the DATE line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_date_object(line.data) def func_address_city(self, line, state): """ Parses the CITY line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_city(line.data) def func_address_state(self, line, state): """ Parses the STAE line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_state(line.data) def func_address_post(self, line, state): """ Parses the POST line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_postal_code(line.data) def func_address_country(self, line, state): """ Parses the country line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_country(line.data) def func_address_phone(self, line, state): """ Parses the PHON line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.set_phone(line.data) def func_address_sour(self, line, state): """ Parses the SOUR line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ state.addr.add_source_reference(self.handle_source(line, state.level+1)) def func_address_note(self, line, state): """ Parses the NOTE line of an ADDR tag @param line: The current line in GedLine format @type line: GedLine @param state: The current state @type state: CurrentState """ note = self.parse_note(line, state.addr, state.level+1, '') def parse_place_as_address(self, street, level): note = None location = RelLib.Location() if street: location.set_street(street) added = True else: added = False while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token in (TOKEN_ADDR, TOKEN_ADR1, TOKEN_ADR2): val = location.get_street() if val: val = "%s, %s" % (val, line.data.strip()) else: val = line.data.strip() location.set_street(val.replace('\n',' ')) added = True elif line.token == TOKEN_DATE: location.set_date_object(line.data) added = True elif line.token == TOKEN_CITY: location.set_city(line.data) added = True elif line.token == TOKEN_STAE: location.set_state(line.data) added = True elif line.token == TOKEN_POST: location.set_postal_code(line.data) added = True elif line.token == TOKEN_CTRY: location.set_country(line.data) added = True elif line.token == TOKEN_NOTE: note = self.parse_note_simple(line, level+1) added = True elif line.token in (TOKEN__LOC, TOKEN__NAME, TOKEN_PHON): continue # ignore unsupported extended location syntax else: self.not_recognized(level+1) if added: return (location, note) else: return (None, None) def parse_event_detail(self, event_ref, event, func_map, level): """ n TYPE {0:1} p.* n DATE {0:1} p.*/* n <> {0:1} p.* n <> {0:1} p.* n AGE {0:1} p.* n AGNC {0:1} p.* n CAUS {0:1} p.* n <> {0:M} p.* n <> {0:M} p.*,* n <> {0:M} p. """ while True: line = self.get_next() if self.level_is_finished(line, level): break else: func = func_map.get(line.token, self.func_event_undef) func(line, event_ref, event, level+1) def func_event_ignore(self, line, event_ref, event, level): self.skip_subordinate_levels(level) def func_event_undef(self, line, event_ref, event, level): self.not_recognized(level) def func_event_type(self, line, event_ref, event, level): """ Parses the TYPE line for an event. """ if event.get_type().is_custom(): if GED_2_GRAMPS.has_key(line.data): name = RelLib.EventType(GED_2_GRAMPS[line.data]) else: val = self.gedsource.tag2gramps(line.data) if val: name = RelLib.EventType((RelLib.EventType.CUSTOM, val)) else: name = RelLib.EventType((RelLib.EventType.CUSTOM, line[3])) event.set_type(name) else: try: if not GED_2_GRAMPS.has_key(line.data) and \ not GED_2_FAMILY.has_key(line.data) and \ line.data[0] != 'Y': event.set_description(line.data) except IndexError: pass def func_event_privacy(self, line, event_ref, event, level): event.set_privacy(True) def func_person_adopt_famc(self, line, event_ref, event, level): gid = line.data.strip()[1:-1] handle = self.find_family_handle(gid) family = self.find_or_create_family(gid) mrel, frel = self.parse_adopt_famc(level); if self.person.get_main_parents_family_handle() == handle: self.person.set_main_parent_family_handle(None) self.person.add_parent_family_handle(handle) reflist = [ ref for ref in family.get_child_ref_list() \ if ref.ref == self.person.handle ] if reflist: ref = reflist[0] ref.set_father_relation(frel) ref.set_mother_relation(mrel) else: ref = RelLib.ChildRef() ref.ref = self.person.handle ref.set_father_relation(frel) ref.set_mother_relation(mrel) family.add_child_ref(ref) self.db.commit_family(family, self.trans) def func_person_birth_famc(self, line, event_ref, event, level): handle = self.find_family_handle(line.data.strip()[1:-1]) if self.person.get_main_parents_family_handle() == handle: self.person.set_main_parent_family_handle(None) self.person.add_parent_family_handle(handle) frel = mrel = RelLib.ChildRefType.BIRTH family = self.db.find_family_from_handle(handle, self.trans) reflist = [ ref for ref in family.get_child_ref_list() \ if ref.ref == self.person.handle ] if reflist: ref = reflist[0] ref.set_father_relation(frel) ref.set_mother_relation(mrel) else: ref = RelLib.ChildRef() ref.ref = self.person.handle ref.set_father_relation(frel) ref.set_mother_relation(mrel) family.add_child_ref(ref) self.db.commit_family(family, self.trans) def func_event_note(self, line, event_ref, event, level): self.parse_note(line, event, level+1,'') def func_event_date(self, line, event_ref, event, level): event.set_date_object(line.data) def func_event_source(self, line, event_ref, event, level): event.add_source_reference(self.handle_source(line, level)) def func_event_addr(self, line, event_ref, event, level): (location, note) = self.parse_place_as_address(line.data, level) if location: index = line.data + location.get_street() else: index = line.data place_handle = event.get_place_handle() if place_handle: place = self.db.get_place_from_handle(place_handle) main_loc = place.get_main_location() if main_loc and main_loc.get_street() != location.get_street(): old_title = place.get_title() place = self.find_or_create_place(index) place.set_title(old_title) place_handle = place.handle else: place = self.find_or_create_place(index) place.set_title(line.data) place_handle = place.handle #load_place_values(place, line.data) if location: place.set_main_location(location) if note: place.set_note(note) event.set_place_handle(place_handle) self.db.commit_place(place, self.trans) def func_event_place(self, line, event_ref, event, level): """ Parse the place portion of a event. A special case has to be made for Family Tree Maker, which violates the GEDCOM spec. It uses the PLAC field to store the description or value assocated with the event. n PLAC {1:1} +1 FORM {0:1} +1 <> {0:M} +1 <> {0:M} """ val = line.data n = event.get_type() if self.is_ftw and int(n) in [RelLib.EventType.OCCUPATION, RelLib.EventType.RELIGION, RelLib.EventType.DEGREE]: event.set_description(val) else: place = self.find_or_create_place(val) place_handle = place.handle place.set_title(val) event.set_place_handle(place_handle) pf = self.place_parser while True: line = self.get_next() if self.level_is_finished(line, level): pf.load_place(place, place.get_title()) break elif line.token == TOKEN_NOTE: note = self.parse_note(line, place, level+1, '') place.set_note(note) elif line.token == TOKEN_FORM: pf = GedcomUtils.PlaceParser(line) elif line.token == TOKEN_OBJE: self.func_place_object(line, place, level+1) elif line.token == TOKEN_SOUR: place.add_source_reference( self.handle_source(line, level+1)) self.db.commit_place(place, self.trans) def func_event_cause(self, line, event_ref, event, level): a = RelLib.Attribute() a.set_type(RelLib.AttributeType.CAUSE) a.set_value(line.data) event.add_attribute(a) self.parse_cause(a, level+1) def func_event_age(self, line, event_ref, event, level): a = RelLib.Attribute() a.set_type(RelLib.AttributeType.AGE) a.set_value(line.data) event_ref.add_attribute(a) def func_event_husb(self, line, event_ref, event, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_AGE: a = RelLib.Attribute() a.set_type(RelLib.AttributeType.FATHER_AGE) a.set_value(line.data) event_ref.add_attribute(a) def func_event_wife(self, line, event_ref, event, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_AGE: a = RelLib.Attribute() a.set_type(RelLib.AttributeType.MOTHER_AGE) a.set_value(line.data) event_ref.add_attribute(a) def func_event_agnc(self, line, event_ref, event, level): a = RelLib.Attribute() a.set_type(RelLib.AttributeType.AGENCY) a.set_value(line.data) event.add_attribute(a) def parse_adopt_famc(self, level): mrel = TYPE_BIRTH frel = TYPE_BIRTH while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_ADOP: if line.data == "HUSB": frel = TYPE_ADOPT elif line.data == "WIFE": mrel = TYPE_ADOPT else: self.not_recognized(level+1) return (mrel, frel) def parse_person_attr(self, attr, level): """ GRAMPS uses an Attribute to store some information. Technically, GEDCOM does not make a distinction between Attributes and Events, so what GRAMPS considers to be an Attribute can have more information than what we allow. """ while True: line = self.get_next() if self.level_is_finished(line, level): break else: func = self.person_attr_parse_tbl.get(line.token, self.func_person_attr_undef) func(attr, line, level+1) def func_person_attr_undef(self, attr, line, level): """ Called when an undefined token is found """ self.not_recognized(level) def func_person_attr_ignore(self, attr, line, level): """ Called when an attribute is found that we know we want to ignore """ self.skip_subordinate_levels(level) def func_person_attr_type(self, attr, line, level): if attr.get_type() == "": if GED_2_GRAMPS.has_key(line.data): name = GED_2_GRAMPS[line.data] else: val = self.gedsource.tag2gramps(line.data) if val: name = val else: name = line.data attr.set_type(name) def func_person_attr_source(self, attr, line, level): attr.add_source_reference(self.handle_source(line, level)) def func_person_attr_place(self, attr, line, level): val = line.data if attr.get_value() == "": attr.set_value(val) self.skip_subordinate_levels(level) def func_person_attr_note(self, attr, line, level): info = self.parse_note(line, attr, level+1, '') attr.set_note(info) def parse_source_reference(self, src_ref, level, handle): """Reads the data associated with a SOUR reference""" state = CurrentState() state.level = level state.src_ref = src_ref state.handle = handle self.parse_level(state, self.srcref_parse_tbl, self.func_srcref_ignore) def func_srcref_page(self, line, state): state.src_ref.set_page(line.data) def func_srcref_date(self, line, state): state.src_ref.set_date_object(line.data) def func_srcref_data(self, line, state): date, text = self.parse_source_data(state.level+1) if date: import DateHandler d = DateHandler.parser.parse(date) state.src_ref.set_date_object(d) state.src_ref.set_text(text) def func_srcref_obje(self, line, state): if line.data and line.data[0] == '@': self.not_recognized(state.level) else: src = self.db.get_source_from_handle(state.handle) (form, filename, title, note) = self.func_obje(state.level) self.build_media_object(src, form, filename, title, note) self.db.commit_source(src, self.trans) def func_srcref_refn(self, line, state): self.skip_subordinate_levels(state.level+1) def func_srcref_ignore(self, line, state): self.skip_subordinate_levels(state.level+1) def func_srcref_quay(self, line, state): try: val = int(line.data) except ValueError: return # If value is greater than 3, cap at 3 val = min(val, 3) if val > 1: state.src_ref.set_confidence_level(val+1) else: state.src_ref.set_confidence_level(val) def func_srcref_note(self, line, state): note = self.parse_comment(line, state.src_ref, state.level+1, '') state.src_ref.set_note(note) def func_srcref_text(self, line, state): note = self.parse_comment(line, state.src_ref, state.level+1, '') state. src_ref.set_text(note) def parse_source_data(self, level): """Parses the source data""" date = "" note = "" while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_DATE: date = line.data elif line.token == TOKEN_TEXT: note = line.data else: self.not_recognized(level+1) return (date, note) def parse_header_head(self): """validiates that this is a valid GEDCOM file""" line = self.lexer.readline() if line.token != TOKEN_HEAD: raise Errors.GedcomError("%s is not a GEDCOM file" % self.filename) def parse_header_source(self): genby = "" note = "" while True: line = self.get_next() if self.level_is_finished(line, 1): return elif line.token == TOKEN_SOUR: self.gedsource = self.gedmap.get_from_source_tag(line.data) self.lexer.set_broken_conc(self.gedsource.get_conc()) if line.data == "FTW": self.is_ftw = 1 genby = line.data elif line.token == TOKEN_NAME: pass elif line.token == TOKEN_VERS: if self.use_def_src: self.def_src.set_data_item('Generated by',"%s %s" % (genby, line.data)) elif line.token == TOKEN_FILE: if self.use_def_src: filename = os.path.basename(line.data).split('\\')[-1] self.def_src.set_title(_("Import from %s") % filename) elif line.token == TOKEN_COPR: if self.use_def_src: self.def_src.set_publication_info(line.data) elif line.token == TOKEN_SUBM: self.parse_subm(2) elif line.token in (TOKEN_CORP, TOKEN_DATA, TOKEN_SUBN, TOKEN_LANG, TOKEN_TIME): self.skip_subordinate_levels(2) elif line.token == TOKEN_DEST: if genby == "GRAMPS": self.gedsource = self.gedmap.get_from_source_tag(line.data) self.lexer.set_broken_conc(self.gedsource.get_conc()) elif line.token == TOKEN_CHAR and not self.override: if line.data == "ANSEL": self.lexer.set_charset_fn(ansel_to_utf8) elif line.data not in ("UNICODE","UTF-8","UTF8"): self.lexer.set_charset_fn(latin_to_utf8) self.skip_subordinate_levels(2) elif line.token == TOKEN_GEDC: self.skip_subordinate_levels(2) elif line.token == TOKEN__SCHEMA: self.parse_ftw_schema(2) elif line.token == TOKEN_PLAC: self.parse_place_form(2) elif line.token == TOKEN_DATE: self.parse_date(2) if self.use_def_src: self.def_src.set_data_item('Creation date', line.data) elif line.token == TOKEN_NOTE: if self.use_def_src: note = self.parse_note(line, self.def_src, 2, '') elif line.token == TOKEN_UNKNOWN: self.skip_subordinate_levels(2) else: self.not_recognized(2) def parse_subm(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_NAME: if self.use_def_src: self.def_src.set_author(line.data) else: self.skip_subordinate_levels(2) def parse_ftw_schema(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_INDI: self.parse_ftw_indi_schema(level+1) elif line.token == TOKEN_FAM: self.parse_ftw_fam_schema(level+1) else: self.not_recognized(2) def parse_ftw_indi_schema(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break else: GED_2_GRAMPS[line.token] = self.parse_label(level+1) def parse_label(self, level): value = None while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_LABL: value = line.data else: self.not_recognized(2) return value def parse_ftw_fam_schema(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break else: GED_2_FAMILY_CUSTOM[line.token_text] = self.parse_label(level+1) def skip_subordinate_levels(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break def ignore_change_data(self, level): line = self.get_next() if line.token == TOKEN_CHAN: self.skip_subordinate_levels(level+1) else: self.backup() def parse_place_form(self, level): while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_FORM: self.place_parser.parse_form(line) else: self.not_recognized(level+1) def parse_date(self, level): while True: line = self.get_next() print line if self.level_is_finished(line, level): break elif line.token == TOKEN_TIME: pass else: self.not_recognized(level+1) def handle_source(self, line, level): source_ref = RelLib.SourceRef() if line.data and line.data[0] != "@": title = line.data handle = self.inline_srcs.get(title, Utils.create_id()) self.inline_srcs[title] = handle self.parse_source_reference(source_ref, level, handle) else: handle = self.find_or_create_source(line.data.strip()[1:-1]).handle self.parse_source_reference(source_ref, level, handle) source_ref.set_reference_handle(handle) return source_ref def resolve_refns(self): return prefix = self.db.iprefix index = 0 new_pmax = self.db.pmap_index for pid in self.added: index = index + 1 if self.refn.has_key(pid): val = self.refn[pid] new_key = prefix % val new_pmax = max(new_pmax, val) person = self.db.get_person_from_handle(pid, self.trans) # new ID is not used if not self.db.has_person_handle(new_key): self.db.remove_person(pid, self.trans) person.set_handle(new_key) person.set_gramps_id(new_key) self.db.add_person(person, self.trans) else: tp = self.db.get_person_from_handle(new_key, self.trans) # same person, just change it if person == tp: self.db.remove_person(pid, self.trans) person.set_handle(new_key) person.set_gramps_id(new_key) self.db.add_person(person, self.trans) # give up trying to use the refn as a key else: pass self.db.pmap_index = new_pmax def invert_year(self, subdate): return (subdate[0], subdate[1], -subdate[2], subdate[3]) #-------------------------------------------------------------------- # # # #-------------------------------------------------------------------- def func_person_chan(self, line, state): self.parse_change(line, state.person, state.level+1) def func_source_chan(self, line, source, level): self.parse_change(line, source, level+1) def parse_change(self, line, obj, level): """ CHANGE_DATE:= > n CHAN {1:1} > +1 DATE {1:1} > +2 TIME {0:1} > +1 <> {0:M} The Note structure is ignored, since we have nothing corresponding in GRAMPS. Based on the values calculated, attempt to convert to a valid change time using time.strptime. If this fails (and it shouldn't unless the value is meaningless and doesn't conform to the GEDCOM spec), the value is ignored. """ tstr = None dstr = None while True: line = self.get_next() if self.level_is_finished(line, level): break elif line.token == TOKEN_TIME: tstr = line.data elif line.token == TOKEN_DATE: dstr = line.data elif line.token == TOKEN_NOTE: self.skip_subordinate_levels(level+1) else: self.not_recognized(level+1) # Attempt to convert the values to a valid change time if tstr: try: if dstr: tstruct = time.strptime("%s %s" % (dstr, tstr), "%d %b %Y %H:%M:%S") else: tstruct = time.strptime(tstr, "%d %b %Y") val = time.mktime(tstruct) obj.change = val except ValueError: # parse of time structure failed, so ignore pass def func_person_alt_name(self, line, state): """ Parse a altername name, usually indicated by a AKA or _AKA tag. This is not valid GEDCOM, but several programs will add this just to make life interesting. Odd, since GEDCOM supports multiple NAME indicators, which is the correct way of handling multiple names. """ name = GedcomUtils.parse_name_personal(line.data) name.set_type(RelLib.NameType.AKA) state.person.add_alternate_name(name) # Create a new state, and parse the remainder of the NAME level sub_state = CurrentState() sub_state.person = state.person sub_state.name = name sub_state.level = 2 self.parse_level(sub_state, self.name_parse_tbl, self.func_name_undefined) def func_person_object(self, line, state): """ Currently, the embedded form is not supported Embedded form > n OBJE @@ {1:1} Linked form > n OBJE {1:1} > +1 FORM {1:1} > +1 TITL {0:1} > +1 FILE {1:1} > +1 <> {0:M} """ if line.data and line.data[0] == '@': ref = RelLib.MediaRef() handle = self.find_object_handle(line.data[1:-1]) ref.set_reference_handle(handle) self.person.add_media_reference(ref) else: (form, filename, title, note) = self.func_obje(state.level+1) self.build_media_object(state.person, form, filename, title, note) def build_media_object(self, obj, form, filename, title, note): if form == "url": url = RelLib.Url() url.set_path(filename) url.set_description(title) obj.add_url(url) else: (ok, path) = self.find_file(filename, self.dir_path) if not ok: self.warn(_("Could not import %s") % filename) path = filename.replace('\\', os.path.sep) photo_handle = self.media_map.get(path) if photo_handle == None: photo = RelLib.MediaObject() photo.set_path(path) photo.set_description(title) full_path = os.path.abspath(path) if os.path.isfile(full_path): photo.set_mime_type(Mime.get_type(full_path)) else: photo.set_mime_type(MIME_MAP.get(form.lower(),'unknown')) self.db.add_object(photo, self.trans) self.media_map[path] = photo.handle else: photo = self.db.get_object_from_handle(photo_handle) oref = RelLib.MediaRef() oref.set_reference_handle(photo.handle) oref.set_note(note) obj.add_media_reference(oref) def func_person_note(self, line, state): self.note = self.parse_note(line, self.person, 1, state.note) def func_person_rnote(self, line, state): self.note = self.parse_note(line, self.person, 1, state.note) def func_person_addr(self, line, state): """ Parses the Address structure by calling parse_address. """ state.addr = RelLib.Address() state.addr.set_street(line.data) self.parse_level(state, self.parse_addr_tbl, self.func_ignore) #self.parse_address(addr, 2) state.person.add_address(state.addr) def func_person_phon(self, line, state): addr = RelLib.Address() addr.set_street("Unknown") addr.set_phone(line.data) state.person.add_address(addr) def func_person_titl(self, line, state): event = RelLib.Event() event_ref = RelLib.EventRef() event.set_gramps_id(self.emapper.find_next()) event.set_type(RelLib.EventType.NOB_TITLE) self.parse_event_detail(event_ref, event, self.event_parse_tbl, 2) person_event_name(event, state.person) self.db.add_event(event, self.trans) def func_person_attr_plac(self, line, state): if state.attr.get_value() == "": state.attr.set_value(line.data) def _build_event_pair(self, state, event_type, event_map, description): event = RelLib.Event() event_ref = RelLib.EventRef() event.set_gramps_id(self.emapper.find_next()) event.set_type(event_type) if description and description != 'Y': event.set_description(description) self.parse_event_detail(event_ref, event, event_map, 2) person_event_name(event, state.person) self.db.add_event(event, self.trans) event_ref.set_reference_handle(event.handle) return event_ref def _build_family_event_pair(self, state, event_type, event_map, description): event = RelLib.Event() event_ref = RelLib.EventRef() event.set_gramps_id(self.emapper.find_next()) event.set_type(event_type) if description and description != 'Y': event.set_description(description) self.parse_event_detail(event_ref, event, event_map, 2) family_event_name(event, state.family) self.db.add_event(event, self.trans) event_ref.set_reference_handle(event.handle) return event_ref def func_person_sour(self, line, state): source_ref = self.handle_source(line, state.level+1) state.person.add_source_reference(source_ref) def func_person_refn(self, line, state): if INT_RE.match(line.data): try: self.refn[self.person.handle] = int(line.data) except: pass def func_person_attr(self, line, state): attr = RelLib.Attribute() attr.set_type((RelLib.AttributeType.CUSTOM, line.token_text)) attr.set_value(line.data) state.person.add_attribute(attr) def func_person_event(self, line, state): event_ref = self._build_event_pair(state, RelLib.EventType.CUSTOM, self.event_parse_tbl, line.data) state.person.add_event_ref(event_ref) #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- def func_name_undefined(self, line, state): self.not_recognized(state.level+1) def func_name_note(self, line, state): state.add_to_note(self.parse_note(line, state.name, state.level+1, state.note)) def func_name_alia(self, line, state): """ The ALIA tag is supposed to cross reference another person. However, we do not support this. Some systems use the ALIA tag as an alternate NAME tag, which is not legal in GEDCOM, but oddly enough, is easy to support. """ if line.data[0] == '@': aka = RelLib.Name() try: names = NAME_RE.match(line.data).groups() except: names = (line.data, "", "", "", "") if names[0]: aka.set_first_name(names[0].strip()) if names[2]: aka.set_surname(names[2].strip()) if names[4]: aka.set_suffix(names[4].strip()) state.person.add_alternate_name(aka) else: pass def func_name_npfx(self, line, state): state.name.set_title(line.data.strip()) def func_name_givn(self, line, state): state.name.set_first_name(line.data.strip()) def func_name_spfx(self, line, state): state.name.set_surname_prefix(line.data.strip()) def func_name_surn(self, line, state): state.name.set_surname(line.data.strip()) def func_name_marnm(self, line, state): text = line.data.strip() data = text.split() if len(data) == 1: name = RelLib.Name(state.person.primary_name) name.set_surname(data[0].strip()) name.set_type(RelLib.NameType.MARRIED) state.person.add_alternate_name(name) elif len(data) > 1: name = GedcomUtils.parse_name_personal(text) name.set_type(RelLib.NameType.MARRIED) state.person.add_alternate_name(name) def func_name_nsfx(self, line, state): if state.name.get_suffix() == "": state.name.set_suffix(line.data) def func_name_nick(self, line, state): attr = RelLib.Attribute() attr.set_type(RelLib.AttributeType.NICKNAME) attr.set_value(line.data) state.person.add_attribute(attr) def func_name_aka(self, line, state): lname = line.data.split() name_len = len(lname) if name_len == 1: attr = RelLib.Attribute() attr.set_type(RelLib.AttributeType.NICKNAME) attr.set_value(line.data) state.person.add_attribute(attr) else: name = RelLib.Name() name.set_surname(lname[-1].strip()) name.set_first_name(' '.join(lname[0:name_len-1])) state.person.add_alternate_name(name) def func_name_sour(self, line, state): sref = self.handle_source(line, state.level+1) state.name.add_source_reference(sref) def parse_repository(self, repo): state = CurrentState() state.repo = repo while True: line = self.get_next() if self.level_is_finished(line, 1): break else: func = self.repo_parse_tbl.get(line.token, self.func_repo_ignore) func(line, repo, 1) def func_repo_name(self, line, repo, level): repo.set_name(line.data) def func_repo_ignore(self, line, repo, level): self.skip_subordinate_levels(level) def func_repo_addr(self, line, repo, level): """ n ADDR {0:1} +1 CONT {0:M} +1 ADR1 {0:1} +1 ADR2 {0:1} +1 CITY {0:1} +1 STAE {0:1} +1 POST {0:1} +1 CTRY {0:1} n PHON {0:3} Some repositories do not try to break up the address, instead they put everything on a single line. Try to determine if this happened, and try to fix it. """ addr = RelLib.Address() addr.set_street(line.data) self.parse_address(addr, 2) text = addr.get_street() if not addr.get_city() and not addr.get_state() and \ not addr.get_postal_code() and not addr.get_country(): match = ADDR_RE.match(text) if match: groups = match.groups() addr.set_street(groups[0].strip()) addr.set_city(groups[2].strip()) addr.set_state(groups[3].strip()) addr.set_postal_code(groups[4].strip()) addr.set_country(groups[5].strip()) matched = True match = ADDR2_RE.match(text) if match: groups = match.groups() addr.set_street(groups[0].strip()) addr.set_city(groups[2].strip()) addr.set_state(groups[3].strip()) addr.set_postal_code(groups[4].strip()) matched = True match = ADDR3_RE.match(text) if match: groups = match.groups() addr.set_street(groups[0].strip()) addr.set_city(groups[2].strip()) addr.set_state(groups[3].strip()) matched = True repo.add_address(addr) def parse_source(self, name, level): """ n @@ SOUR {1:1} +1 DATA {0:1} +2 EVEN {0:M} +3 DATE {0:1} +3 PLAC {0:1} +2 AGNC {0:1} +2 <> {0:M} +1 AUTH {0:1} +1 TITL {0:1} +1 ABBR {0:1} +1 PUBL {0:1} +1 TEXT {0:1} +1 <> {0:1} +1 <> {0:M} +1 <> {0:M} +1 REFN {0:M} +2 TYPE {0:1} +1 RIN {0:1} +1 <> {0:1} """ self.source = self.find_or_create_source(name) self.source.set_title("No title - ID %s" % self.source.get_gramps_id()) while True: line = self.get_next() if self.level_is_finished(line, level): break else: func = self.source_func.get(line.token, self.func_source_undef) func(line, self.source, level) self.db.commit_source(self.source, self.trans) def func_source_undef(self, line, source, level): self.not_recognized(level+1) def func_source_ignore(self, line, source, level): self.skip_subordinate_levels(level+1) def func_source_repo(self, line, source, level): if line.data and line.data[0] == '@': gid = line.data.strip()[1:-1] repo = self.find_or_create_repository(gid) else: gid = self.repo2id.get(line.data) repo = self.find_or_create_repository(gid) self.repo2id[line.data] = repo.get_gramps_id() repo.set_name(line.data) self.db.commit_repository(repo, self.trans) repo_ref = RelLib.RepoRef() repo_ref.set_reference_handle(repo.handle) self.parse_repo_ref(line, repo_ref, level+1) source.add_repo_reference(repo_ref) def func_source_abbr(self, line, source, level): source.set_abbreviation(line.data) def func_source_agnc(self, line, source, level): attr = RelLib.Attribute() attr.set_type(RelLib.AttributeType.AGENCY) attr.set_value(line.data) source.add_attribute(attr) def func_source_text(self, line, source, level): source.set_note(line.data) def func_source_note(self, line, source, level): note = self.parse_note(line, source, level+1, '') source.set_note(note) def func_source_auth(self, line, source, level): source.set_author(line.data) def func_source_publ(self, line, source, level): source.set_publication_info(line.data) def func_source_title(self, line, source, level): title = line.data title = title.replace('\n',' ') source.set_title(title) def func_source_taxt_peri(self, line, source, level): if source.get_title() == "": source.set_title(line.data.replace('\n',' ')) def func_obje_form(self, line, media, level): self.skip_subordinate_levels(level+1) def func_obje_file(self, line, media, level): (file_ok, filename) = self.find_file(line.data, self.dir_path) if not file_ok: self.warn(_("Could not import %s") % filename[0]) path = filename[0].replace('\\', os.path.sep) media.set_path(path) media.set_mime_type(Mime.get_type(path)) if not media.get_description(): media.set_description(path) def func_obje_ignore(self, line, media, level): self.skip_subordinate_levels(level+1) def func_obje_title(self, line, media, level): media.set_description(line.data) def func_obje_note(self, line, media, level): note = self.parse_note(line, media, level+1, '') media.set_note(note) def func_obje_blob(self, line, media, level): self.skip_subordinate_levels(level+1) def func_obje_refn(self, line, media, level): self.skip_subordinate_levels(level+1) def func_obje_type(self, line, media, level): self.skip_subordinate_levels(level+1) def func_obje_rin(self, line, media, level): self.skip_subordinate_levels(level+1) def func_obje_chan(self, line, media, level): self.skip_subordinate_levels(level+1) def skip_record(self, line, state): self.skip_subordinate_levels(2) def extract_temple(self, line): def get_code(code): if LdsUtils.temple_to_abrev.has_key(code): return code elif LdsUtils.temple_codes.has_key(code): return LdsUtils.temple_codes[code] code = get_code(line.data) if code: return code ## Not sure why we do this. Kind of ugly. code = get_code(line.data.split()[0]) if code: return code ## Okay we have no clue which temple this is. ## We should tell the user and store it anyway. self.warn("Invalid temple code '%s'" % (line.data,)) return line.data def add_default_source(self, obj): if self.use_def_src and len(obj.get_source_references()) == 0: sref = RelLib.SourceRef() sref.set_reference_handle(self.def_src.handle) obj.add_source_reference(sref) def person_event_name(event, person): if event.get_type().is_custom(): if not event.get_description(): text = EVENT_PERSON_STR % { 'event_name' : str(event.get_type()), 'person' : NameDisplay.displayer.display(person), } event.set_description(text) def family_event_name(event, family): if event.get_type().is_custom(): if not event.get_description(): text = EVENT_FAMILY_STR % { 'event_name' : str(event.get_type()), 'family' : "", } event.set_description(text) #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- def create_id(): return Utils.create_id() if __name__ == "__main__": import const import sys import hotshot#, hotshot.stats from GrampsDb import gramps_db_factory def callback(val): pass log_msg_fmt = "%(levelname)s: %(filename)s: line %(lineno)d: %(message)s" form = logging.Formatter(fmt=log_msg_fmt) stderrh = logging.StreamHandler(sys.stderr) stderrh.setFormatter(form) stderrh.setLevel(logging.DEBUG) # Setup the base level logger, this one gets # everything. l = logging.getLogger() l.setLevel(logging.DEBUG) l.addHandler(stderrh) code_set = None db_class = gramps_db_factory(const.app_gramps) database = db_class() database.load("test.grdb", lambda x: None, mode="w") f = open(sys.argv[1],"rU") np = NoteParser(f, False, 0) g = GedcomParser(database, f, sys.argv[1], callback, code_set, np.get_map(), np.get_lines(),np.get_persons()) if False: pr = hotshot.Profile('mystats.profile') print "Start" pr.runcall(g.parse_gedcom_file, False) print "Finished" pr.close() else: t = time.time() g.parse_gedcom_file(False) print time.time() - t database.close()