Filter improvements
svn: r4852
This commit is contained in:
parent
cf9bfde518
commit
9e39e88d2b
@ -1,3 +1,7 @@
|
||||
2005-06-20 Don Allingham <don@gramps-project.org>
|
||||
* src/GenericFilter.py: optimize looping and reducing fetches
|
||||
* src/Utils.py: fix event handle accesses
|
||||
|
||||
2005-06-16 Alex Roitman <shura@gramps-project.org>
|
||||
* src/EventEdit.py: Remove unused variable.
|
||||
|
||||
|
@ -109,8 +109,8 @@ class Rule:
|
||||
def check(self):
|
||||
return len(self.list) == len(self.labels)
|
||||
|
||||
def apply(self,db,handle):
|
||||
return 1
|
||||
def apply(self,db,person):
|
||||
return True
|
||||
|
||||
def display_values(self):
|
||||
v = []
|
||||
@ -131,8 +131,8 @@ class Everyone(Rule):
|
||||
category = _('General filters')
|
||||
description = _('Matches everyone in the database')
|
||||
|
||||
def apply(self,db,handle):
|
||||
return 1
|
||||
def apply(self,db,person):
|
||||
return True
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -147,8 +147,7 @@ class Disconnected(Rule):
|
||||
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)
|
||||
def apply(self,db,person):
|
||||
return (not person.get_main_parents_family_handle() and
|
||||
not len(person.get_family_handle_list()))
|
||||
|
||||
@ -205,8 +204,8 @@ class RelationshipPathBetween(Rule):
|
||||
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 apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,p1_handle,p2_handle):
|
||||
firstMap = {}
|
||||
@ -262,8 +261,8 @@ class HasIdOf(Rule):
|
||||
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]
|
||||
def apply(self,db,person):
|
||||
return person.gramps_id == self.list[0]
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -277,11 +276,11 @@ class IsDefaultPerson(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
def_person = db.get_default_person()
|
||||
if def_person:
|
||||
return person.handle == def_person.handle
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -295,10 +294,10 @@ class IsBookmarked(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
if person.handle in db.get_bookmarks():
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -312,8 +311,8 @@ class HasCompleteRecord(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
return person.get_complete_flag() == 1
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -327,8 +326,8 @@ class IsFemale(Rule):
|
||||
category = _('General filters')
|
||||
description = _('Matches all females')
|
||||
|
||||
def apply(self,db,handle):
|
||||
return db.get_person_from_handle(handle).get_gender() == RelLib.Person.FEMALE
|
||||
def apply(self,db,person):
|
||||
return person.gender == RelLib.Person.FEMALE
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -342,8 +341,8 @@ class HasUnknownGender(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
return person.get_gender() == RelLib.Person.UNKNOWN
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -364,11 +363,11 @@ class IsDescendantOf(Rule):
|
||||
self.map = {}
|
||||
try:
|
||||
if int(self.list[1]):
|
||||
first = 0
|
||||
first = False
|
||||
else:
|
||||
first = 1
|
||||
first = True
|
||||
except IndexError:
|
||||
first = 1
|
||||
first = True
|
||||
try:
|
||||
root_handle = db.get_person_from_gramps_id(self.list[0]).get_handle()
|
||||
self.init_list(root_handle,first)
|
||||
@ -378,8 +377,8 @@ class IsDescendantOf(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,handle,first):
|
||||
if not handle:
|
||||
@ -433,8 +432,8 @@ class IsDescendantOfFilterMatch(IsDescendantOf):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -463,8 +462,8 @@ class IsLessThanNthGenerationDescendantOf(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,handle,gen):
|
||||
if not handle:
|
||||
@ -508,8 +507,8 @@ class IsMoreThanNthGenerationDescendantOf(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,handle,gen):
|
||||
if not handle:
|
||||
@ -550,8 +549,8 @@ class IsChildOfFilterMatch(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,handle):
|
||||
if not handle:
|
||||
@ -588,8 +587,8 @@ class IsSiblingOfFilterMatch(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,handle):
|
||||
if not handle:
|
||||
@ -617,9 +616,9 @@ class IsDescendantFamilyOf(Rule):
|
||||
description = _("Matches people that are descendants or the spouse "
|
||||
"of a descendant of a specified person")
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
self.map = {}
|
||||
self.orig_handle = handle
|
||||
self.orig_handle = person.handle
|
||||
self.db = db
|
||||
return self.search(handle,1)
|
||||
|
||||
@ -627,9 +626,9 @@ class IsDescendantFamilyOf(Rule):
|
||||
try:
|
||||
if handle == self.db.get_person_from_gramps_id(self.list[0]).get_handle():
|
||||
self.map[handle] = 1
|
||||
return 1
|
||||
return True
|
||||
except:
|
||||
return 0
|
||||
return False
|
||||
|
||||
p = self.db.get_person_from_handle(handle)
|
||||
for (f,r1,r2) in p.get_parent_family_handle_list():
|
||||
@ -637,7 +636,7 @@ class IsDescendantFamilyOf(Rule):
|
||||
for person_handle in [family.get_mother_handle(),family.get_father_handle()]:
|
||||
if person_handle:
|
||||
if self.search(person_handle,0):
|
||||
return 1
|
||||
return True
|
||||
if val:
|
||||
for family_handle in p.get_family_handle_list():
|
||||
family = self.db.get_family_from_handle(family_handle)
|
||||
@ -647,8 +646,8 @@ class IsDescendantFamilyOf(Rule):
|
||||
spouse_id = family.get_father_handle()
|
||||
if spouse_id:
|
||||
if self.search(spouse_id,0):
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -683,8 +682,8 @@ class IsAncestorOf(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_ancestor_list(self,db,handle,first):
|
||||
if not handle:
|
||||
@ -744,8 +743,8 @@ class IsAncestorOfFilterMatch(IsAncestorOf):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -774,8 +773,8 @@ class IsLessThanNthGenerationAncestorOf(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_ancestor_list(self,handle,gen):
|
||||
# if self.map.has_key(p.get_handle()) == 1:
|
||||
@ -826,8 +825,8 @@ class IsMoreThanNthGenerationAncestorOf(Rule):
|
||||
def reset(self):
|
||||
self.map = []
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_ancestor_list(self,handle,gen):
|
||||
# if self.map.has_key(p.get_handle()) == 1:
|
||||
@ -876,8 +875,8 @@ class IsParentOfFilterMatch(Rule):
|
||||
def reset(self):
|
||||
self.map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
return self.map.has_key(handle)
|
||||
def apply(self,db,person):
|
||||
return self.map.has_key(person.handle)
|
||||
|
||||
def init_list(self,handle):
|
||||
p = self.db.get_person_from_handle(handle)
|
||||
@ -919,12 +918,13 @@ class HasCommonAncestorWith(Rule):
|
||||
def init(self,handle): self.ancestor_cache[handle] = 1
|
||||
for_each_ancestor(db,[handle],init,self)
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
# 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)
|
||||
handle = person.handle
|
||||
return for_each_ancestor(db,[handle],
|
||||
lambda self,handle: self.ancestor_cache.has_key(handle),
|
||||
self);
|
||||
@ -969,8 +969,8 @@ class IsMale(Rule):
|
||||
category = _('General filters')
|
||||
description = _('Matches all males')
|
||||
|
||||
def apply(self,db,handle):
|
||||
return db.get_person_from_handle(handle).get_gender() == RelLib.Person.MALE
|
||||
def apply(self,db,person):
|
||||
return person.gender == RelLib.Person.MALE
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -995,9 +995,8 @@ class HasEvent(Rule):
|
||||
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():
|
||||
def apply(self,db,person):
|
||||
for event_handle in person.get_event_list():
|
||||
if not event_handle:
|
||||
continue
|
||||
event = db.get_event_from_handle(event_handle)
|
||||
@ -1018,8 +1017,8 @@ class HasEvent(Rule):
|
||||
if pn.upper().find(self.list[2].upper()) == -1:
|
||||
val = 0
|
||||
if val == 1:
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1045,9 +1044,8 @@ class HasFamilyEvent(Rule):
|
||||
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():
|
||||
def apply(self,db,person):
|
||||
for f_id in person.get_family_handle_list():
|
||||
f = db.get_family_from_handle(f_id)
|
||||
for event_handle in f.get_event_list():
|
||||
if not event_handle:
|
||||
@ -1069,8 +1067,8 @@ class HasFamilyEvent(Rule):
|
||||
if self.list[2] and pn.find(self.list[2].upper()) == -1:
|
||||
val = 0
|
||||
if val == 1:
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1087,14 +1085,13 @@ class HasRelationship(Rule):
|
||||
description = _("Matches people with a particular relationship")
|
||||
category = _('Family filters')
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
rel_type = 0
|
||||
cnt = 0
|
||||
p = db.get_person_from_handle(handle)
|
||||
num_rel = len(p.get_family_handle_list())
|
||||
num_rel = len(person.get_family_handle_list())
|
||||
|
||||
# count children and look for a relationship type match
|
||||
for f_id in p.get_family_handle_list():
|
||||
for f_id in person.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():
|
||||
@ -1105,24 +1102,24 @@ class HasRelationship(Rule):
|
||||
try:
|
||||
v = int(self.list[0])
|
||||
except:
|
||||
return 0
|
||||
return False
|
||||
if v != num_rel:
|
||||
return 0
|
||||
return False
|
||||
|
||||
# number of childred
|
||||
if self.list[2]:
|
||||
try:
|
||||
v = int(self.list[2])
|
||||
except:
|
||||
return 0
|
||||
return False
|
||||
if v != cnt:
|
||||
return 0
|
||||
return False
|
||||
|
||||
# relation
|
||||
if self.list[1]:
|
||||
return rel_type == 1
|
||||
else:
|
||||
return 1
|
||||
return True
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1144,9 +1141,8 @@ class HasBirth(Rule):
|
||||
else:
|
||||
self.date = None
|
||||
|
||||
def apply(self,db,handle):
|
||||
p = db.get_person_from_handle(handle)
|
||||
event_handle = p.get_birth_handle()
|
||||
def apply(self,db,person):
|
||||
event_handle = person.get_birth_handle()
|
||||
if not event_handle:
|
||||
return False
|
||||
event = db.get_event_from_handle(event_handle)
|
||||
@ -1184,25 +1180,24 @@ class HasDeath(Rule):
|
||||
else:
|
||||
self.date = None
|
||||
|
||||
def apply(self,db,handle):
|
||||
p = db.get_person_from_handle(handle)
|
||||
event_handle = p.get_death_handle()
|
||||
def apply(self,db,person):
|
||||
event_handle = person.get_death_handle()
|
||||
if not event_handle:
|
||||
return 0
|
||||
return False
|
||||
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
|
||||
return False
|
||||
if self.date:
|
||||
if date_cmp(self.date,event.get_date_object()) == 0:
|
||||
return 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 self.list[1] and pn.find(self.list[1].upper()) == -1:
|
||||
return 0
|
||||
return 1
|
||||
return False
|
||||
return True
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1217,17 +1212,16 @@ class HasAttribute(Rule):
|
||||
description = _("Matches people with the personal attribute of a particular value")
|
||||
category = _('General filters')
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
if not self.list[0]:
|
||||
return 0
|
||||
p = db.get_person_from_handle(handle)
|
||||
for attr in p.get_attribute_list():
|
||||
return False
|
||||
for attr in person.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
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1242,19 +1236,18 @@ class HasFamilyAttribute(Rule):
|
||||
description = _("Matches people with the family attribute of a particular value")
|
||||
category = _('General filters')
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
if not self.list[0]:
|
||||
return 0
|
||||
p = db.get_person_from_handle(handle)
|
||||
for f_id in p.get_family_handle_list():
|
||||
return False
|
||||
for f_id in person.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
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1272,13 +1265,12 @@ class HasNameOf(Rule):
|
||||
description = _("Matches people with a specified (partial) name")
|
||||
category = _('General filters')
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
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():
|
||||
for name in [person.get_primary_name()] + person.get_alternate_names():
|
||||
val = 1
|
||||
if self.f and name.get_first_name().upper().find(self.f.upper()) == -1:
|
||||
val = 0
|
||||
@ -1289,8 +1281,8 @@ class HasNameOf(Rule):
|
||||
if self.t and name.get_title().upper().find(self.t.upper()) == -1:
|
||||
val = 0
|
||||
if val == 1:
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1323,14 +1315,13 @@ class IncompleteNames(Rule):
|
||||
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():
|
||||
def apply(self,db,person):
|
||||
for name in [person.get_primary_name()] + person.get_alternate_names():
|
||||
if name.get_first_name() == "":
|
||||
return 1
|
||||
return True
|
||||
if name.get_surname() == "":
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1372,7 +1363,7 @@ class MatchesFilter(Rule):
|
||||
for filt in CustomFilters.get_filters():
|
||||
if filt.get_name() == self.list[0]:
|
||||
return filt.check(db,handle)
|
||||
return 0
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1388,10 +1379,9 @@ class IsSpouseOfFilterMatch(Rule):
|
||||
description = _("Matches people married to anybody matching a filter")
|
||||
category = _('Family filters')
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
filt = MatchesFilter (self.list)
|
||||
p = db.get_person_from_handle(handle)
|
||||
for family_handle in p.get_family_handle_list ():
|
||||
for family_handle in person.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:
|
||||
@ -1399,8 +1389,8 @@ class IsSpouseOfFilterMatch(Rule):
|
||||
if spouse_id == handle:
|
||||
continue
|
||||
if filt.apply (db, spouse_id):
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People who were adopted"
|
||||
@ -1412,12 +1402,11 @@ class HaveAltFamilies(Rule):
|
||||
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():
|
||||
def apply(self,db,person):
|
||||
for (fam,rel1,rel2) in person.get_parent_family_handle_list():
|
||||
if rel1 == RelLib.Person.CHILD_ADOPTED or rel2 == RelLib.Person.CHILD_ADOPTED:
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
@ -1430,9 +1419,8 @@ class HavePhotos(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
return len( person.get_media_list()) > 0
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People with children"
|
||||
@ -1444,9 +1432,8 @@ class HaveChildren(Rule):
|
||||
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():
|
||||
def apply(self,db,person):
|
||||
for family_handle in person.get_family_handle_list():
|
||||
family = db.get_family_from_handle(family_handle)
|
||||
return len(family.get_child_handle_list()) > 0
|
||||
|
||||
@ -1460,9 +1447,8 @@ class NeverMarried(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
return len(person.get_family_handle_list()) == 0
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People with multiple marriage records"
|
||||
@ -1474,9 +1460,8 @@ class MultipleMarriages(Rule):
|
||||
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
|
||||
def apply(self,db,person):
|
||||
return len(person.get_family_handle_list()) > 1
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People without a birth date"
|
||||
@ -1488,15 +1473,14 @@ class NoBirthdate(Rule):
|
||||
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()
|
||||
def apply(self,db,person):
|
||||
birth_handle = person.get_birth_handle()
|
||||
if not birth_handle:
|
||||
return 1
|
||||
return True
|
||||
birth = db.get_event_from_handle(birth_handle)
|
||||
if not birth.get_date_object():
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People with incomplete events"
|
||||
@ -1508,16 +1492,15 @@ class PersonWithIncompleteEvent(Rule):
|
||||
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()]:
|
||||
def apply(self,db,person):
|
||||
for event_handle in person.get_event_list() + [person.get_birth_handle(), person.get_death_handle()]:
|
||||
event = db.get_event_from_handle(event_handle)
|
||||
if event:
|
||||
if not event.get_place_handle():
|
||||
return 1
|
||||
return True
|
||||
if not event.get_date_object():
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "Families with incomplete events"
|
||||
@ -1529,18 +1512,17 @@ class FamilyWithIncompleteEvent(Rule):
|
||||
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():
|
||||
def apply(self,db,person):
|
||||
for family_handle in person.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
|
||||
return True
|
||||
if not event.get_date_object():
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People probably alive"
|
||||
@ -1559,9 +1541,8 @@ class ProbablyAlive(Rule):
|
||||
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)
|
||||
def apply(self,db,person):
|
||||
return probably_alive(person,db,self.current_year)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "People marked private"
|
||||
@ -1573,9 +1554,8 @@ class PeoplePrivate(Rule):
|
||||
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()
|
||||
def apply(self,db,person):
|
||||
return person.get_privacy()
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# "Witnesses"
|
||||
@ -1596,8 +1576,8 @@ class IsWitness(Rule):
|
||||
def reset(self):
|
||||
self.map = []
|
||||
|
||||
def apply(self,db,handle):
|
||||
return handle in self.map
|
||||
def apply(self,db,person):
|
||||
return person.handle in self.map
|
||||
|
||||
def build_witness_list(self):
|
||||
event_type = None
|
||||
@ -1679,26 +1659,25 @@ class HasTextMatchingSubstringOf(Rule):
|
||||
self.place_map = {}
|
||||
self.media_map = {}
|
||||
|
||||
def apply(self,db,handle):
|
||||
if handle in self.person_map: # Cached by matching Source?
|
||||
def apply(self,db,person):
|
||||
if person.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.match_object(person): # first match the person itself
|
||||
return True
|
||||
for event_handle in person.get_event_list()+[person.get_birth_handle(), person.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
|
||||
return True
|
||||
for family_handle in person.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
|
||||
return True
|
||||
for media_ref in person.get_media_list(): # match Media object
|
||||
if self.search_media(media_ref.get_reference_handle()):
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
def search_family(self,family_handle):
|
||||
if not family_handle:
|
||||
return 0
|
||||
return False
|
||||
# search inside the family and cache the result to not search a family twice
|
||||
if not family_handle in self.family_map:
|
||||
match = 0
|
||||
@ -1712,13 +1691,13 @@ class HasTextMatchingSubstringOf(Rule):
|
||||
break
|
||||
for media_ref in family.get_media_list(): # match Media object
|
||||
if self.search_media(media_ref.get_reference_handle()):
|
||||
return 1
|
||||
return True
|
||||
self.family_map[family_handle] = match
|
||||
return self.family_map[family_handle]
|
||||
|
||||
def search_event(self,event_handle):
|
||||
if not event_handle:
|
||||
return 0
|
||||
return False
|
||||
# search inside the event and cache the result (event sharing)
|
||||
if not event_handle in self.event_map:
|
||||
match = 0
|
||||
@ -1732,13 +1711,13 @@ class HasTextMatchingSubstringOf(Rule):
|
||||
match = 1
|
||||
for media_ref in event.get_media_list(): # match Media object
|
||||
if self.search_media(media_ref.get_reference_handle()):
|
||||
return 1
|
||||
return True
|
||||
self.event_map[event_handle] = match
|
||||
return self.event_map[event_handle]
|
||||
|
||||
def search_place(self,place_handle):
|
||||
if not place_handle:
|
||||
return 0
|
||||
return False
|
||||
# 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)
|
||||
@ -1747,7 +1726,7 @@ class HasTextMatchingSubstringOf(Rule):
|
||||
|
||||
def search_media(self,media_handle):
|
||||
if not media_handle:
|
||||
return 0
|
||||
return False
|
||||
# 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)
|
||||
@ -1787,7 +1766,7 @@ class HasTextMatchingSubstringOf(Rule):
|
||||
|
||||
def match_object(self,obj):
|
||||
if not obj:
|
||||
return 0
|
||||
return False
|
||||
if self.regexp_match:
|
||||
return obj.matches_regexp(self.list[0],self.case_sensitive)
|
||||
return obj.matches_string(self.list[0],self.case_sensitive)
|
||||
@ -1832,11 +1811,10 @@ class HasSourceOf(Rule):
|
||||
except:
|
||||
self.source_handle = None
|
||||
|
||||
def apply(self,db,handle):
|
||||
def apply(self,db,person):
|
||||
if not self.source_handle:
|
||||
return False
|
||||
p = db.get_person_from_handle(handle)
|
||||
return p.has_source_reference( self.source_handle)
|
||||
return person.has_source_reference( self.source_handle)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
#
|
||||
@ -1862,7 +1840,7 @@ class GenericFilter:
|
||||
self.name = ''
|
||||
self.comment = ''
|
||||
self.logical_op = 'and'
|
||||
self.invert = 0
|
||||
self.invert = False
|
||||
|
||||
def set_logical_op(self,val):
|
||||
if val in GenericFilter.logical_functions:
|
||||
@ -1903,49 +1881,62 @@ class GenericFilter:
|
||||
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
|
||||
def check_func(self,db,id_list,task):
|
||||
final_list = []
|
||||
|
||||
if id_list == None:
|
||||
cursor = db.get_person_cursor()
|
||||
data = cursor.next()
|
||||
while data:
|
||||
person = RelLib.Person()
|
||||
person.unserialize(data[1])
|
||||
if self.invert ^ task(db,person):
|
||||
final_list.append(data[0])
|
||||
else:
|
||||
for handle in id_list:
|
||||
person = db.get_person_from_handle(handle)
|
||||
if self.invert ^ task(db,person):
|
||||
final_list.append(handle)
|
||||
return final_list
|
||||
|
||||
def check_or(self,db,id_list):
|
||||
return self.check_func(db,id_list,self.or_test)
|
||||
|
||||
def check_and(self,db,id_list):
|
||||
return self.check_func(db,id_list,self.and_test)
|
||||
|
||||
def check_one(self,db,id_list):
|
||||
return self.check_func(db,id_list,self.one_test)
|
||||
|
||||
def check_xor(self,db,id_list):
|
||||
return self.check_func(db,id_list,self.xor_test)
|
||||
|
||||
def xor_test(self,db,person):
|
||||
test = False
|
||||
for rule in self.flist:
|
||||
test = test ^ rule.apply(db,handle)
|
||||
return test
|
||||
|
||||
def check_xor(self,db,handle):
|
||||
test = 0
|
||||
def and_test(self,db,person):
|
||||
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
|
||||
if not rule.apply(db,person):
|
||||
return False
|
||||
return True
|
||||
|
||||
def check_one(self,db,handle):
|
||||
def one_test(self,db,person):
|
||||
count = 0
|
||||
for rule in self.flist:
|
||||
if rule.apply(db,handle):
|
||||
count = count + 1
|
||||
if count > 1:
|
||||
break
|
||||
if self.invert:
|
||||
if rule.apply(db,person):
|
||||
if count:
|
||||
return False
|
||||
count += 1
|
||||
return count != 1
|
||||
else:
|
||||
return count == 1
|
||||
|
||||
def check_and(self,db,handle):
|
||||
test = 1
|
||||
def and_or(self,db,person):
|
||||
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
|
||||
if rule.apply(db,person):
|
||||
return True
|
||||
return False
|
||||
|
||||
def get_check_func(self):
|
||||
try:
|
||||
@ -1957,14 +1948,11 @@ class GenericFilter:
|
||||
def check(self,db,handle):
|
||||
return self.get_check_func()(db,handle)
|
||||
|
||||
def apply(self,db,id_list):
|
||||
def apply(self,db,id_list=None):
|
||||
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)
|
||||
res = m(db,id_list)
|
||||
for rule in self.flist:
|
||||
rule.reset()
|
||||
return res
|
||||
|
39
src/Utils.py
39
src/Utils.py
@ -711,8 +711,8 @@ def probably_alive(person,db,current_year=None):
|
||||
death_year = None
|
||||
# If the recorded death year is before current year then
|
||||
# things are simple.
|
||||
if person.death_handle:
|
||||
death = db.get_event_from_handle(person.death_handle)
|
||||
if person.death_ref:
|
||||
death = db.get_event_from_handle(person.death_ref.ref)
|
||||
if death.get_date_object().get_start_date() != Date.EMPTY:
|
||||
death_year = death.get_date_object().get_year()
|
||||
if death.get_date_object().get_year() < current_year:
|
||||
@ -720,9 +720,10 @@ def probably_alive(person,db,current_year=None):
|
||||
|
||||
# Look for Cause Of Death, Burial or Cremation events.
|
||||
# These are fairly good indications that someone's not alive.
|
||||
for ev_handle in person.event_list:
|
||||
ev = db.get_event_from_handle(ev_handle)
|
||||
if ev and ev.name in ["Cause Of Death", "Burial", "Cremation"]:
|
||||
for ev_ref in person.event_ref_list:
|
||||
ev = db.get_event_from_handle(ev_ref.ref)
|
||||
if ev and ev.get_type()[0] in [RelLib.Event.CAUSE_DEATH, RelLib.Event.BURIAL,
|
||||
RelLib.Event.CREMATION]:
|
||||
if not death_year:
|
||||
death_year = ev.get_date_object().get_year()
|
||||
if ev.get_date_object().get_start_date() != Date.EMPTY:
|
||||
@ -732,8 +733,8 @@ def probably_alive(person,db,current_year=None):
|
||||
birth_year = None
|
||||
# If they were born within 100 years before current year then
|
||||
# assume they are alive (we already know they are not dead).
|
||||
if person.birth_handle:
|
||||
birth = db.get_event_from_handle(person.birth_handle)
|
||||
if person.birth_ref:
|
||||
birth = db.get_event_from_handle(person.birth_ref.ref)
|
||||
if birth.get_date_object().get_start_date() != Date.EMPTY:
|
||||
if not birth_year:
|
||||
birth_year = birth.get_date_object().get_year()
|
||||
@ -766,8 +767,8 @@ def probably_alive(person,db,current_year=None):
|
||||
|
||||
for child_handle in family.get_child_handle_list():
|
||||
child = db.get_person_from_handle(child_handle)
|
||||
if child.birth_handle:
|
||||
child_birth = db.get_event_from_handle(child.birth_handle)
|
||||
if child.birth_ref:
|
||||
child_birth = db.get_event_from_handle(child.birth_ref.ref)
|
||||
dobj = child_birth.get_date_object()
|
||||
if dobj.get_start_date() != Date.EMPTY:
|
||||
d = Date.Date(dobj)
|
||||
@ -777,8 +778,8 @@ def probably_alive(person,db,current_year=None):
|
||||
if not not_too_old (d,current_year):
|
||||
return True
|
||||
|
||||
if child.death_handle:
|
||||
child_death = db.get_event_from_handle(child.death_handle)
|
||||
if child.death_ref:
|
||||
child_death = db.get_event_from_handle(child.death_ref.ref)
|
||||
dobj = child_death.get_date_object()
|
||||
if dobj.get_start_date() != Date.EMPTY:
|
||||
if not not_too_old (dobj,current_year):
|
||||
@ -806,8 +807,8 @@ def probably_alive(person,db,current_year=None):
|
||||
father_handle = family.get_father_handle()
|
||||
if father_handle:
|
||||
father = db.get_person_from_handle(father_handle)
|
||||
if father.birth_handle:
|
||||
father_birth = db.get_event_from_handle(father.birth_handle)
|
||||
if father.birth_ref:
|
||||
father_birth = db.get_event_from_handle(father.birth_ref.ref)
|
||||
dobj = father_birth.get_date_object()
|
||||
if dobj.get_start_date() != Date.EMPTY:
|
||||
if not not_too_old (dobj,year - average_generation_gap):
|
||||
@ -817,8 +818,8 @@ def probably_alive(person,db,current_year=None):
|
||||
#print father.get_primary_name().get_name(), " father of ", person.get_primary_name().get_name(), " is NOT too old by birth. birth year ", dobj.get_year(), " test year ", year - average_generation_gap
|
||||
|
||||
|
||||
if father.death_handle:
|
||||
father_death = db.get_event_from_handle(father.death_handle)
|
||||
if father.death_ref:
|
||||
father_death = db.get_event_from_handle(father.death_ref.ref)
|
||||
dobj = father_death.get_date_object()
|
||||
if dobj.get_start_date() != Date.EMPTY:
|
||||
if dobj.get_year() < year - average_generation_gap:
|
||||
@ -831,8 +832,8 @@ def probably_alive(person,db,current_year=None):
|
||||
mother_handle = family.get_mother_handle()
|
||||
if mother_handle:
|
||||
mother = db.get_person_from_handle(mother_handle)
|
||||
if mother.birth_handle:
|
||||
mother_birth = db.get_event_from_handle(mother.birth_handle)
|
||||
if mother.birth_ref:
|
||||
mother_birth = db.get_event_from_handle(mother.birth_ref.ref)
|
||||
dobj = mother_birth.get_date_object()
|
||||
if dobj.get_start_date() != Date.EMPTY:
|
||||
if not not_too_old (dobj,year - average_generation_gap):
|
||||
@ -842,8 +843,8 @@ def probably_alive(person,db,current_year=None):
|
||||
#print mother.get_primary_name().get_name(), " mother of ", person.get_primary_name().get_name(), " is NOT too old by birth. birth year ", dobj.get_year(), " test year ", year - average_generation_gap
|
||||
|
||||
|
||||
if mother.death_handle:
|
||||
mother_death = db.get_event_from_handle(mother.death_handle)
|
||||
if mother.death_ref:
|
||||
mother_death = db.get_event_from_handle(mother.death_ref.ref)
|
||||
dobj = mother_death.get_date_object()
|
||||
if dobj.get_start_date() != Date.EMPTY:
|
||||
if dobj.get_year() < year - average_generation_gap:
|
||||
|
Loading…
Reference in New Issue
Block a user