Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Match alternative names in HasPlace filter

Browse Source
This commit is contained in:
Nick Hall 2014-05-24 20:43:32 +01:00
parent dd77bf74d5
commit 2578252d11
3 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
gramps/gen/filters/rules/place/_hasplace.py
View File

@ -90,7 +90,16 @@ class HasPlace(Rule):
Check each location for a match.
"""
for place_type, field in self.TYPE2FIELD.items():
name = location.get(place_type, '')
if not self.match_substring(field, name):
name_list = location.get(place_type, [''])
if not self.match_name(field, name_list):
return False
return True
def match_name(self, field, name_list):
"""
Match any name in a list of names.
"""
for name in name_list:
if self.match_substring(field, name):
return True
return False

9
gramps/gen/lib/place.py
View File

@ -321,6 +321,15 @@ class Place(CitationBase, NoteBase, MediaBase, UrlBase, PrimaryObject):
"""
return self.name
def get_all_names(self):
"""
Return a list of all names of the Place object.
:returns: Returns a list of all names of the Place
:rtype: list
"""
return [self.name] + self.alt_names
def set_longitude(self, longitude):
"""
Set the longitude of the Place object.

4
gramps/gen/utils/location.py
View File

@ -64,14 +64,14 @@ def get_locations(db, place):
containing dictionaries of place types and names.
"""
locations = []
todo = [(place, [(int(place.get_type()), place.get_name())])]
todo = [(place, [(int(place.get_type()), place.get_all_names())])]
while len(todo):
place, tree = todo.pop()
if len(place.get_placeref_list()):
for parent in place.get_placeref_list():
parent_place = db.get_place_from_handle(parent.ref)
parent_tree = tree + [(int(parent_place.get_type()),
parent_place.get_name())]
parent_place.get_all_names())]
todo.append((parent_place, parent_tree))
else:
locations.append(dict(tree))