# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000 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 # from RelLib import * from Researcher import Researcher import string import os from xml.sax import handler #------------------------------------------------------------------------- # # Try to abstract SAX1 from SAX2 # #------------------------------------------------------------------------- try: import xml.sax.saxexts sax = 1 except: sax = 2 from latin_utf8 import utf8_to_latin #------------------------------------------------------------------------- # # Remove extraneous spaces # #------------------------------------------------------------------------- def fix_spaces(text_list): val = "" for text in text_list: val = val + string.join(string.split(text),' ') + "\n" return val #------------------------------------------------------------------------- # # Gramps database parsing class. Derived from SAX XML parser # #------------------------------------------------------------------------- class GrampsParser(handler.ContentHandler): #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def __init__(self,database,callback,base,is_import): self.call = None self.stext_list = [] self.scomments_list = [] self.note_list = [] self.use_p = 0 self.in_note = 0 self.in_attribute = 0 self.in_old_attr = 0 self.in_stext = 0 self.in_scomments = 0 self.in_people = 0 self.db = database self.base = base self.in_family = 0 self.in_source_ref = 0 self.in_source = 0 self.in_address = 0 self.in_event = 0 self.person = None self.family = None self.address = None self.source = None self.source_ref = None self.is_import = is_import self.resname = "" self.resaddr = "" self.rescity = "" self.resstate = "" self.rescon = "" self.respos = "" self.resphone = "" self.resemail = "" self.pmap = {} self.fmap = {} self.smap = {} self.callback = callback self.entries = 0 self.count = 0 self.increment = 50 self.event = None self.name = None self.tempDefault = None self.owner = Researcher() self.data = "" self.func_list = [None]*50 self.func_index = 0 self.func = None handler.ContentHandler.__init__(self) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def setDocumentLocator(self,locator): self.locator = locator #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def endDocument(self): self.db.setResearcher(self.owner) if self.tempDefault != None: id = self.tempDefault if self.db.personMap.has_key(id): person = self.db.personMap[id] self.db.setDefaultPerson(person) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_event(self,attrs): self.event = Event() self.event_type = string.capwords(attrs["type"]) self.in_event = 1 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_attribute(self,attrs): self.in_attribute = 1 self.attribute = Attribute() if attrs.has_key('type'): self.in_old_attr = 1 self.attribute.setType(string.capwords(attrs["type"])) else: self.in_old_attr = 0 if self.in_people: self.person.addAttribute(self.attribute) elif self.in_family: self.family.addAttribute(self.attribute) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_address(self,attrs): self.in_address = 1 self.address = Address() self.person.addAddress(self.address) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_bmark(self,attrs): if self.is_import: person = self.db.findPerson("x%s" % attrs["ref"],self.pmap) else: person = self.db.findPersonNoMap(attrs["ref"]) self.db.bookmarks.append(person) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_person(self,attrs): if self.count % self.increment == 0: self.callback(float(self.count)/float(self.entries)) self.count = self.count + 1 if self.is_import: self.person = self.db.findPerson("x%s" % attrs["id"],self.pmap) else: self.person = self.db.findPersonNoMap(attrs["id"]) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_people(self,attrs): self.in_family = 0 self.in_people = 1 self.in_source = 0 if self.is_import == 0 and attrs.has_key("default"): self.tempDefault = int(attrs["default"]) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_father(self,attrs): if self.is_import: father = self.db.findPerson("x%s" % attrs["ref"],self.pmap) else: father = self.db.findPersonNoMap(attrs["ref"]) self.family.setFather(father) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_mother(self,attrs): if self.is_import: mother = self.db.findPerson("x%s" % attrs["ref"],self.pmap) else: mother = self.db.findPersonNoMap(attrs["ref"]) self.family.setMother(mother) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_child(self,attrs): if self.is_import: child = self.db.findPerson("x%s" % attrs["ref"],self.pmap) else: child = self.db.findPersonNoMap(attrs["ref"]) self.family.addChild(child) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_url(self,attrs): if not attrs.has_key("href"): return try: desc = attrs["description"] except KeyError: desc = "" try: url = Url(attrs["href"],desc) self.person.addUrl(url) except KeyError: return #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_family(self,attrs): if self.count % self.increment == 0: self.callback(float(self.count)/float(self.entries)) self.count = self.count + 1 if self.is_import: self.family = self.db.findFamily(attrs["id"],self.fmap) else: self.family = self.db.findFamilyNoMap(attrs["id"]) if attrs.has_key("type"): self.family.setRelationship(attrs["type"]) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_childof(self,attrs): if self.is_import: family = self.db.findFamily(attrs["ref"],self.fmap) else: family = self.db.findFamilyNoMap(attrs["ref"]) if attrs.has_key("type"): type = attrs["type"] self.person.addAltFamily(family,type) else: self.person.setMainFamily(family) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_parentin(self,attrs): if self.is_import: family = self.db.findFamily(attrs["ref"],self.fmap) else: family = self.db.findFamilyNoMap(attrs["ref"]) self.person.addFamily(family) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_name(self,attrs): self.name = Name() #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_note(self,attrs): self.in_note = 1 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_families(self,attrs): self.in_family = 1 self.in_people = 0 self.in_source = 0 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_sources(self,attrs): self.in_family = 0 self.in_people = 0 self.in_source = 1 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_sourceref(self,attrs): self.source_ref = SourceRef() self.in_source_ref = 1 if self.is_import: self.source = self.db.findSource(attrs["ref"],self.smap) else: self.source = self.db.findSourceNoMap(attrs["ref"]) self.source_ref.setBase(self.source) if self.in_address: self.address.setSourceRef(self.source_ref) elif self.in_event: self.event.setSourceRef(self.source_ref) elif self.in_attribute: self.attribute.setSourceRef(self.source_ref) else: print "Sorry, I'm lost" #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_source(self,attrs): if self.is_import: self.source = self.db.findSource(attrs["id"],self.smap) else: self.source = self.db.findSourceNoMap(attrs["id"]) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_photo(self,attrs): photo = Photo() if attrs.has_key("descrip"): photo.setDescription(attrs["descrip"]) src = attrs["src"] if src[0] != os.sep: photo.setPath("%s%s%s" % (self.base,os.sep,src)) photo.setPrivate(1) else: photo.setPath(src) photo.setPrivate(0) if self.in_family == 1: self.family.addPhoto(photo) if self.in_source == 1: self.source.addPhoto(photo) else: self.person.addPhoto(photo) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def start_created(self,attrs): self.entries = string.atoi(attrs["people"]) + \ string.atoi(attrs["families"]) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_attribute(self,tag): self.in_attribute = 0 if self.in_old_attr: self.attribute.setValue(tag) self.in_old_attr = 0 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_attr_type(self,tag): self.attribute.setType(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_attr_value(self,tag): self.attribute.setValue(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_address(self,tag): self.in_address = 0 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_event(self,tag): self.in_event = 0 self.event.setName(self.event_type) if self.event_type == "Birth": self.person.setBirth(self.event) elif self.event_type == "Death": self.person.setDeath(self.event) elif self.event_type == "Marriage": self.family.setMarriage(self.event) elif self.event_type == "Divorce": self.family.setDivorce(self.event) elif self.in_people == 1: self.person.addEvent(self.event) else: self.family.addEvent(self.event) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_name(self,tag): self.person.setPrimaryName(self.name) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_place(self,tag): self.event.setPlace(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_uid(self,tag): self.person.setPafUid(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_date(self,tag): if tag: if self.in_address: self.address.setDate(tag) else: self.event.getDateObj().quick_set(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_first(self,tag): self.name.setFirstName(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_description(self,tag): self.event.setDescription(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_gender(self,tag): if tag == "M": self.person.setGender(Person.male) else: self.person.setGender(Person.female) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_stitle(self,tag): self.source.setTitle(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_sourceref(self,tag): self.in_source_ref = 0 #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_sauthor(self,tag): self.source.setAuthor(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_sdate(self,tag): date = Date() date.quick_set(tag) self.source_ref.setDate(date) def stop_street(self,tag): self.address.setStreet(tag) def stop_city(self,tag): self.address.setCity(tag) def stop_state(self,tag): self.address.setState(tag) def stop_country(self,tag): self.address.setCountry(tag) def stop_postal(self,tag): self.address.setPostal(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_spage(self,tag): self.source_ref.setPage(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_spubinfo(self,tag): self.source.setPubInfo(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_scallno(self,tag): self.source.setCallNumber(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_stext(self,tag): if self.use_p: self.use_p = 0 note = fix_spaces(self.stext_list) else: note = tag self.source_ref.setText(note) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_scomments(self,tag): if self.use_p: self.use_p = 0 note = fix_spaces(self.scomments_list) else: note = tag self.source_ref.setComments(note) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_last(self,tag): self.name.setSurname(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_suffix(self,tag): self.name.setSuffix(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_title(self,tag): self.name.setTitle(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_nick(self,tag): self.person.setNickName(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_note(self,tag): self.in_note = 0 if self.use_p: self.use_p = 0 note = fix_spaces(self.note_list) else: note = tag if self.in_address == 1: self.address.setNote(note) if self.in_attribute == 1: self.attribute.setNote(note) elif self.in_source == 1: self.source.setNote(note) elif self.in_event == 1: self.event.setNote(note) elif self.in_people == 1: self.person.setNote(note) elif self.in_family == 1: self.family.setNote(note) self.note_list = [] #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_research(self,tag): self.owner.set(self.resname, self.resaddr, self.rescity, self.resstate, self.rescon, self.respos, self.resphone, self.resemail) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_resname(self,tag): self.resname = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_resaddr(self,tag): self.resaddr = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_rescity(self,tag): self.rescity = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_resstate(self,tag): self.resstate = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_rescountry(self,tag): self.rescon = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_respostal(self,tag): self.respos = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_resphone(self,tag): self.resphone = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_resemail(self,tag): self.resemail = tag #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_ptag(self,tag): self.use_p = 1 if self.in_note: self.note_list.append(tag) elif self.in_stext: self.stext_list.append(tag) elif self.in_scomments: self.scomments_list.append(tag) #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def stop_aka(self,tag): self.person.addAlternateName(self.name) func_map = { "address" : (start_address, stop_address), "aka" : (start_name, stop_aka), "attribute" : (start_attribute, stop_attribute), "attr_type" : (None,stop_attr_type), "attr_value" : (None,stop_attr_value), "bookmark" : (start_bmark, None), "bookmarks" : (None, None), "child" : (start_child,None), "childof" : (start_childof,None), "childlist" : (None,None), "city" : (None, stop_city), "country" : (None, stop_country), "created" : (start_created, None), "database" : (None, None), "date" : (None, stop_date), "description": (None, stop_description), "event" : (start_event, stop_event), "families" : (start_families, None), "family" : (start_family, None), "father" : (start_father, None), "first" : (None, stop_first), "gender" : (None, stop_gender), "header" : (None, None), "last" : (None, stop_last), "mother" : (start_mother,None), "name" : (start_name, stop_name), "nick" : (None, stop_nick), "note" : (start_note, stop_note), "p" : (None, stop_ptag), "parentin" : (start_parentin,None), "people" : (start_people, None), "person" : (start_person, None), "img" : (start_photo, None), "place" : (None, stop_place), "postal" : (None, stop_postal), "researcher" : (None, stop_research), "resname" : (None, stop_resname ), "resaddr" : (None, stop_resaddr ), "rescity" : (None, stop_rescity ), "resstate" : (None, stop_resstate ), "rescountry" : (None, stop_rescountry), "respostal" : (None, stop_respostal), "resphone" : (None, stop_resphone), "resemail" : (None, stop_resemail), "sauthor" : (None, stop_sauthor), "scallno" : (None, stop_scallno), "scomments" : (None, stop_scomments), "sdate" : (None,stop_sdate), "source" : (start_source, None), "sourceref" : (start_sourceref, stop_sourceref), "sources" : (start_sources, None), "spage" : (None, stop_spage), "spubinfo" : (None, stop_spubinfo), "state" : (None, stop_state), "stext" : (None, stop_stext), "stitle" : (None, stop_stitle), "street" : (None, stop_street), "suffix" : (None, stop_suffix), "title" : (None, stop_title), "uid" : (None, stop_uid), "url" : (start_url, None) } #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- def startElement(self,tag,attrs): self.func_list[self.func_index] = (self.func,self.data) self.func_index = self.func_index + 1 self.data = "" try: f,self.func = GrampsParser.func_map[tag] if f: f(self,attrs) except: GrampsParser.func_map[tag] = (None,None) self.func = None #--------------------------------------------------------------------- # # # #--------------------------------------------------------------------- if sax == 1: def endElement(self,tag): if self.func: self.func(self,utf8_to_latin(self.data)) self.func_index = self.func_index - 1 self.func,self.data = self.func_list[self.func_index] def characters(self, data, offset, length): if self.func: self.data = self.data + data else: def endElement(self,tag): if self.func: self.func(self,self.data) self.func_index = self.func_index - 1 self.func,self.data = self.func_list[self.func_index] def characters(self, data): if self.func: self.data = self.data + data