# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2002-2005 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$ """Generic Filtering Routines""" __author__ = "Don Allingham" #------------------------------------------------------------------------- # # Try to abstract SAX1 from SAX2 # #------------------------------------------------------------------------- from xml.sax import make_parser,handler,SAXParseException #------------------------------------------------------------------------- # # Standard Python modules # #------------------------------------------------------------------------- import os from gettext import gettext as _ #------------------------------------------------------------------------- # # gtk # #------------------------------------------------------------------------- import gtk #------------------------------------------------------------------------- # # GRAMPS modules # #------------------------------------------------------------------------- import const import RelLib import Date import DateHandler import NameDisplay from TransTable import TransTable from Utils import for_each_ancestor,probably_alive,get_source_referents #------------------------------------------------------------------------- # # date_cmp # #------------------------------------------------------------------------- def date_cmp(rule,value): sd = rule.get_start_date() s = rule.get_modifier() od = value.get_start_date() cmp_rule = (sd[2],sd[1],sd[0]) cmp_value = (od[2],od[1],od[0]) if s == Date.MOD_BEFORE: return cmp_rule > cmp_value elif s == Date.MOD_AFTER: return cmp_rule < cmp_value else: return cmp_rule == cmp_value #------------------------------------------------------------------------- # # Rule # #------------------------------------------------------------------------- class Rule: """Base rule class""" labels = [] name = '' category = _('Miscellaneous filters') description = _('No description') def __init__(self,list): self.set_list(list) def prepare(self,db): pass def reset(self): pass def set_list(self,list): assert type(list) == type([]) or list == None, "Argument is not a list" self.list = list def values(self): return self.list def check(self): return len(self.list) == len(self.labels) def apply(self,db,handle): return 1 def display_values(self): v = [] for i in range(0,len(self.list)): if self.list[i]: v.append('%s="%s"' % (_(self.labels[i]),_(self.list[i]))) return ';'.join(v) #------------------------------------------------------------------------- # # Everyone # #------------------------------------------------------------------------- class Everyone(Rule): """Matches Everyone""" name = _('Everyone') category = _('General filters') description = _('Matches everyone in the database') def apply(self,db,handle): return 1 #------------------------------------------------------------------------- # # Disconnected # #------------------------------------------------------------------------- class Disconnected(Rule): """Matches disconnected people""" name = _('Disconnected people') category = _('General filters') description = _('Matches people that have no family relationships ' 'to any other person in the database') def apply(self,db,handle): person = db.get_person_from_handle(handle) return (not person.get_main_parents_family_handle() and not len(person.get_family_handle_list())) #------------------------------------------------------------------------- # # RelationshipPathBetween # #------------------------------------------------------------------------- class RelationshipPathBetween(Rule): """Rule that checks for a person that is a descendant of a specified person not more than N generations away""" labels = [ _('ID:'), _('ID:') ] name = _("Relationship path between ") category = _('Relationship filters') description = _("Matches the ancestors of two persons back to a common ancestor, " "producing the relationship path between two persons.") def prepare(self,db): self.db = db self.map = {} try: root1_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() root2_handle = db.get_person_from_gramps_id(self.list[1]).get_handle() self.init_list(root1_handle,root2_handle) except: pass def reset(self): self.map = {} def desc_list(self, handle, map, first): if not first: map[handle] = 1 p = self.db.get_person_from_handle(handle) for fam_id in p.get_family_handle_list(): fam = self.db.get_family_from_handle(fam_id) if fam: for child_handle in fam.get_child_handle_list(): if child_handle: self.desc_list(child_handle,map,0) def apply_filter(self,rank,handle,plist,pmap): person = self.db.get_person_from_handle(handle) if person == None: return plist.append(person) pmap[person.get_handle()] = rank fam_id = person.get_main_parents_family_handle() family = self.db.get_family_from_handle(fam_id) if family != None: self.apply_filter(rank+1,family.get_father_handle(),plist,pmap) self.apply_filter(rank+1,family.get_mother_handle(),plist,pmap) def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,p1_handle,p2_handle): firstMap = {} firstList = [] secondMap = {} secondList = [] common = [] rank = 9999999 self.apply_filter(0,p1_handle,firstList,firstMap) self.apply_filter(0,p2_handle,secondList,secondMap) for person_handle in firstList: if person_handle in secondList: new_rank = firstMap[person_handle] if new_rank < rank: rank = new_rank common = [ person_handle ] elif new_rank == rank: common.append(person_handle) path1 = { p1_handle : 1} path2 = { p2_handle : 1} for person_handle in common: new_map = {} self.desc_list(person_handle,new_map,1) self.get_intersection(path1,firstMap,new_map) self.get_intersection(path2,secondMap,new_map) for e in path1: self.map[e] = 1 for e in path2: self.map[e] = 1 for e in common: self.map[e] = 1 def get_intersection(self,target, map1, map2): for e in map1.keys(): if map2.has_key(e): target[e] = map2[e] #------------------------------------------------------------------------- # # HasIdOf # #------------------------------------------------------------------------- class HasIdOf(Rule): """Rule that checks for a person with a specific GRAMPS ID""" labels = [ _('ID:') ] name = _('People with ') description = _("Matches people with a specified GRAMPS ID") category = _('General filters') def apply(self,db,handle): return db.get_person_from_handle(handle).get_gramps_id() == self.list[0] #------------------------------------------------------------------------- # # HasIdOf # #------------------------------------------------------------------------- class IsDefaultPerson(Rule): """Rule that checks for a default person in the database""" name = _('Default person') category = _('General filters') description = _("Matches the default person") def apply(self,db,handle): person = db.get_default_person() if person: return handle == person.get_handle() return 0 #------------------------------------------------------------------------- # # IsBookmarked # #------------------------------------------------------------------------- class IsBookmarked(Rule): """Rule that checks for the bookmark list in the database""" name = _('Bookmarked people') category = _('General filters') description = _("Matches the people on the bookmark list") def apply(self,db,handle): if handle in db.get_bookmarks(): return 1 return 0 #------------------------------------------------------------------------- # # HasCompleteRecord # #------------------------------------------------------------------------- class HasCompleteRecord(Rule): """Rule that checks for a person whose record is complete""" name = _('People with complete records') category = _('General filters') description = _('Matches all people whose records are complete') def apply(self,db,handle): return db.get_person_from_handle(handle).get_complete_flag() == 1 #------------------------------------------------------------------------- # # IsFemale # #------------------------------------------------------------------------- class IsFemale(Rule): """Rule that checks for a person that is a female""" name = _('Females') category = _('General filters') description = _('Matches all females') def apply(self,db,handle): return db.get_person_from_handle(handle).get_gender() == RelLib.Person.FEMALE #------------------------------------------------------------------------- # # HasUnknownGender # #------------------------------------------------------------------------- class HasUnknownGender(Rule): """Rule that checks for a person that has unknown gender""" name = _('People with unknown gender') category = _('General filters') description = _('Matches all people with unknown gender') def apply(self,db,handle): return db.get_person_from_handle(handle).get_gender() == RelLib.Person.UNKNOWN #------------------------------------------------------------------------- # # IsDescendantOf # #------------------------------------------------------------------------- class IsDescendantOf(Rule): """Rule that checks for a person that is a descendant of a specified person""" labels = [ _('ID:'), _('Inclusive:') ] name = _('Descendants of ') category = _('Descendant filters') description = _('Matches all descendants for the specified person') def prepare(self,db): self.db = db self.map = {} try: if int(self.list[1]): first = 0 else: first = 1 except IndexError: first = 1 try: root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() self.init_list(root_handle,first) except: pass def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,handle,first): if not handle: return if not first: self.map[handle] = 1 p = self.db.get_person_from_handle(handle) for fam_id in p.get_family_handle_list(): fam = self.db.get_family_from_handle(fam_id) if fam: for child_handle in fam.get_child_handle_list(): self.init_list(child_handle,0) #------------------------------------------------------------------------- # # IsDescendantOfFilterMatch # #------------------------------------------------------------------------- class IsDescendantOfFilterMatch(IsDescendantOf): """Rule that checks for a person that is a descendant of someone matched by a filter""" labels = [ _('Filter name:'), _('Inclusive:') ] name = _('Descendants of match') category = _('Descendant filters') description = _("Matches people that are descendants of anybody matched by a filter") def __init__(self,list): IsDescendantOf.__init__(self,list) def prepare(self,db): self.db = db self.map = {} try: if int(self.list[1]): first = 0 else: first = 1 except IndexError: first = 1 filt = MatchesFilter(self.list) filt.prepare(db) for person_handle in db.get_person_handles(sort_handles=False): if filt.apply (db, person_handle): self.init_list (person_handle,first) filt.reset() def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) #------------------------------------------------------------------------- # # IsLessThanNthGenerationDescendantOf # #------------------------------------------------------------------------- class IsLessThanNthGenerationDescendantOf(Rule): """Rule that checks for a person that is a descendant of a specified person not more than N generations away""" labels = [ _('ID:'), _('Number of generations:') ] name = _('Descendants of not more than generations away') category = _('Descendant filters') description = _("Matches people that are descendants of a specified person " "not more than N generations away") def prepare(self,db): self.db = db self.map = {} try: root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() self.init_list(root_handle,0) except: pass def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,handle,gen): if not handle: return if gen: self.map[handle] = 1 if gen >= int(self.list[1]): return p = self.db.get_person_from_handle(handle) for fam_id in p.get_family_handle_list(): fam = self.db.get_family_from_handle(fam_id) for child_handle in fam.get_child_handle_list(): self.init_list(child_handle,gen+1) #------------------------------------------------------------------------- # # IsMoreThanNthGenerationDescendantOf # #------------------------------------------------------------------------- class IsMoreThanNthGenerationDescendantOf(Rule): """Rule that checks for a person that is a descendant of a specified person at least N generations away""" labels = [ _('ID:'), _('Number of generations:') ] name = _('Descendants of at least generations away') category = _("Descendant filters") description = _("Matches people that are descendants of a specified " "person at least N generations away") def prepare(self,db): self.db = db self.map = {} try: root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() self.init_list(root_handle,0) except: pass def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,handle,gen): if not handle: return if gen >= int(self.list[1]): self.map[handle] = 1 p = self.db.get_person_from_handle(handle) for fam_id in p.get_family_handle_list(): fam = self.db.get_family_from_handle(fam_id) for child_handle in fam.get_child_handle_list(): self.init_list(child_handle,gen+1) #------------------------------------------------------------------------- # # IsChildOfFilterMatch # #------------------------------------------------------------------------- class IsChildOfFilterMatch(Rule): """Rule that checks for a person that is a child of someone matched by a filter""" labels = [ _('Filter name:') ] name = _('Children of match') category = _('Family filters') description = _("Matches children of anybody matched by a filter") def prepare(self,db): self.db = db self.map = {} filt = MatchesFilter(self.list) filt.prepare(db) for person_handle in db.get_person_handles(sort_handles=False): if filt.apply (db, person_handle): self.init_list (person_handle) filt.reset() def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,handle): if not handle: return p = self.db.get_person_from_handle(handle) for fam_id in p.get_family_handle_list(): fam = self.db.get_family_from_handle(fam_id) for child_handle in fam.get_child_handle_list(): self.map[child_handle] = 1 #------------------------------------------------------------------------- # # IsSiblingOfFilterMatch # #------------------------------------------------------------------------- class IsSiblingOfFilterMatch(Rule): """Rule that checks for siblings of someone matched by a filter""" labels = [ _('Filter name:') ] name = _('Siblings of match') category = _('Family filters') description = _("Matches siblings of anybody matched by a filter") def prepare(self,db): self.db = db self.map = {} filt = MatchesFilter(self.list) filt.prepare(db) for person_handle in db.get_person_handles(sort_handles=False): if filt.apply (db, person_handle): self.init_list (person_handle) filt.reset() def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,handle): if not handle: return p = self.db.get_person_from_handle(handle) fam_id = p.get_main_parents_family_handle() fam = self.db.get_family_from_handle(fam_id) if fam: for child_handle in fam.get_child_handle_list(): if child_handle != handle: self.map[child_handle] = 1 #------------------------------------------------------------------------- # # IsDescendantFamilyOf # #------------------------------------------------------------------------- class IsDescendantFamilyOf(Rule): """Rule that checks for a person that is a descendant or the spouse of a descendant of a specified person""" labels = [ _('ID:') ] name = _('Descendant family members of ') category = _('Descendant filters') description = _("Matches people that are descendants or the spouse " "of a descendant of a specified person") def apply(self,db,handle): self.map = {} self.orig_handle = handle self.db = db return self.search(handle,1) def search(self,handle,val): try: if handle == self.db.get_person_from_gramps_id(self.list[0]).get_handle(): self.map[handle] = 1 return 1 except: return 0 p = self.db.get_person_from_handle(handle) for (f,r1,r2) in p.get_parent_family_handle_list(): family = self.db.get_family_from_handle(f) for person_handle in [family.get_mother_handle(),family.get_father_handle()]: if person_handle: if self.search(person_handle,0): return 1 if val: for family_handle in p.get_family_handle_list(): family = self.db.get_family_from_handle(family_handle) if handle == family.get_father_handle(): spouse_id = family.get_mother_handle() else: spouse_id = family.get_father_handle() if spouse_id: if self.search(spouse_id,0): return 1 return 0 #------------------------------------------------------------------------- # # IsAncestorOf # #------------------------------------------------------------------------- class IsAncestorOf(Rule): """Rule that checks for a person that is an ancestor of a specified person""" labels = [ _('ID:'), _('Inclusive:') ] name = _('Ancestors of ') category = _("Ancestral filters") description = _("Matches people that are ancestors of a specified person") def prepare(self,db): """Assume that if 'Inclusive' not defined, assume inclusive""" self.db = db self.map = {} try: if int(self.list[1]): first = 0 else: first = 1 except IndexError: first = 1 try: root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() self.init_ancestor_list(db,root_handle,first) except: pass def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_ancestor_list(self,db,handle,first): if not handle: return if not first: self.map[handle] = 1 p = db.get_person_from_handle(handle) fam_id = p.get_main_parents_family_handle() fam = db.get_family_from_handle(fam_id) if fam: f_id = fam.get_father_handle() m_id = fam.get_mother_handle() if f_id: self.init_ancestor_list(db,f_id,0) if m_id: self.init_ancestor_list(db,m_id,0) #------------------------------------------------------------------------- # # IsAncestorOfFilterMatch # #------------------------------------------------------------------------- class IsAncestorOfFilterMatch(IsAncestorOf): """Rule that checks for a person that is an ancestor of someone matched by a filter""" labels = [ _('Filter name:'), _('Inclusive:') ] name = _('Ancestors of match') category = _("Ancestral filters") description = _("Matches people that are ancestors " "of anybody matched by a filter") def __init__(self,list): IsAncestorOf.__init__(self,list) def prepare(self,db): self.db = db self.map = {} try: if int(self.list[1]): first = 0 else: first = 1 except IndexError: first = 1 filt = MatchesFilter(self.list) filt.prepare(db) for person_handle in db.get_person_handles(sort_handles=False): if filt.apply (db, person_handle): self.init_ancestor_list (db,person_handle,first) filt.reset() def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) #------------------------------------------------------------------------- # # IsLessThanNthGenerationAncestorOf # #------------------------------------------------------------------------- class IsLessThanNthGenerationAncestorOf(Rule): """Rule that checks for a person that is an ancestor of a specified person not more than N generations away""" labels = [ _('ID:'), _('Number of generations:') ] name = _('Ancestors of not more than generations away') category = _("Ancestral filters") description = _("Matches people that are ancestors " "of a specified person not more than N generations away") def prepare(self,db): self.db = db self.map = {} try: root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() self.init_ancestor_list(root_handle,0) except: pass def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_ancestor_list(self,handle,gen): # if self.map.has_key(p.get_handle()) == 1: # loop_error(self.orig,p) if not handle: return if gen: self.map[handle] = 1 if gen >= int(self.list[1]): return p = self.db.get_person_from_handle(handle) fam_id = p.get_main_parents_family_handle() fam = self.db.get_family_from_handle(fam_id) if fam: f_id = fam.get_father_handle() m_id = fam.get_mother_handle() if f_id: self.init_ancestor_list(f_id,gen+1) if m_id: self.init_ancestor_list(m_id,gen+1) #------------------------------------------------------------------------- # # IsMoreThanNthGenerationAncestorOf # #------------------------------------------------------------------------- class IsMoreThanNthGenerationAncestorOf(Rule): """Rule that checks for a person that is an ancestor of a specified person at least N generations away""" labels = [ _('ID:'), _('Number of generations:') ] name = _('Ancestors of at least generations away') category = _("Ancestral filters") description = _("Matches people that are ancestors " "of a specified person at least N generations away") def prepare(self,db): self.db = db self.map = {} try: root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle() self.init_ancestor_list(root_handle,0) except: pass def reset(self): self.map = [] def apply(self,db,handle): return self.map.has_key(handle) def init_ancestor_list(self,handle,gen): # if self.map.has_key(p.get_handle()) == 1: # loop_error(self.orig,p) if not handle: return if gen >= int(self.list[1]): self.map[handle] = 1 p = self.db.get_person_from_handle(handle) fam_id = p.get_main_parents_family_handle() fam = self.db.get_family_from_handle(fam_id) if fam: f_id = fam.get_father_handle() m_id = fam.get_mother_handle() if f_id: self.init_ancestor_list(f_id,gen+1) if m_id: self.init_ancestor_list(m_id,gen+1) #------------------------------------------------------------------------- # # IsParentOfFilterMatch # #------------------------------------------------------------------------- class IsParentOfFilterMatch(Rule): """Rule that checks for a person that is a parent of someone matched by a filter""" labels = [ _('Filter name:') ] name = _('Parents of match') category = _('Family filters') description = _("Matches parents of anybody matched by a filter") def prepare(self,db): self.db = db self.map = {} filt = MatchesFilter(self.list) filt.prepare(db) for person_handle in db.get_person_handles(sort_handles=False): if filt.apply (db, person_handle): self.init_list (person_handle) filt.reset() def reset(self): self.map = {} def apply(self,db,handle): return self.map.has_key(handle) def init_list(self,handle): p = self.db.get_person_from_handle(handle) for fam_id,frel,mrel in p.get_parent_family_handle_list(): fam = self.db.get_family_from_handle(fam_id) for parent_id in [fam.get_father_handle (), fam.get_mother_handle ()]: if parent_id: self.map[parent_id] = 1 #------------------------------------------------------------------------- # # HasCommonAncestorWith # #------------------------------------------------------------------------- class HasCommonAncestorWith(Rule): """Rule that checks for a person that has a common ancestor with a specified person""" labels = [ _('ID:') ] name = _('People with a common ancestor with ') category = _("Ancestral filters") description = _("Matches people that have a common ancestor " "with a specified person") def prepare(self,db): self.db = db # Keys in `ancestor_cache' are ancestors of list[0]. # We delay the computation of ancestor_cache until the # first use, because it's not uncommon to instantiate # this class and not use it. self.ancestor_cache = {} def reset(self): self.ancestor_cache = {} def init_ancestor_cache(self,db): # list[0] is an Id, but we need to pass a Person to for_each_ancestor. handle = db.get_person_from_gramps_id(self.list[0]).get_handle() if handle: def init(self,handle): self.ancestor_cache[handle] = 1 for_each_ancestor(db,[handle],init,self) def apply(self,db,handle): # On the first call, we build the ancestor cache for the # reference person. Then, for each person to test, # we browse his ancestors until we found one in the cache. if len(self.ancestor_cache) == 0: self.init_ancestor_cache(db) return for_each_ancestor(db,[handle], lambda self,handle: self.ancestor_cache.has_key(handle), self); #------------------------------------------------------------------------- # # HasCommonAncestorWithFilterMatch # #------------------------------------------------------------------------- class HasCommonAncestorWithFilterMatch(HasCommonAncestorWith): """Rule that checks for a person that has a common ancestor with someone matching a filter""" labels = [ _('Filter name:') ] name = _('People with a common ancestor with match') description = _("Matches people that have a common ancestor " "with anybody matched by a filter") category = _("Ancestral filters") def __init__(self,list): HasCommonAncestorWith.__init__(self,list) def init_ancestor_cache(self,db): filt = MatchesFilter(self.list) filt.prepare(db) def init(self,h): self.ancestor_cache[h] = 1 for handle in db.get_person_handles(sort_handles=False): if (not self.ancestor_cache.has_key (handle) and filt.apply (db, handle)): for_each_ancestor(db,[handle],init,self) filt.reset() #------------------------------------------------------------------------- # # IsMale # #------------------------------------------------------------------------- class IsMale(Rule): """Rule that checks for a person that is a male""" name = _('Males') category = _('General filters') description = _('Matches all males') def apply(self,db,handle): return db.get_person_from_handle(handle).get_gender() == RelLib.Person.MALE #------------------------------------------------------------------------- # # HasEvent # #------------------------------------------------------------------------- class HasEvent(Rule): """Rule that checks for a person with a particular value""" labels = [ _('Personal event:'), _('Date:'), _('Place:'), _('Description:') ] name = _('People with the personal ') description = _("Matches people with a personal event of a particular value") category = _('Event filters') def prepare(self,list): self.date = None try: if self.list and self.list[1]: self.date = DateHandler.parser.parse(self.list[1]) except: pass def apply(self,db,handle): p = db.get_person_from_handle(handle) for event_handle in p.get_event_list(): if not event_handle: continue event = db.get_event_from_handle(event_handle) val = 1 if self.list[0] and event.get_name() != self.list[0]: val = 0 if self.list[3] and event.get_description().upper().find( self.list[3].upper())==-1: val = 0 if self.date: if date_cmp(self.date,event.get_date_object()): val = 0 if self.list[2]: pl_id = event.get_place_handle() if pl_id: pl = db.get_place_from_handle(pl_id) pn = pl.get_title() if pn.upper().find(self.list[2].upper()) == -1: val = 0 if val == 1: return 1 return 0 #------------------------------------------------------------------------- # # HasFamilyEvent # #------------------------------------------------------------------------- class HasFamilyEvent(Rule): """Rule that checks for a person who has a relationship event with a particular value""" labels = [ _('Family event:'), _('Date:'), _('Place:'), _('Description:') ] name = _('People with the family ') description = _("Matches people with a family event of a particular value") category = _('Event filters') def prepare(self,list): self.date = None try: if self.list and self.list[1]: self.date = DateHandler.parser.parse(self.list[1]) except: pass def apply(self,db,handle): p = db.get_person_from_handle(handle) for f_id in p.get_family_handle_list(): f = db.get_family_from_handle(f_id) for event_handle in f.get_event_list(): if not event_handle: continue event = db.get_event_from_handle(event_handle) val = 1 if self.list[0] and event.get_name() != self.list[0]: val = 0 v = self.list[3] if v and event.get_description().upper().find(v.upper())==-1: val = 0 if self.date: if date_cmp(self.date,event.get_date_object()): val = 0 pl_id = event.get_place_handle() if pl_id: pl = db.get_place_from_handle(pl_id) pn = pl.get_title() if self.list[2] and pn.find(self.list[2].upper()) == -1: val = 0 if val == 1: return 1 return 0 #------------------------------------------------------------------------- # # HasRelationship # #------------------------------------------------------------------------- class HasRelationship(Rule): """Rule that checks for a person who has a particular relationship""" labels = [ _('Number of relationships:'), _('Relationship type:'), _('Number of children:') ] name = _('People with the ') description = _("Matches people with a particular relationship") category = _('Family filters') def apply(self,db,handle): rel_type = 0 cnt = 0 p = db.get_person_from_handle(handle) num_rel = len(p.get_family_handle_list()) # count children and look for a relationship type match for f_id in p.get_family_handle_list(): f = db.get_family_from_handle(f_id) cnt = cnt + len(f.get_child_handle_list()) if self.list[1] and int(self.list[1]) == f.get_relationship(): rel_type = 1 # if number of relations specified if self.list[0]: try: v = int(self.list[0]) except: return 0 if v != num_rel: return 0 # number of childred if self.list[2]: try: v = int(self.list[2]) except: return 0 if v != cnt: return 0 # relation if self.list[1]: return rel_type == 1 else: return 1 #------------------------------------------------------------------------- # # HasBirth # #------------------------------------------------------------------------- class HasBirth(Rule): """Rule that checks for a person with a birth of a particular value""" labels = [ _('Date:'), _('Place:'), _('Description:') ] name = _('People with the ') description = _("Matches people with birth data of a particular value") category = _('Event filters') def __init__(self,list): Rule.__init__(self,list) if self.list and self.list[0]: self.date = DateHandler.parser.parse(self.list[0]) else: self.date = None def apply(self,db,handle): p = db.get_person_from_handle(handle) event_handle = p.get_birth_handle() if not event_handle: return False event = db.get_event_from_handle(event_handle) ed = event.get_description().upper() if len(self.list) > 2 and ed.find(self.list[2].upper())==-1: return False if self.date: if date_cmp(self.date,event.get_date_object()) == 0: return False pl_id = event.get_place_handle() if pl_id: pl = db.get_place_from_handle(pl_id) pn = pl.get_title() if len(self.list) > 1 and pn.find(self.list[1].upper()) == -1: return False return True #------------------------------------------------------------------------- # # HasDeath # #------------------------------------------------------------------------- class HasDeath(Rule): """Rule that checks for a person with a death of a particular value""" labels = [ _('Date:'), _('Place:'), _('Description:') ] name = _('People with the ') description = _("Matches people with death data of a particular value") category = _('Event filters') def __init__(self,list): Rule.__init__(self,list) if self.list and self.list[0]: self.date = DateHandler.parser.parse(self.list[0]) else: self.date = None def apply(self,db,handle): p = db.get_person_from_handle(handle) event_handle = p.get_death_handle() if not event_handle: return 0 event = db.get_event_from_handle(event_handle) ed = event.get_description().upper() if self.list[2] and ed.find(self.list[2].upper())==-1: return 0 if self.date: if date_cmp(self.date,event.get_date_object()) == 0: return 0 pl_id = event.get_place_handle() if pl_id: pl = db.get_place_from_handle(pl_id) pn = pl.get_title() if self.list[1] and pn.find(self.list[1].upper()) == -1: return 0 return 1 #------------------------------------------------------------------------- # # HasAttribute # #------------------------------------------------------------------------- class HasAttribute(Rule): """Rule that checks for a person with a particular personal attribute""" labels = [ _('Personal attribute:'), _('Value:') ] name = _('People with the personal ') description = _("Matches people with the personal attribute of a particular value") category = _('General filters') def apply(self,db,handle): if not self.list[0]: return 0 p = db.get_person_from_handle(handle) for attr in p.get_attribute_list(): name_match = self.list[0] == attr.get_type() value_match = \ attr.get_value().upper().find(self.list[1].upper()) != -1 if name_match and value_match: return 1 return 0 #------------------------------------------------------------------------- # # HasFamilyAttribute # #------------------------------------------------------------------------- class HasFamilyAttribute(Rule): """Rule that checks for a person with a particular family attribute""" labels = [ _('Family attribute:'), _('Value:') ] name = _('People with the family ') description = _("Matches people with the family attribute of a particular value") category = _('General filters') def apply(self,db,handle): if not self.list[0]: return 0 p = db.get_person_from_handle(handle) for f_id in p.get_family_handle_list(): f = db.get_family_from_handle(f_id) for attr in f.get_attribute_list(): name_match = self.list[0] == attr.get_type() value_match = \ attr.get_value().upper().find(self.list[1].upper()) != -1 if name_match and value_match: return 1 return 0 #------------------------------------------------------------------------- # # HasNameOf # #------------------------------------------------------------------------- class HasNameOf(Rule): """Rule that checks for full or partial name matches""" labels = [ _('Given name:'), _('Family name:'), _('Suffix:'), _('Title:')] name = _('People with the ') description = _("Matches people with a specified (partial) name") category = _('General filters') def apply(self,db,handle): self.f = self.list[0] self.l = self.list[1] self.s = self.list[2] self.t = self.list[3] p = db.get_person_from_handle(handle) for name in [p.get_primary_name()] + p.get_alternate_names(): val = 1 if self.f and name.get_first_name().upper().find(self.f.upper()) == -1: val = 0 if self.l and name.get_surname().upper().find(self.l.upper()) == -1: val = 0 if self.s and name.get_suffix().upper().find(self.s.upper()) == -1: val = 0 if self.t and name.get_title().upper().find(self.t.upper()) == -1: val = 0 if val == 1: return 1 return 0 #------------------------------------------------------------------------- # # HasNameOf # #------------------------------------------------------------------------- class SearchName(Rule): """Rule that checks for full or partial name matches""" labels = [_('Substring:')] name = _('People matching the ') description = _("Matches people with a specified (partial) name") category = _('General filters') def apply(self,db,handle): self.f = self.list[0] p = db.get_person_from_handle(handle) n = NameDisplay.displayer.display(p) return self.f and n.upper().find(self.f.upper()) != -1 #------------------------------------------------------------------------- # # IncompleteNames # #------------------------------------------------------------------------- class IncompleteNames(Rule): """People with incomplete names""" name = _('People with incomplete names') description = _("Matches people with firstname or lastname missing") category = _('General filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) for name in [p.get_primary_name()] + p.get_alternate_names(): if name.get_first_name() == "": return 1 if name.get_surname() == "": return 1 return 0 #------------------------------------------------------------------------- # # MatchesFilter # #------------------------------------------------------------------------- class MatchesFilter(Rule): """Rule that checks against another filter""" labels = [_('Filter name:')] name = _('People matching the ') description = _("Matches people macthed by the specified filter name") category = _('General filters') def prepare(self,db): for filt in SystemFilters.get_filters(): if filt.get_name() == self.list[0]: for rule in filt.flist: rule.prepare(db) for filt in CustomFilters.get_filters(): if filt.get_name() == self.list[0]: for rule in filt.flist: rule.prepare(db) def reset(self): for filt in SystemFilters.get_filters(): if filt.get_name() == self.list[0]: for rule in filt.flist: rule.reset() for filt in CustomFilters.get_filters(): if filt.get_name() == self.list[0]: for rule in filt.flist: rule.reset() def apply(self,db,handle): for filt in SystemFilters.get_filters(): if filt.get_name() == self.list[0]: return filt.check(handle) for filt in CustomFilters.get_filters(): if filt.get_name() == self.list[0]: return filt.check(db,handle) return 0 #------------------------------------------------------------------------- # # IsSpouseOfFilterMatch # #------------------------------------------------------------------------- class IsSpouseOfFilterMatch(Rule): """Rule that checks for a person married to someone matching a filter""" labels = [_('Filter name:')] name = _('Spouses of match') description = _("Matches people married to anybody matching a filter") category = _('Family filters') def apply(self,db,handle): filt = MatchesFilter (self.list) p = db.get_person_from_handle(handle) for family_handle in p.get_family_handle_list (): family = db.get_family_from_handle(family_handle) for spouse_id in [family.get_father_handle (), family.get_mother_handle ()]: if not spouse_id: continue if spouse_id == handle: continue if filt.apply (db, spouse_id): return 1 return 0 #------------------------------------------------------------------------- # "People who were adopted" #------------------------------------------------------------------------- class HaveAltFamilies(Rule): """People who were adopted""" name = _('Adopted people') description = _("Matches people who were adopted") category = _('Family filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) for (fam,rel1,rel2) in p.get_parent_family_handle_list(): if rel1 == RelLib.Person.CHILD_REL_ADOPT or rel2 == RelLib.Person.CHILD_REL_ADOPT: return 1 return 0 #------------------------------------------------------------------------- # "People who have images" #------------------------------------------------------------------------- class HavePhotos(Rule): """People who have images""" name = _('People with images') description = _("Matches people with images in the gallery") category = _('General filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) return len( p.get_media_list()) > 0 #------------------------------------------------------------------------- # "People with children" #------------------------------------------------------------------------- class HaveChildren(Rule): """People with children""" name = _('People with children') description = _("Matches people who have children") category = _('Family filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) for family_handle in p.get_family_handle_list(): family = db.get_family_from_handle(family_handle) return len(family.get_child_handle_list()) > 0 #------------------------------------------------------------------------- # "People with no marriage records" #------------------------------------------------------------------------- class NeverMarried(Rule): """People with no marriage records""" name = _('People with no marriage records') description = _("Matches people who have no spouse") category = _('Family filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) return len(p.get_family_handle_list()) == 0 #------------------------------------------------------------------------- # "People with multiple marriage records" #------------------------------------------------------------------------- class MultipleMarriages(Rule): """People with multiple marriage records""" name = _('People with multiple marriage records') description = _("Matches people who have more than one spouse") category = _('Family filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) return len(p.get_family_handle_list()) > 1 #------------------------------------------------------------------------- # "People without a birth date" #------------------------------------------------------------------------- class NoBirthdate(Rule): """People without a birth date""" name = _('People without a known birth date') description = _("Matches people without a known birthdate") category = _('General filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) birth_handle = p.get_birth_handle() if not birth_handle: return 1 birth = db.get_event_from_handle(birth_handle) if not birth.get_date_object(): return 1 return 0 #------------------------------------------------------------------------- # "People with incomplete events" #------------------------------------------------------------------------- class PersonWithIncompleteEvent(Rule): """People with incomplete events""" name = _('People with incomplete events') description = _("Matches people with missing date or place in an event") category = _('Event filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) for event_handle in p.get_event_list() + [p.get_birth_handle(), p.get_death_handle()]: event = db.get_event_from_handle(event_handle) if event: if not event.get_place_handle(): return 1 if not event.get_date_object(): return 1 return 0 #------------------------------------------------------------------------- # "Families with incomplete events" #------------------------------------------------------------------------- class FamilyWithIncompleteEvent(Rule): """Families with incomplete events""" name = _('Families with incomplete events') description = _("Matches people with missing date or place in an event of the family") category = _('Event filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) for family_handle in p.get_family_handle_list(): family = db.get_family_from_handle(family_handle) for event_handle in family.get_event_list(): event = db.get_event_from_handle(event_handle) if event: if not event.get_place_handle(): return 1 if not event.get_date_object(): return 1 return 0 #------------------------------------------------------------------------- # "People probably alive" #------------------------------------------------------------------------- class ProbablyAlive(Rule): """People probably alive""" labels = [_("On year:")] name = _('People probably alive') description = _("Matches people without indications of death that are not too old") category = _('General filters') def prepare(self,db): try: self.current_year = int(self.list[0]) except: self.current_year = None def apply(self,db,handle): p = db.get_person_from_handle(handle) return probably_alive(p,db,self.current_year) #------------------------------------------------------------------------- # "People marked private" #------------------------------------------------------------------------- class PeoplePrivate(Rule): """People marked private""" name = _('People marked private') description = _("Matches people that are indicated as private") category = _('General filters') def apply(self,db,handle): p = db.get_person_from_handle(handle) return p.get_privacy() #------------------------------------------------------------------------- # "Witnesses" #------------------------------------------------------------------------- class IsWitness(Rule): """Witnesses""" labels = [_('Personal event:'), _('Family event:')] name = _('Witnesses') description = _("Matches people who are witnesses in any event") category = _('Event filters') def prepare(self,db): self.db = db self.map = [] self.build_witness_list() def reset(self): self.map = [] def apply(self,db,handle): return handle in self.map def build_witness_list(self): event_type = None if self.list and self.list[0]: event_type = self.list[0] if not self.list or event_type: for person_handle in self.db.get_person_handles(): p = self.db.get_person_from_handle(person_handle) self.get_witness_of_events( event_type, p.get_event_list()+[p.get_birth_handle(), p.get_death_handle()]) event_type = None if self.list and self.list[1]: event_type = self.list[1] if not self.list or event_type: for family_handle in self.db.get_family_handles(): f = self.db.get_family_from_handle(family_handle) self.get_witness_of_events(event_type, f.get_event_list()) def get_witness_of_events(self, event_type, event_list): if not event_list: return for event_handle in event_list: event = self.db.get_event_from_handle(event_handle) if event: if event_type and not event.get_name() == event_type: continue wlist = event.get_witness_list() if wlist: for w in wlist: if w.get_type() == RelLib.Event.ID: self.map.append(w.get_value()) #------------------------------------------------------------------------- # "HasTextMatchingSubstringOf" #------------------------------------------------------------------------- class HasTextMatchingSubstringOf(Rule): """Rule that checks for string matches in any textual information""" labels = [ _('Substring:'), _('Case sensitive:'), _('Regular-Expression matching:')] name = _('People with records containing ') description = _("Matches people whose records contain text matching a substring") category = _('General filters') def prepare(self,db): self.db = db self.person_map = {} self.event_map = {} self.source_map = {} self.family_map = {} self.place_map = {} self.media_map = {} try: if int(self.list[1]): self.case_sensitive = True else: self.case_sensitive = False except IndexError: self.case_sensitive = False try: if int(self.list[2]): self.regexp_match = True else: self.regexp_match = False except IndexError: self.regexp_match = False self.cache_sources() def reset(self): self.person_map = {} self.event_map = {} self.source_map = {} self.family_map = {} self.place_map = {} self.media_map = {} def apply(self,db,handle): if handle in self.person_map: # Cached by matching Source? return self.person_map[handle] p = db.get_person_from_handle(handle) if self.match_object(p): # first match the person itself return 1 for event_handle in p.get_event_list()+[p.get_birth_handle(), p.get_death_handle()]: if self.search_event(event_handle): # match referenced events return 1 for family_handle in p.get_family_handle_list(): # match families if self.search_family(family_handle): return 1 for media_ref in p.get_media_list(): # match Media object if self.search_media(media_ref.get_reference_handle()): return 1 return 0 def search_family(self,family_handle): if not family_handle: return 0 # search inside the family and cache the result to not search a family twice if not family_handle in self.family_map: match = 0 family = self.db.get_family_from_handle(family_handle) if self.match_object(family): match = 1 else: for event_handle in family.get_event_list(): if self.search_event(event_handle): match = 1 break for media_ref in family.get_media_list(): # match Media object if self.search_media(media_ref.get_reference_handle()): return 1 self.family_map[family_handle] = match return self.family_map[family_handle] def search_event(self,event_handle): if not event_handle: return 0 # search inside the event and cache the result (event sharing) if not event_handle in self.event_map: match = 0 event = self.db.get_event_from_handle(event_handle) if self.match_object(event): match = 1 elif event: place_handle = event.get_place_handle() if place_handle: if self.search_place(place_handle): match = 1 for media_ref in event.get_media_list(): # match Media object if self.search_media(media_ref.get_reference_handle()): return 1 self.event_map[event_handle] = match return self.event_map[event_handle] def search_place(self,place_handle): if not place_handle: return 0 # search inside the place and cache the result if not place_handle in self.place_map: place = self.db.get_place_from_handle(place_handle) self.place_map[place_handle] = self.match_object(place) return self.place_map[place_handle] def search_media(self,media_handle): if not media_handle: return 0 # search inside the place and cache the result if not media_handle in self.media_map: media = self.db.get_object_from_handle(media_handle) self.media_map[media_handle] = self.match_object(media) return self.media_map[media_handle] def cache_sources(self): # search all sources and match all referents of a matching source for source_handle in self.db.get_source_handles(): source = self.db.get_source_from_handle(source_handle) if self.match_object(source): (person_list,family_list,event_list, place_list,source_list,media_list ) = get_source_referents(source_handle,self.db) for handle in person_list: self.person_map[handle] = 1 for handle in family_list: self.family_map[handle] = 1 for handle in event_list: self.event_map[handle] = 1 for handle in place_list: self.place_map[handle] = 1 for handle in media_list: self.media_map[handle] = 1 def match_object(self,obj): if not obj: return 0 if self.regexp_match: return obj.matches_regexp(self.list[0],self.case_sensitive) return obj.matches_string(self.list[0],self.case_sensitive) #------------------------------------------------------------------------- # "HasTextMatchingRegexOf" #------------------------------------------------------------------------- class HasTextMatchingRegexpOf(HasTextMatchingSubstringOf): """This is wrapping HasTextMatchingSubstringOf to enable the regex_match parameter""" def __init__(self,list): HasTextMatchingSubstringOf.__init__(self,list) def prepare(self,db): self.db = db self.person_map = {} self.event_map = {} self.source_map = {} self.family_map = {} self.place_map = {} self.media_map = {} self.case_sensitive = False self.regexp_match = True self.cache_sources() #------------------------------------------------------------------------- # # HasSourceOf # #------------------------------------------------------------------------- class HasSourceOf(Rule): """Rule that checks people that have a particular source.""" labels = [ _('Source ID:') ] name = _('People with the ') category = _('General filters') description = _('Matches people who have a particular source') def prepare(self,db): try: self.source_handle = db.get_source_from_gramps_id(self.list[0]).get_handle() except: self.source_handle = None def apply(self,db,handle): if not self.source_handle: return False p = db.get_person_from_handle(handle) return p.has_source_reference( self.source_handle) #------------------------------------------------------------------------- # # GenericFilter # #------------------------------------------------------------------------- class GenericFilter: """Filter class that consists of several rules""" logical_functions = ['or', 'and', 'xor', 'one'] def __init__(self,source=None): if source: self.need_param = source.need_param self.flist = source.flist[:] self.name = source.name self.comment = source.comment self.logical_op = source.logical_op self.invert = source.invert else: self.need_param = 0 self.flist = [] self.name = '' self.comment = '' self.logical_op = 'and' self.invert = 0 def set_logical_op(self,val): if val in GenericFilter.logical_functions: self.logical_op = val else: self.logical_op = 'and' def get_logical_op(self): return self.logical_op def set_invert(self, val): self.invert = not not val def get_invert(self): return self.invert def get_name(self): return self.name def set_name(self,name): self.name = name def set_comment(self,comment): self.comment = comment def get_comment(self): return self.comment def add_rule(self,rule): self.flist.append(rule) def delete_rule(self,rule): self.flist.remove(rule) def set_rules(self,rules): self.flist = rules def get_rules(self): return self.flist def check_or(self,db,handle): test = 0 for rule in self.flist: test = test or rule.apply(db,handle) if test: break if self.invert: return not test else: return test def check_xor(self,db,handle): test = 0 for rule in self.flist: temp = rule.apply(db,handle) test = ((not test) and temp) or (test and (not temp)) if self.invert: return not test else: return test def check_one(self,db,handle): count = 0 for rule in self.flist: if rule.apply(db,handle): count = count + 1 if count > 1: break if self.invert: return count != 1 else: return count == 1 def check_and(self,db,handle): test = 1 for rule in self.flist: test = test and rule.apply(db,handle) if not test: break if self.invert: return not test else: return test def get_check_func(self): try: m = getattr(self, 'check_' + self.logical_op) except AttributeError: m = self.check_and return m def check(self,db,handle): return self.get_check_func()(db,handle) def apply(self,db,id_list): m = self.get_check_func() res = [] for rule in self.flist: rule.prepare(db) for handle in id_list: if m(db,handle): res.append(handle) for rule in self.flist: rule.reset() return res #------------------------------------------------------------------------- # # Name to class mappings # #------------------------------------------------------------------------- # This dict is mapping from old names to new names, so that the existing # custom_filters.xml will continue working old_names_2_class = { "Everyone" : Everyone, "Is default person" : IsDefaultPerson, "Is bookmarked person" : IsBookmarked, "Has the Id" : HasIdOf, "Has a name" : HasNameOf, "Has the relationships" : HasRelationship, "Has the death" : HasDeath, "Has the birth" : HasBirth, "Is a descendant of" : IsDescendantOf, "Is a descendant family member of" : IsDescendantFamilyOf, "Is a descendant of filter match": IsDescendantOfFilterMatch, "Is a descendant of person not more than N generations away": IsLessThanNthGenerationDescendantOf, "Is a descendant of person at least N generations away": IsMoreThanNthGenerationDescendantOf, "Is an descendant of person at least N generations away" : IsMoreThanNthGenerationDescendantOf, "Is a child of filter match" : IsChildOfFilterMatch, "Is an ancestor of" : IsAncestorOf, "Is an ancestor of filter match": IsAncestorOfFilterMatch, "Is an ancestor of person not more than N generations away" : IsLessThanNthGenerationAncestorOf, "Is an ancestor of person at least N generations away": IsMoreThanNthGenerationAncestorOf, "Is a parent of filter match" : IsParentOfFilterMatch, "Has a common ancestor with" : HasCommonAncestorWith, "Has a common ancestor with filter match" :HasCommonAncestorWithFilterMatch, "Is a female" : IsFemale, "Is a male" : IsMale, "Has complete record" : HasCompleteRecord, "Has the personal event" : HasEvent, "Has the family event" : HasFamilyEvent, "Has the personal attribute" : HasAttribute, "Has the family attribute" : HasFamilyAttribute, "Has source of" : HasSourceOf, "Matches the filter named" : HasSourceOf, "Is spouse of filter match" : IsSpouseOfFilterMatch, "Is a sibling of filter match" : IsSiblingOfFilterMatch, "Relationship path between two people" : RelationshipPathBetween, "People who were adopted" : HaveAltFamilies, "People who have images" : HavePhotos, "People with children" : HaveChildren, "People with incomplete names" : IncompleteNames, "People with no marriage records" : NeverMarried, "People with multiple marriage records": MultipleMarriages, "People without a birth date" : NoBirthdate, "People with incomplete events" : PersonWithIncompleteEvent, "Families with incomplete events" :FamilyWithIncompleteEvent, "People probably alive" : ProbablyAlive, "People marked private" : PeoplePrivate, "Witnesses" : IsWitness, "Has text matching substring of":HasTextMatchingSubstringOf, } editor_rule_list = [ Everyone, IsFemale, HasUnknownGender, IsMale, IsDefaultPerson, IsBookmarked, HasIdOf, HasNameOf, HasRelationship, HasDeath, HasBirth, HasCompleteRecord, HasEvent, HasFamilyEvent, HasAttribute, HasFamilyAttribute, HasSourceOf, HaveAltFamilies, HavePhotos, HaveChildren, IncompleteNames, NeverMarried, MultipleMarriages, NoBirthdate, PersonWithIncompleteEvent, FamilyWithIncompleteEvent, ProbablyAlive, PeoplePrivate, IsWitness, IsDescendantOf, IsDescendantFamilyOf, IsDescendantOfFilterMatch, IsLessThanNthGenerationDescendantOf, IsMoreThanNthGenerationDescendantOf, IsAncestorOf, IsAncestorOfFilterMatch, IsLessThanNthGenerationAncestorOf, IsMoreThanNthGenerationAncestorOf, HasCommonAncestorWith, HasCommonAncestorWithFilterMatch, MatchesFilter, IsChildOfFilterMatch, IsParentOfFilterMatch, IsSpouseOfFilterMatch, IsSiblingOfFilterMatch, RelationshipPathBetween, HasTextMatchingSubstringOf, ] #------------------------------------------------------------------------- # # GenericFilterList # #------------------------------------------------------------------------- class GenericFilterList: """Container class for the generic filters. Stores, saves, and loads the filters.""" def __init__(self,file): self.filter_list = [] self.file = os.path.expanduser(file) def get_filters(self): return self.filter_list def add(self,filt): self.filter_list.append(filt) def load(self): try: parser = make_parser() parser.setContentHandler(FilterParser(self)) if self.file[0:7] != "file://": parser.parse("file://" + self.file) else: parser.parse(self.file) except (IOError,OSError): pass except SAXParseException: print "Parser error" def fix(self,line): l = line.strip() l = l.replace('&','&') l = l.replace('>','>') l = l.replace('<','<') return l.replace('"','"') def save(self): f = open(self.file.encode('utf-8'),'w') f.write("\n") f.write('\n') for i in self.filter_list: f.write(' \n') for rule in i.get_rules(): rule_module_name = rule.__module__ rule_class_name = rule.__class__.__name__ rule_save_name = "%s.%s" % (rule_module_name,rule_class_name) f.write(' \n' % rule_save_name) for v in rule.values(): f.write(' \n' % self.fix(v)) f.write(' \n') f.write(' \n') f.write('\n') f.close() #------------------------------------------------------------------------- # # FilterParser # #------------------------------------------------------------------------- class FilterParser(handler.ContentHandler): """Parses the XML file and builds the list of filters""" def __init__(self,gfilter_list): handler.ContentHandler.__init__(self) self.gfilter_list = gfilter_list self.f = None self.r = None self.a = [] self.cname = None def setDocumentLocator(self,locator): self.locator = locator def startElement(self,tag,attrs): if tag == "filter": self.f = GenericFilter() self.f.set_name(attrs['name']) if attrs.has_key('function'): try: if int(attrs['function']): op = 'or' else: op = 'and' except ValueError: op = attrs['function'] self.f.set_logical_op(op) if attrs.has_key('comment'): self.f.set_comment(attrs['comment']) if attrs.has_key('invert'): try: self.f.set_invert(int(attrs['invert'])) except ValueError: pass self.gfilter_list.add(self.f) elif tag == "rule": save_name = attrs['class'] if save_name in old_names_2_class.keys(): self.r = old_names_2_class[save_name] else: mc_match = save_name.split('.') if len(mc_match) != 2: # rule has an old style name, that is not in old_names_2_class # or is not in the format "module.class" print "ERROR: Filter rule '%s' in filter '%s' not found!" % ( save_name,self.f.get_name()) self.r = None return module_name,class_name = mc_match try: if module_name == self.__module__: exec 'self.r = %s' % class_name else: exec 'import %s' % module_name exec 'self.r = %s.%s' % (module_name,class_name) except (ImportError,NameError): print "ERROR: Filter rule '%s' in filter '%s' not found!" % ( save_name,self.f.get_name()) self.r = None return self.a = [] elif tag == "arg": self.a.append(attrs['value']) def endElement(self,tag): if tag == "rule" and self.r != None: rule = self.r(self.a) self.f.add_rule(rule) def characters(self, data): pass class ParamFilter(GenericFilter): def __init__(self,source=None): GenericFilter.__init__(self,source) self.need_param = 1 self.param_list = [] def set_parameter(self,param): self.param_list = [param] def apply(self,db,id_list): for rule in self.flist: rule.set_list(self.param_list) for rule in self.flist: rule.prepare(db) result = GenericFilter.apply(self,db,id_list) for rule in self.flist: rule.reset() return result #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- SystemFilters = None CustomFilters = None def reload_system_filters(): global SystemFilters SystemFilters = GenericFilterList(const.system_filters) SystemFilters.load() def reload_custom_filters(): global CustomFilters CustomFilters = GenericFilterList(const.custom_filters) CustomFilters.load() if not SystemFilters: reload_system_filters() if not CustomFilters: reload_custom_filters() class GrampsFilterComboBox(gtk.ComboBox): def set(self,local_filters,default=""): self.store = gtk.ListStore(str) self.set_model(self.store) cell = gtk.CellRendererText() self.pack_start(cell,True) self.add_attribute(cell,'text',0) self.map = {} active = 0 cnt = 0 for filt in local_filters: self.store.append(row=[filt.get_name()]) self.map[filt.get_name()] = filt if default != "" and default == filt.get_name(): active = cnt cnt += 1 for filt in SystemFilters.get_filters(): self.store.append(row=[_(filt.get_name())]) self.map[filt.get_name()] = filt if default != "" and default == filt.get_name(): active = cnt cnt += 1 for filt in CustomFilters.get_filters(): self.store.append(row=[_(filt.get_name())]) self.map[filt.get_name()] = filt if default != "" and default == filt.get_name(): active = cnt cnt += 1 if active: self.set_active(active) elif len(local_filters): self.set_active(2) elif len(SystemFilters.get_filters()): self.set_active(4 + len(local_filters)) elif len(CustomFilters.get_filters()): self.set_active(6 + len(local_filters) + len(SystemFilters.get_filters())) else: self.set_active(0) def get_value(self): active = self.get_active() if active < 0: return None key = self.store[active][0] return self.map[key] class FilterStore(gtk.ListStore): def __init__(self,local_filters=[], default=""): gtk.ListStore.__init__(self,str) self.list_map = [] self.def_index = 0 cnt = 0 for filt in local_filters: name = filt.get_name() self.append(row=[name]) self.list_map.append(filt) if default != "" and default == name: self.def_index = cnt cnt += 1 for filt in SystemFilters.get_filters() + CustomFilters.get_filters(): name = _(filt.get_name()) self.append(row=[name]) self.list_map.append(filt) if default != "" and default == name: self.def_index = cnt cnt += 1 def default_index(self): return self.def_index def get_filter(self,index): return self.list_map[index] def build_filter_menu(local_filters = [], default=""): menu = gtk.Menu() active = 0 cnt = 0 for filt in local_filters: menuitem = gtk.MenuItem(filt.get_name()) menuitem.show() menu.append(menuitem) menuitem.set_data("filter", filt) if default != "" and default == filt.get_name(): active = cnt cnt += 1 for filt in SystemFilters.get_filters(): menuitem = gtk.MenuItem(_(filt.get_name())) menuitem.show() menu.append(menuitem) menuitem.set_data("filter", filt) if default != "" and default == filt.get_name(): active = cnt cnt += 1 for filt in CustomFilters.get_filters(): menuitem = gtk.MenuItem(_(filt.get_name())) menuitem.show() menu.append(menuitem) menuitem.set_data("filter", filt) if default != "" and default == filt.get_name(): active = cnt cnt += 1 if active: menu.set_active(active) elif len(local_filters): menu.set_active(2) elif len(SystemFilters.get_filters()): menu.set_active(4 + len(local_filters)) elif len(CustomFilters.get_filters()): menu.set_active(6 + len(local_filters) + len(SystemFilters.get_filters())) else: menu.set_active(0) return menu