2384 lines
80 KiB
Python
2384 lines
80 KiB
Python
#
|
|
# 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 <persons>")
|
|
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 <Id>')
|
|
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 <person>')
|
|
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 <filter> 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 <person> not more than <N> 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 <person> at least <N> 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 <filter> 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 <filter> 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 <person>')
|
|
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 <person>')
|
|
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 <filter> 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 <person> not more than <N> 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 <person> at least <N> 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 <filter> 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 <person>')
|
|
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 <filter> 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 <event>')
|
|
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 <event>')
|
|
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 <relationships>')
|
|
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 <birth data>')
|
|
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 <death data>')
|
|
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 <attribute>')
|
|
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 <attribute>')
|
|
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 <name>')
|
|
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 <name>')
|
|
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 <filter>')
|
|
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 <filter> 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 <substring>')
|
|
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 <source>')
|
|
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("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n")
|
|
f.write('<filters>\n')
|
|
for i in self.filter_list:
|
|
f.write(' <filter name="%s"' % self.fix(i.get_name()))
|
|
if i.get_invert():
|
|
f.write(' invert="1"')
|
|
f.write(' function="%s"' % i.get_logical_op())
|
|
comment = i.get_comment()
|
|
if comment:
|
|
f.write(' comment="%s"' % self.fix(comment))
|
|
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(' <rule class="%s">\n' % rule_save_name)
|
|
for v in rule.values():
|
|
f.write(' <arg value="%s"/>\n' % self.fix(v))
|
|
f.write(' </rule>\n')
|
|
f.write(' </filter>\n')
|
|
f.write('</filters>\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
|