Pylint cleanups and minor optimizations gen.utils.db
This commit is contained in:
Christopher Horn
Nick Hall
parent
262440fcea
commit
72553fdfcf
@ -21,75 +21,86 @@
|
||||
"""
|
||||
Utilities for getting information from the database.
|
||||
"""
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Standard python modules
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
import logging
|
||||
LOG = logging.getLogger(".gui.utils.db")
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Gramps modules
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
from ..lib.nameorigintype import NameOriginType
|
||||
from ..lib.surname import Surname
|
||||
# -------------------------------------------------------------------------
|
||||
from ..const import GRAMPS_LOCALE as glocale
|
||||
from ..display.name import displayer as name_displayer
|
||||
from ..display.place import displayer as place_displayer
|
||||
from ..const import GRAMPS_LOCALE as glocale
|
||||
from ..lib import EventType, EventRoleType, NameOriginType, Surname
|
||||
|
||||
_ = glocale.translation.sgettext
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
LOG = logging.getLogger(".gui.utils.db")
|
||||
|
||||
PRIMARY_EVENT_ROLES = [EventRoleType.FAMILY, EventRoleType.PRIMARY]
|
||||
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Fallback functions
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
def get_birth_or_fallback(db, person, format=None):
|
||||
"""
|
||||
Get BIRTH event from a person, or fallback to an event around
|
||||
the time of birth.
|
||||
"""
|
||||
birth_ref = person.get_birth_ref()
|
||||
if birth_ref: # regular birth found
|
||||
if birth_ref: # regular birth found
|
||||
event = db.get_event_from_handle(birth_ref.ref)
|
||||
if event:
|
||||
return event
|
||||
|
||||
# now search the event list for fallbacks
|
||||
for event_ref in person.get_primary_event_ref_list():
|
||||
if event_ref:
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (event
|
||||
and event.type.is_birth_fallback()
|
||||
and event_ref.role.is_primary()):
|
||||
if format:
|
||||
event.date.format = format
|
||||
return event
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (
|
||||
event
|
||||
and event.type.is_birth_fallback()
|
||||
and event_ref.role.is_primary()
|
||||
):
|
||||
if format:
|
||||
event.date.format = format
|
||||
return event
|
||||
return None
|
||||
|
||||
|
||||
def get_death_or_fallback(db, person, format=None):
|
||||
"""
|
||||
Get a DEATH event from a person, or fallback to an
|
||||
event around the time of death.
|
||||
"""
|
||||
death_ref = person.get_death_ref()
|
||||
if death_ref: # regular death found
|
||||
if death_ref: # regular death found
|
||||
event = db.get_event_from_handle(death_ref.ref)
|
||||
if event:
|
||||
return event
|
||||
|
||||
# now search the event list for fallbacks
|
||||
for event_ref in person.get_primary_event_ref_list():
|
||||
if event_ref:
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (event
|
||||
and event.type.is_death_fallback()
|
||||
and event_ref.role.is_primary()):
|
||||
if format:
|
||||
event.date.format = format
|
||||
return event
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (
|
||||
event
|
||||
and event.type.is_death_fallback()
|
||||
and event_ref.role.is_primary()
|
||||
):
|
||||
if format:
|
||||
event.date.format = format
|
||||
return event
|
||||
return None
|
||||
|
||||
|
||||
def get_age(db, person, fallback=True, calendar="gregorian"):
|
||||
"""
|
||||
Compute the age of person.
|
||||
@ -108,24 +119,23 @@ def get_age(db, person, fallback=True, calendar="gregorian"):
|
||||
death = get_death_or_fallback(db, person)
|
||||
else:
|
||||
birth_ref = person.get_birth_ref()
|
||||
if birth_ref: # regular birth found
|
||||
event = db.get_event_from_handle(birth_ref.ref)
|
||||
if event:
|
||||
birth = event
|
||||
if birth_ref: # regular birth found
|
||||
birth = db.get_event_from_handle(birth_ref.ref)
|
||||
death_ref = person.get_death_ref()
|
||||
if death_ref: # regular death found
|
||||
event = db.get_event_from_handle(death_ref.ref)
|
||||
if event:
|
||||
death = event
|
||||
if death_ref: # regular death found
|
||||
death = db.get_event_from_handle(death_ref.ref)
|
||||
|
||||
age = None
|
||||
if birth is not None:
|
||||
birth_date = birth.get_date_object().to_calendar("gregorian")
|
||||
if (birth_date and birth_date.get_valid()
|
||||
and not birth_date.is_empty()):
|
||||
if birth_date and birth_date.get_valid() and not birth_date.is_empty():
|
||||
if death is not None:
|
||||
death_date = death.get_date_object().to_calendar("gregorian")
|
||||
if (death_date and death_date.get_valid()
|
||||
and not death_date.is_empty()):
|
||||
if (
|
||||
death_date
|
||||
and death_date.get_valid()
|
||||
and not death_date.is_empty()
|
||||
):
|
||||
age = death_date - birth_date
|
||||
if not age.is_valid():
|
||||
age = None
|
||||
@ -133,6 +143,7 @@ def get_age(db, person, fallback=True, calendar="gregorian"):
|
||||
age = age.tuple()
|
||||
return age
|
||||
|
||||
|
||||
def get_timeperiod(db, person):
|
||||
"""
|
||||
Compute the timeperiod a person lived in
|
||||
@ -143,114 +154,121 @@ def get_timeperiod(db, person):
|
||||
if isinstance(person, str):
|
||||
# a handle is passed
|
||||
person = db.get_person_from_handle(person)
|
||||
|
||||
# the period is the year of birth
|
||||
birth = get_birth_or_fallback(db, person)
|
||||
if birth is not None:
|
||||
birth_date = birth.get_date_object().to_calendar("gregorian")
|
||||
if (birth_date and birth_date.get_valid()
|
||||
and not birth_date.is_empty()):
|
||||
if birth_date and birth_date.get_valid() and not birth_date.is_empty():
|
||||
return birth_date.get_year()
|
||||
|
||||
death = get_death_or_fallback(db, person)
|
||||
# no birth, period is death - 20
|
||||
if death is not None:
|
||||
death_date = death.get_date_object().to_calendar("gregorian")
|
||||
if (death_date and death_date.get_valid()
|
||||
and not death_date.is_empty()):
|
||||
if death_date and death_date.get_valid() and not death_date.is_empty():
|
||||
return death_date.get_year() - 20
|
||||
|
||||
# no birth and death, look for another event date we can use
|
||||
for event_ref in person.get_primary_event_ref_list():
|
||||
if event_ref:
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if event:
|
||||
event_date = event.get_date_object().to_calendar("gregorian")
|
||||
if (event_date and event_date.get_valid()
|
||||
and not event_date.is_empty()):
|
||||
if (
|
||||
event_date
|
||||
and event_date.get_valid()
|
||||
and not event_date.is_empty()
|
||||
):
|
||||
return event_date.get_year()
|
||||
return None
|
||||
|
||||
|
||||
def get_event_ref(db, family, event_type):
|
||||
"""
|
||||
Return a reference to a primary family event of the given event type.
|
||||
"""
|
||||
from ..lib.eventroletype import EventRoleType
|
||||
if family:
|
||||
for event_ref in family.get_event_ref_list():
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (event and event.get_type() == event_type and
|
||||
(event_ref.get_role() == EventRoleType.FAMILY or
|
||||
event_ref.get_role() == EventRoleType.PRIMARY)):
|
||||
if (
|
||||
event
|
||||
and event.get_type() == event_type
|
||||
and event_ref.get_role() in PRIMARY_EVENT_ROLES
|
||||
):
|
||||
return event_ref
|
||||
return None
|
||||
|
||||
|
||||
def get_primary_event_ref_list(db, family):
|
||||
"""
|
||||
Return a reference to the primary events of the family.
|
||||
"""
|
||||
from ..lib.eventroletype import EventRoleType
|
||||
retval = []
|
||||
events = []
|
||||
for event_ref in family.get_event_ref_list():
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (event and
|
||||
(event_ref.get_role() == EventRoleType.FAMILY or
|
||||
event_ref.get_role() == EventRoleType.PRIMARY)):
|
||||
retval.append(event_ref)
|
||||
return retval
|
||||
if event and event_ref.get_role() in PRIMARY_EVENT_ROLES:
|
||||
events.append(event_ref)
|
||||
return events
|
||||
|
||||
|
||||
def get_marriage_or_fallback(db, family, format=None):
|
||||
"""
|
||||
Get a MARRIAGE event from a family, or fallback to an
|
||||
alternative event type.
|
||||
"""
|
||||
from ..lib.eventroletype import EventRoleType
|
||||
from ..lib.eventtype import EventType
|
||||
marriage_ref = get_event_ref(db, family, EventType.MARRIAGE)
|
||||
if marriage_ref: # regular marriage found
|
||||
if marriage_ref: # regular marriage found
|
||||
event = db.get_event_from_handle(marriage_ref.ref)
|
||||
if event:
|
||||
return event
|
||||
|
||||
# now search the event list for fallbacks
|
||||
for event_ref in get_primary_event_ref_list(db, family):
|
||||
if event_ref:
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (event
|
||||
if (
|
||||
event
|
||||
and event.type.is_marriage_fallback()
|
||||
and (event_ref.role == EventRoleType.FAMILY or
|
||||
event_ref.role == EventRoleType.PRIMARY)):
|
||||
and event_ref.role in PRIMARY_EVENT_ROLES
|
||||
):
|
||||
if format:
|
||||
event.date.format = format
|
||||
return event
|
||||
return None
|
||||
|
||||
|
||||
def get_divorce_or_fallback(db, family, format=None):
|
||||
"""
|
||||
Get a DIVORCE event from a family, or fallback to an
|
||||
alternative event type.
|
||||
"""
|
||||
from ..lib.eventroletype import EventRoleType
|
||||
from ..lib.eventtype import EventType
|
||||
divorce_ref = get_event_ref(db, family, EventType.DIVORCE)
|
||||
if divorce_ref: # regular marriage found
|
||||
if divorce_ref: # regular marriage found
|
||||
event = db.get_event_from_handle(divorce_ref.ref)
|
||||
if event:
|
||||
return event
|
||||
|
||||
# now search the event list for fallbacks
|
||||
for event_ref in get_primary_event_ref_list(db, family):
|
||||
if event_ref:
|
||||
event = db.get_event_from_handle(event_ref.ref)
|
||||
if (event
|
||||
if (
|
||||
event
|
||||
and event.type.is_divorce_fallback()
|
||||
and (event_ref.role == EventRoleType.FAMILY or
|
||||
event_ref.role == EventRoleType.PRIMARY)):
|
||||
and event_ref.role in PRIMARY_EVENT_ROLES
|
||||
):
|
||||
if format:
|
||||
event.date.format = format
|
||||
return event
|
||||
return None
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Function to return the name of the main participant of an event
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
def get_participant_from_event(db, event_handle, all_=False):
|
||||
"""
|
||||
Obtain the first primary or family participant to an event we find in the
|
||||
@ -261,22 +279,27 @@ def get_participant_from_event(db, event_handle, all_=False):
|
||||
"""
|
||||
participant = ""
|
||||
ellipses = False
|
||||
result_list = list(db.find_backlink_handles(
|
||||
event_handle, include_classes=['Person', 'Family']))
|
||||
result_list = list(
|
||||
db.find_backlink_handles(
|
||||
event_handle, include_classes=["Person", "Family"]
|
||||
)
|
||||
)
|
||||
|
||||
#obtain handles without duplicates
|
||||
people = set([x[1] for x in result_list if x[0] == 'Person'])
|
||||
families = set([x[1] for x in result_list if x[0] == 'Family'])
|
||||
for personhandle in people:
|
||||
person = db.get_person_from_handle(personhandle)
|
||||
# obtain handles without duplicates
|
||||
people = set([x[1] for x in result_list if x[0] == "Person"])
|
||||
families = set([x[1] for x in result_list if x[0] == "Family"])
|
||||
for person_handle in people:
|
||||
person = db.get_person_from_handle(person_handle)
|
||||
if not person:
|
||||
continue
|
||||
for event_ref in person.get_event_ref_list():
|
||||
if event_handle == event_ref.ref and \
|
||||
event_ref.get_role().is_primary():
|
||||
if (
|
||||
event_handle == event_ref.ref
|
||||
and event_ref.get_role().is_primary()
|
||||
):
|
||||
if participant:
|
||||
if all_:
|
||||
participant += ', %s' % name_displayer.display(person)
|
||||
participant += f", {name_displayer.display(person)}"
|
||||
else:
|
||||
ellipses = True
|
||||
else:
|
||||
@ -285,16 +308,18 @@ def get_participant_from_event(db, event_handle, all_=False):
|
||||
if ellipses:
|
||||
break
|
||||
if ellipses:
|
||||
return _('%s, ...') % participant
|
||||
return _("%s, ...") % participant
|
||||
|
||||
for familyhandle in families:
|
||||
family = db.get_family_from_handle(familyhandle)
|
||||
for family_handle in families:
|
||||
family = db.get_family_from_handle(family_handle)
|
||||
for event_ref in family.get_event_ref_list():
|
||||
if event_handle == event_ref.ref and \
|
||||
event_ref.get_role().is_family():
|
||||
if (
|
||||
event_handle == event_ref.ref
|
||||
and event_ref.get_role().is_family()
|
||||
):
|
||||
if participant:
|
||||
if all_:
|
||||
participant += ', %s' % family_name(family, db)
|
||||
participant += f", {family_name(family, db)}"
|
||||
else:
|
||||
ellipses = True
|
||||
else:
|
||||
@ -304,151 +329,160 @@ def get_participant_from_event(db, event_handle, all_=False):
|
||||
break
|
||||
|
||||
if ellipses:
|
||||
return _('%s, ...') % participant
|
||||
else:
|
||||
return participant
|
||||
return _("%s, ...") % participant
|
||||
return participant
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Function to return a label to display the active object in the status bar
|
||||
# and to describe bookmarked objects.
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
def navigation_label(db, nav_type, handle):
|
||||
# -------------------------------------------------------------------------
|
||||
def navigation_label(db, nav_type, handle_or_obj):
|
||||
"""
|
||||
Return a descriptive label for an object.
|
||||
"""
|
||||
if isinstance(handle_or_obj, str):
|
||||
get_method = db.method("get_%s_from_handle", nav_type)
|
||||
obj = get_method(handle_or_obj)
|
||||
else:
|
||||
obj = handle_or_obj
|
||||
|
||||
if obj is None:
|
||||
return (None, None)
|
||||
|
||||
label = None
|
||||
obj = None
|
||||
if nav_type == 'Person':
|
||||
obj = db.get_person_from_handle(handle)
|
||||
if obj:
|
||||
label = name_displayer.display(obj)
|
||||
elif nav_type == 'Family':
|
||||
obj = db.get_family_from_handle(handle)
|
||||
if obj:
|
||||
label = family_name(obj, db)
|
||||
elif nav_type == 'Event':
|
||||
obj = db.get_event_from_handle(handle)
|
||||
if obj:
|
||||
who = get_participant_from_event(db, handle)
|
||||
desc = obj.get_description()
|
||||
label = obj.get_type()
|
||||
if desc:
|
||||
label = '%s - %s' % (label, desc)
|
||||
if who:
|
||||
label = '%s - %s' % (label, who)
|
||||
elif nav_type == 'Place':
|
||||
obj = db.get_place_from_handle(handle)
|
||||
if obj:
|
||||
label = place_displayer.display(db, obj)
|
||||
elif nav_type == 'Source':
|
||||
obj = db.get_source_from_handle(handle)
|
||||
if obj:
|
||||
label = obj.get_title()
|
||||
elif nav_type == 'Citation':
|
||||
obj = db.get_citation_from_handle(handle)
|
||||
if obj:
|
||||
label = obj.get_page()
|
||||
src = db.get_source_from_handle(obj.get_reference_handle())
|
||||
if src:
|
||||
label = src.get_title() + " " + label
|
||||
elif nav_type == 'Repository':
|
||||
obj = db.get_repository_from_handle(handle)
|
||||
if obj:
|
||||
label = obj.get_name()
|
||||
elif nav_type == 'Media' or nav_type == 'Media':
|
||||
obj = db.get_media_from_handle(handle)
|
||||
if obj:
|
||||
label = obj.get_description()
|
||||
elif nav_type == 'Note':
|
||||
obj = db.get_note_from_handle(handle)
|
||||
if obj:
|
||||
label = obj.get()
|
||||
# When strings are cut, make sure they are unicode
|
||||
#otherwise you may end of with cutting within an utf-8 sequence
|
||||
label = str(label)
|
||||
label = " ".join(label.split())
|
||||
if len(label) > 40:
|
||||
label = label[:40] + "..."
|
||||
if nav_type == "Person":
|
||||
label = name_displayer.display(obj)
|
||||
elif nav_type == "Family":
|
||||
label = family_name(obj, db)
|
||||
elif nav_type == "Event":
|
||||
who = get_participant_from_event(db, obj.get_handle())
|
||||
desc = obj.get_description()
|
||||
label = obj.get_type()
|
||||
if desc:
|
||||
label = f"{label} - {desc}"
|
||||
if who:
|
||||
label = f"{label} - {who}"
|
||||
elif nav_type == "Place":
|
||||
label = place_displayer.display(db, obj)
|
||||
elif nav_type == "Source":
|
||||
label = obj.get_title()
|
||||
elif nav_type == "Citation":
|
||||
label = obj.get_page()
|
||||
src = db.get_source_from_handle(obj.get_reference_handle())
|
||||
if src:
|
||||
label = f"{src.get_title()} {label}"
|
||||
elif nav_type == "Repository":
|
||||
label = obj.get_name()
|
||||
elif nav_type == "Media":
|
||||
label = obj.get_description()
|
||||
elif nav_type == "Note":
|
||||
label = obj.get()
|
||||
# When strings are cut, make sure they are unicode
|
||||
# otherwise you may end of with cutting within an utf-8 sequence
|
||||
label = str(label)
|
||||
label = " ".join(label.split())
|
||||
if len(label) > 40:
|
||||
label = label[:40] + "..."
|
||||
|
||||
if label and obj:
|
||||
label = '[%s] %s' % (obj.get_gramps_id(), label)
|
||||
label = f"[{obj.get_gramps_id()}] {label}"
|
||||
|
||||
return (label, obj)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Function to return children's list of a person
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
def find_children(db,p):
|
||||
# -------------------------------------------------------------------------
|
||||
def find_children(db, person):
|
||||
"""
|
||||
Return the list of all children's IDs for a person.
|
||||
"""
|
||||
childlist = []
|
||||
for family_handle in p.get_family_handle_list():
|
||||
children = set()
|
||||
for family_handle in person.get_family_handle_list():
|
||||
family = db.get_family_from_handle(family_handle)
|
||||
if family:
|
||||
for child_ref in family.get_child_ref_list():
|
||||
childlist.append(child_ref.ref)
|
||||
return childlist
|
||||
children.add(child_ref.ref)
|
||||
return list(children)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Function to return parent's list of a person
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
def find_parents(db,p):
|
||||
# -------------------------------------------------------------------------
|
||||
def find_parents(db, person):
|
||||
"""
|
||||
Return the unique list of all parents' IDs for a person.
|
||||
"""
|
||||
parentlist = []
|
||||
for f in p.get_parent_family_handle_list():
|
||||
family = db.get_family_from_handle(f)
|
||||
parents = set()
|
||||
for family_handle in person.get_parent_family_handle_list():
|
||||
family = db.get_family_from_handle(family_handle)
|
||||
father_handle = family.get_father_handle()
|
||||
if father_handle:
|
||||
parents.add(father_handle)
|
||||
mother_handle = family.get_mother_handle()
|
||||
if father_handle not in parentlist:
|
||||
parentlist.append(father_handle)
|
||||
if mother_handle not in parentlist:
|
||||
parentlist.append(mother_handle)
|
||||
return parentlist
|
||||
if mother_handle:
|
||||
parents.add(mother_handle)
|
||||
return list(parents)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Function to return persons, that share the same event.
|
||||
# This for example links witnesses to the tree
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
def find_witnessed_people(db,p):
|
||||
people = []
|
||||
for event_ref in p.get_event_ref_list():
|
||||
for l in db.find_backlink_handles( event_ref.ref):
|
||||
if l[0] == 'Person' and l[1] != p.get_handle() and l[1] not in people:
|
||||
people.append(l[1])
|
||||
if l[0] == 'Family':
|
||||
fam = db.get_family_from_handle(l[1])
|
||||
if fam:
|
||||
father_handle = fam.get_father_handle()
|
||||
if father_handle and father_handle != p.get_handle() and father_handle not in people:
|
||||
people.append(father_handle)
|
||||
mother_handle = fam.get_mother_handle()
|
||||
if mother_handle and mother_handle != p.get_handle() and mother_handle not in people:
|
||||
people.append(mother_handle)
|
||||
for f in p.get_family_handle_list():
|
||||
family = db.get_family_from_handle(f)
|
||||
for event_ref in family.get_event_ref_list():
|
||||
for l in db.find_backlink_handles( event_ref.ref):
|
||||
if l[0] == 'Person' and l[1] != p.get_handle() and l[1] not in people:
|
||||
people.append(l[1])
|
||||
for pref in p.get_person_ref_list():
|
||||
if pref.ref != p.get_handle and pref.ref not in people:
|
||||
people.append(pref.ref)
|
||||
return people
|
||||
# -------------------------------------------------------------------------
|
||||
def find_witnessed_people(db, person):
|
||||
"""
|
||||
Return list of all person handles associated with an event.
|
||||
"""
|
||||
person_handle = person.get_handle()
|
||||
people = {
|
||||
person_ref.ref
|
||||
for person_ref in person.get_person_ref_list()
|
||||
if person_ref.ref != person_handle
|
||||
}
|
||||
for event_ref in person.get_event_ref_list():
|
||||
for backlink in db.find_backlink_handles(
|
||||
event_ref.ref, include_classes=["Person", "Family"]
|
||||
):
|
||||
if backlink[0] == "Person":
|
||||
if backlink[1] != person_handle:
|
||||
people.add(backlink[1])
|
||||
continue
|
||||
if backlink[0] == "Family":
|
||||
family = db.get_family_from_handle(backlink[1])
|
||||
if family:
|
||||
father_handle = family.get_father_handle()
|
||||
if father_handle and father_handle != person_handle:
|
||||
people.add(father_handle)
|
||||
mother_handle = family.get_mother_handle()
|
||||
if mother_handle and mother_handle != person_handle:
|
||||
people.add(mother_handle)
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
for family_handle in person.get_family_handle_list():
|
||||
family = db.get_family_from_handle(family_handle)
|
||||
for event_ref in family.get_event_ref_list():
|
||||
for backlink in db.find_backlink_handles(
|
||||
event_ref.ref, include_classes=["Person"]
|
||||
):
|
||||
if backlink[1] != person_handle:
|
||||
people.add(backlink[1])
|
||||
|
||||
return list(people)
|
||||
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Iterate over ancestors.
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
def for_each_ancestor(db, start, func, data):
|
||||
"""
|
||||
Recursively iterate (breadth-first) over ancestors of
|
||||
@ -460,95 +494,104 @@ def for_each_ancestor(db, start, func, data):
|
||||
todo = start
|
||||
done_ids = set()
|
||||
while len(todo):
|
||||
p_handle = todo.pop()
|
||||
p = db.get_person_from_handle(p_handle)
|
||||
person_handle = todo.pop()
|
||||
# Don't process the same handle twice. This can happen
|
||||
# if there is a cycle in the database, or if the
|
||||
# initial list contains X and some of X's ancestors.
|
||||
if p_handle in done_ids:
|
||||
if person_handle in done_ids:
|
||||
continue
|
||||
done_ids.add(p_handle)
|
||||
if func(data, p_handle):
|
||||
if func(data, person_handle):
|
||||
return 1
|
||||
for fam_handle in p.get_parent_family_handle_list():
|
||||
fam = db.get_family_from_handle(fam_handle)
|
||||
if fam:
|
||||
f_handle = fam.get_father_handle()
|
||||
m_handle = fam.get_mother_handle()
|
||||
if f_handle: todo.append(f_handle)
|
||||
if m_handle: todo.append(m_handle)
|
||||
person = db.get_person_from_handle(person_handle)
|
||||
for family_handle in person.get_parent_family_handle_list():
|
||||
family = db.get_family_from_handle(family_handle)
|
||||
if family:
|
||||
father_handle = family.get_father_handle()
|
||||
if father_handle:
|
||||
todo.append(father_handle)
|
||||
mother_handle = family.get_mother_handle()
|
||||
if mother_handle:
|
||||
todo.append(mother_handle)
|
||||
done_ids.add(person_handle)
|
||||
return 0
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Preset a name with a name of family member
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
def preset_name(basepers, name, sibling=False):
|
||||
"""Fill up name with all family common names of basepers.
|
||||
# -------------------------------------------------------------------------
|
||||
def preset_name(base_person, name, sibling=False):
|
||||
"""
|
||||
Fill up name with all family common names of base person.
|
||||
If sibling=True, pa/matronymics are retained.
|
||||
"""
|
||||
surnlist = []
|
||||
primname = basepers.get_primary_name()
|
||||
prim = False
|
||||
for surn in primname.get_surname_list():
|
||||
if (not sibling) and (surn.get_origintype().value in
|
||||
[NameOriginType.PATRONYMIC,
|
||||
NameOriginType.MATRONYMIC]):
|
||||
surnames = []
|
||||
primary_name = base_person.get_primary_name()
|
||||
primary = False
|
||||
for surname in primary_name.get_surname_list():
|
||||
if (not sibling) and (
|
||||
surname.get_origintype().value
|
||||
in [NameOriginType.PATRONYMIC, NameOriginType.MATRONYMIC]
|
||||
):
|
||||
continue
|
||||
surnlist.append(Surname(source=surn))
|
||||
if surn.primary:
|
||||
prim=True
|
||||
if not surnlist:
|
||||
surnlist = [Surname()]
|
||||
name.set_surname_list(surnlist)
|
||||
if not prim:
|
||||
surnames.append(Surname(source=surname))
|
||||
if surname.primary:
|
||||
primary = True
|
||||
if not surnames:
|
||||
surnames = [Surname()]
|
||||
name.set_surname_list(surnames)
|
||||
if not primary:
|
||||
name.set_primary_surname(0)
|
||||
name.set_family_nick_name(primname.get_family_nick_name())
|
||||
name.set_group_as(primname.get_group_as())
|
||||
name.set_sort_as(primname.get_sort_as())
|
||||
name.set_family_nick_name(primary_name.get_family_nick_name())
|
||||
name.set_group_as(primary_name.get_group_as())
|
||||
name.set_sort_as(primary_name.get_sort_as())
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Short hand function to return either the person's name, or an empty
|
||||
# string if the person is None
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
def family_name(family, db, noname=_("unknown")):
|
||||
"""Builds a name for the family from the parents names"""
|
||||
|
||||
"""
|
||||
Builds a name for the family from the parents names.
|
||||
"""
|
||||
father = None
|
||||
mother = None
|
||||
father_handle = family.get_father_handle()
|
||||
mother_handle = family.get_mother_handle()
|
||||
if father_handle:
|
||||
father = db.get_person_from_handle(father_handle)
|
||||
|
||||
mother = None
|
||||
mother_handle = family.get_mother_handle()
|
||||
if mother_handle:
|
||||
mother = db.get_person_from_handle(mother_handle)
|
||||
|
||||
if father and mother:
|
||||
fname = name_displayer.display(father)
|
||||
mname = name_displayer.display(mother)
|
||||
name = _("%(father)s and %(mother)s") % {
|
||||
"father" : fname,
|
||||
"mother" : mname}
|
||||
elif father:
|
||||
name = name_displayer.display(father)
|
||||
elif mother:
|
||||
name = name_displayer.display(mother)
|
||||
else:
|
||||
name = noname
|
||||
return name
|
||||
return _("%(father)s and %(mother)s") % {
|
||||
"father": fname,
|
||||
"mother": mname,
|
||||
}
|
||||
if father:
|
||||
return name_displayer.display(father)
|
||||
if mother:
|
||||
return name_displayer.display(mother)
|
||||
return noname
|
||||
|
||||
#-------------------------------------------------------------------------
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
#
|
||||
# Referents functions
|
||||
#
|
||||
#-------------------------------------------------------------------------
|
||||
# -------------------------------------------------------------------------
|
||||
def get_referents(handle, db, primary_objects):
|
||||
""" Find objects that refer to an object.
|
||||
"""
|
||||
Find objects that refer to an object.
|
||||
|
||||
This function is the base for other get_<object>_referents functions.
|
||||
|
||||
"""
|
||||
# Use one pass through the reference map to grab all the references
|
||||
object_list = list(db.find_backlink_handles(handle))
|
||||
@ -558,32 +601,43 @@ def get_referents(handle, db, primary_objects):
|
||||
|
||||
for primary in primary_objects:
|
||||
primary_list = [item[1] for item in object_list if item[0] == primary]
|
||||
the_lists = the_lists + (primary_list, )
|
||||
the_lists = the_lists + (primary_list,)
|
||||
return the_lists
|
||||
|
||||
|
||||
def get_source_referents(source_handle, db):
|
||||
""" Find objects that refer the source.
|
||||
"""
|
||||
Find objects that refer the source.
|
||||
|
||||
This function finds all primary objects that refer (directly or through
|
||||
secondary child-objects) to a given source handle in a given database.
|
||||
|
||||
Only Citations can refer to sources, so that is all we need to check
|
||||
"""
|
||||
_primaries = ('Citation',)
|
||||
_primaries = ("Citation",)
|
||||
|
||||
return get_referents(source_handle, db, _primaries)
|
||||
|
||||
return (get_referents(source_handle, db, _primaries))
|
||||
|
||||
def get_citation_referents(citation_handle, db):
|
||||
""" Find objects that refer the citation.
|
||||
"""
|
||||
Find objects that refer the citation.
|
||||
|
||||
This function finds all primary objects that refer (directly or through
|
||||
secondary child-objects) to a given citation handle in a given database.
|
||||
|
||||
"""
|
||||
_primaries = ('Person', 'Family', 'Event', 'Place',
|
||||
'Source', 'Media', 'Repository')
|
||||
_primaries = (
|
||||
"Person",
|
||||
"Family",
|
||||
"Event",
|
||||
"Place",
|
||||
"Source",
|
||||
"Media",
|
||||
"Repository",
|
||||
)
|
||||
|
||||
return get_referents(citation_handle, db, _primaries)
|
||||
|
||||
return (get_referents(citation_handle, db, _primaries))
|
||||
|
||||
def get_source_and_citation_referents(source_handle, db):
|
||||
"""
|
||||
@ -621,40 +675,50 @@ def get_source_and_citation_referents(source_handle, db):
|
||||
| )
|
||||
"""
|
||||
the_lists = get_source_referents(source_handle, db)
|
||||
LOG.debug('source referents %s' % [the_lists])
|
||||
LOG.debug("source referents %s", [the_lists])
|
||||
# now, for each citation, get the objects that refer to that citation
|
||||
citation_referents_list = []
|
||||
for citation in the_lists[0]:
|
||||
LOG.debug('citation %s' % citation)
|
||||
LOG.debug("citation %s", citation)
|
||||
refs = get_citation_referents(citation, db)
|
||||
citation_referents_list += [(citation, refs)]
|
||||
LOG.debug('citation_referents_list %s' % [citation_referents_list])
|
||||
LOG.debug("citation_referents_list %s", [citation_referents_list])
|
||||
|
||||
(citation_list) = the_lists
|
||||
the_lists = (citation_list, citation_referents_list)
|
||||
|
||||
LOG.debug('the_lists %s' % [the_lists])
|
||||
LOG.debug("the_lists %s", [the_lists])
|
||||
return the_lists
|
||||
|
||||
|
||||
def get_media_referents(media_handle, db):
|
||||
""" Find objects that refer the media object.
|
||||
"""
|
||||
Find objects that refer the media object.
|
||||
|
||||
This function finds all primary objects that refer
|
||||
to a given media handle in a given database.
|
||||
|
||||
"""
|
||||
_primaries = ('Person', 'Family', 'Event', 'Place', 'Source', 'Citation')
|
||||
_primaries = ("Person", "Family", "Event", "Place", "Source", "Citation")
|
||||
|
||||
return get_referents(media_handle, db, _primaries)
|
||||
|
||||
return (get_referents(media_handle, db, _primaries))
|
||||
|
||||
def get_note_referents(note_handle, db):
|
||||
""" Find objects that refer a note object.
|
||||
"""
|
||||
Find objects that refer a note object.
|
||||
|
||||
This function finds all primary objects that refer
|
||||
to a given note handle in a given database.
|
||||
|
||||
"""
|
||||
_primaries = ('Person', 'Family', 'Event', 'Place',
|
||||
'Source', 'Citation', 'Media', 'Repository')
|
||||
_primaries = (
|
||||
"Person",
|
||||
"Family",
|
||||
"Event",
|
||||
"Place",
|
||||
"Source",
|
||||
"Citation",
|
||||
"Media",
|
||||
"Repository",
|
||||
)
|
||||
|
||||
return (get_referents(note_handle, db, _primaries))
|
||||
return get_referents(note_handle, db, _primaries)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user