def test_invalidate_cache():
parser = Parser()
parser.parse_file('tests/files/Musterstammbaum.ged')
assert len(parser.get_element_list()) == 396
assert len(parser.get_element_dictionary()) == 32
parser.invalidate_cache()
assert len(parser.get_element_list()) == 396
assert len(parser.get_element_dictionary()) == 32
def test_parse_file():
parser = Parser()
assert len(parser.get_root_child_elements()) == 0
parser.parse_file('tests/files/Musterstammbaum.ged')
assert len(parser.get_root_child_elements()) == 34
individuals_in_root_child_elements = 0
individuals_in_element_list = 0
for element in parser.get_root_child_elements():
if isinstance(element, IndividualElement):
individuals_in_root_child_elements += 1
for element in parser.get_element_list():
if isinstance(element, IndividualElement):
individuals_in_element_list += 1
assert individuals_in_root_child_elements == 20
assert individuals_in_element_list == 20
def processGEDCOM(file_path):
gedcom_parser = Parser()
gedcom_parser.parse_file(file_path)
root_elements = gedcom_parser.get_element_list()
info = {"INDI": [], "FAM": []}
for element in root_elements:
if str(element.get_tag()) == "INDI" or str(element.get_tag()) == "FAM":
if element.get_tag() == "INDI" and len(
info[element.get_tag()]) > 5000:
raise ValueError("Too many individuals in file")
if element.get_tag() == "FAM" and len(
info[element.get_tag()]) > 1000:
raise ValueError("Too many families in file")
months = {
"JAN": 1,
"FEB": 2,
"MAR": 3,
"APR": 4,
"MAY": 5,
"JUN": 6,
"JUL": 7,
"AUG": 8,
"SEP": 9,
"OCT": 10,
"NOV": 11,
"DEC": 12
}
indiDict = OrderedDict()
myTag = ""
famDict = OrderedDict()
famTag = ""
isMarried = False
isDivorced = False
for element in root_elements:
age = 0
if (element.get_level() == 0 and element.get_tag() == "INDI"):
individualString = element.to_gedcom_string()
individualString = individualString.replace('@',
'').strip().split(" ")
myTag = individualString[1]
indiDict[myTag] = indiClass(myTag)
indiDict[myTag].Set_ID(myTag)
if (element.get_level() == 1) and element.get_tag() == "FAMC":
childString = element.to_gedcom_string()
childString = childString.replace('@', '').strip().split(" ")
indiDict[myTag].Set_child(childString[2])
if (element.get_level() == 1) and element.get_tag() == "FAMS":
spouseString = element.to_gedcom_string()
spouseString = spouseString.replace('@', '').strip().split(" ")
indiDict[myTag].Set_spouse(spouseString[2])
if isinstance(element, IndividualElement):
(first, last) = element.get_name()
indiDict[myTag].Set_name(str(first + " " + last))
indiDict[myTag].Set_gender(element.get_gender())
if (element.is_deceased() == True):
indiDict[myTag].Set_alive(False)
bday = element.get_birth_data()[0]
death = element.get_death_data()[0]
bday = bday.split(" ")
bday = datetime.date(int(bday[2]), int(months[bday[1]]),
int(bday[0]))
death = death.split(" ")
death = datetime.date(int(death[2]), int(months[death[1]]),
int(death[0]))
indiDict[myTag].Set_birthday(bday)
indiDict[myTag].Set_death(death)
age = from_dob_to_death(bday, death)
indiDict[myTag].Set_age(age)
else:
indiDict[myTag].Set_alive(True)
bday = element.get_birth_data()[0].split(" ")
bday = datetime.date(int(bday[2]), int(months[bday[1]]),
int(bday[0]))
indiDict[myTag].Set_birthday(bday)
age = from_dob_to_age(bday)
indiDict[myTag].Set_age(age)
if (element.get_level() == 0 and element.get_tag() == "FAM"):
familyString = element.to_gedcom_string()
familyString = familyString.replace('@', '').strip().split(" ")
famTag = familyString[1]
famDict[famTag] = family(famTag)
famDict[famTag].Set_ID(famTag)
if (element.get_level() == 1 and element.get_tag() == "MARR"):
isMarried = True
if (isMarried and element.get_tag() == "DATE"
and element.get_level() == 2):
marriedDay = element.get_value()
marriedDay = marriedDay.split(" ")
marriedDay = datetime.date(int(marriedDay[2]),
int(months[marriedDay[1]]),
int(marriedDay[0]))
famDict[famTag].Set_married(marriedDay)
isMarried = False
if (element.get_tag() == "DIV" and element.get_level() == 1):
isDivorced = True
if (isDivorced and element.get_tag() == "DATE"
and element.get_level() == 2):
divorcedDay = element.get_value()
divorcedDay = divorcedDay.split(" ")
divorcedDay = datetime.date(int(divorcedDay[2]),
int(months[divorcedDay[1]]),
int(divorcedDay[0]))
famDict[famTag].Set_divorced(divorcedDay)
isDivorced = False
if (element.get_level() == 1 and element.get_tag() == "HUSB"):
husbStr = element.to_gedcom_string()
husbStr = husbStr.replace('@', '').strip().split(" ")[2]
famDict[famTag].Set_husbandID(husbStr)
famDict[famTag].Set_husbandName(indiDict[husbStr].Get_name())
if (element.get_level() == 1 and element.get_tag() == "WIFE"):
wifeStr = element.to_gedcom_string()
wifeStr = wifeStr.replace('@', '').strip().split(" ")[2]
famDict[famTag].Set_wifeID(wifeStr)
famDict[famTag].Set_wifeName(indiDict[wifeStr].Get_name())
if (element.get_level() == 1 and element.get_tag() == "CHIL"):
child = element.to_gedcom_string()
child = child.replace('@', '').strip().split(" ")[2]
famDict[famTag].Set_children(child)
return indiDict, famDict
# US27: Include person's current age when listing individuals
# Author: Kristin Kim
from datetime import date
from gedcom.parser import Parser
from gedcom.element.individual import IndividualElement
file_path = 'DiazJGedcomProject1.ged'
gedcom_parser = Parser()
gedcom_parser.parse_file(file_path, False) # Disable strict parsing
elements = gedcom_parser.get_element_list()
today = date.today()
d4 = today.strftime("%d %b %Y").split(" ") # ['09', 'Oct', '2020']
t_day = int(d4[0])
t_month = d4[1].upper()
t_year = d4[2]
month_list = [
"JAN", "FEB", "MAR", "APR", "MAY", "JUN", "JUL", "AUG", "SEP", "OCT",
"NOV", "DEC"
]
t_month_int = month_list.index(t_month) + 1 # JAN=1, FEB=2, DEC=12
n = []
def individual_age(a):
def processGedFile(file_path):
# Path to your `.ged` file
#file_path ='FamilyTree.ged'
#file_path ='gedcom'
# Initialize the parser
gedcom_parser = Parser()
# Parse your file
gedcom_parser.parse_file(file_path)
root_elements = gedcom_parser.get_element_list()
info = {"INDI": [], "FAM": []}
for element in root_elements:
if str(element.get_tag()) == "INDI" or str(element.get_tag()) == "FAM":
if element.get_tag() == "INDI" and len(
info[element.get_tag()]) > 5000:
raise ValueError("Too many individuals in file")
if element.get_tag() == "FAM" and len(
info[element.get_tag()]) > 1000:
raise ValueError("Too many families in file")
months = {
"JAN": 1,
"FEB": 2,
"MAR": 3,
"APR": 4,
"MAY": 5,
"JUN": 6,
"JUL": 7,
"AUG": 8,
"SEP": 9,
"OCT": 10,
"NOV": 11,
"DEC": 12
}
# Individual dictionary
indiDict = OrderedDict()
myTag = ""
famDict = OrderedDict()
famTag = ""
isMarried = False
isDivorced = False
indiDict["DupI_ID"] = indiClass("DupI_ID")
famDict["DupliID_fam"] = familyClass("DupliID_fam")
#Line Dictionary for the items
lines_dict = dict()
lines_dict["Duplicate Ind"] = list()
lines_dict["Duplicate Fam"] = list()
line_count = 0
for element in root_elements:
age = 0
line_count = line_count + 1
# Fetch Individual ID details
if (element.get_level() == 0 and element.get_tag() == "INDI"):
individualString = element.to_gedcom_string()
individualString = individualString.replace('@',
'').strip().split(" ")
#Fetch Individual ID tag
myTag = individualString[1]
if myTag in indiDict:
indiDict["DupI_ID"].Set_DupliID(myTag)
lines_dict["Duplicate Ind"].append(line_count)
continue
else:
indiDict[myTag] = indiClass(myTag)
indiDict[myTag].Set_ID(myTag)
lines_dict[str(f"{myTag}: ID set")] = line_count
# Fetch ans set Child ID details for individual
if (element.get_level() == 1) and element.get_tag() == "FAMC":
childString = element.to_gedcom_string()
childString = childString.replace('@', '').strip().split(" ")
indiDict[myTag].Set_child(childString[2])
lines_dict[str(
f"{myTag}: {childString[2]} child set")] = line_count
# Fetch ans set Spouse ID details for individual
if (element.get_level() == 1) and element.get_tag() == "FAMS":
spouseString = element.to_gedcom_string()
spouseString = spouseString.replace('@', '').strip().split(" ")
indiDict[myTag].Set_spouse(spouseString[2])
lines_dict[str(
f"{myTag}: {spouseString[2]} spouse set")] = line_count
if isinstance(element, IndividualElement):
# Fetch and set the name
(first, last) = element.get_name()
indiDict[myTag].Set_name(str(first + " " + last))
lines_dict[str(
f"{myTag}: First and last name set")] = line_count + 1
# Fetch the Gender and set the gender
indiDict[myTag].Set_gender(element.get_gender())
lines_dict[str(f"{myTag}: Gender set")] = line_count + 5
# Check if individual is alive
if (element.is_deceased() == True):
# Set Alive status to false
indiDict[myTag].Set_alive(False)
lines_dict[str(f"{myTag}: Living status set")] = line_count + 9
# Fetch Birth and death dates
bday = element.get_birth_data()[0]
death = element.get_death_data()[0]
#fetch birth and death places
bplace = element.get_birth_data()[1]
#dplace = element.get_death_data()[1]
#get occupation
#occupation = element.get_occupation()
# Format Dates in Day Month and Year
bday = bday.split(" ")
bday = datetime.date(int(bday[2]), int(months[bday[1]]),
int(bday[0]))
death = death.split(" ")
death = datetime.date(int(death[2]), int(months[death[1]]),
int(death[0]))
# Set Birthday
indiDict[myTag].Set_birthday(bday)
lines_dict[str(f"{myTag}: Birthday set")] = line_count + 7
# Set Death day
indiDict[myTag].Set_death(death)
#add death day if birth place is not empty
if len(bplace) > 0:
lines_dict[str(
f"{myTag}: Death date set")] = line_count + 10
else:
lines_dict[str(
f"{myTag}: Death date set")] = line_count + 9
# Calculate the age
age = from_dob_to_death(bday, death)
# Set Age
indiDict[myTag].Set_age(age)
else:
# Set Alive status to true
indiDict[myTag].Set_alive(True)
# Fetch Birth dates
bday = element.get_birth_data()[0].split(" ")
bday = datetime.date(int(bday[2]), int(months[bday[1]]),
int(bday[0]))
# Set Birthday
indiDict[myTag].Set_birthday(bday)
lines_dict[str(f"{myTag}: Birthday set")] = line_count + 7
# Calculate the age
age = from_dob_to_age(bday)
# Set Age
indiDict[myTag].Set_age(age)
if (element.get_level() == 0 and element.get_tag() == "FAM"):
familyString = element.to_gedcom_string()
familyString = familyString.replace('@', '').strip().split(" ")
famTag = familyString[1]
if famTag in famDict:
famDict["DupliID_fam"].Set_DupliID_fam(famTag)
lines_dict["Duplicate Fam"].append(line_count)
continue
else:
famDict[famTag] = familyClass(famTag)
famDict[famTag].Set_ID(famTag)
lines_dict[str(f"{famTag}: ID set")] = line_count
if (element.get_level() == 1 and element.get_tag() == "MARR"):
isMarried = True
if (isMarried and element.get_tag() == "DATE"
and element.get_level() == 2):
marriedDay = element.get_value()
marriedDay = marriedDay.split(" ")
marriedDay = datetime.date(int(marriedDay[2]),
int(months[marriedDay[1]]),
int(marriedDay[0]))
famDict[famTag].Set_married(marriedDay)
lines_dict[str(f"{famTag}: Married date set")] = line_count
isMarried = False
if (element.get_tag() == "DIV" and element.get_level() == 1):
isDivorced = True
if (isDivorced and element.get_tag() == "DATE"
and element.get_level() == 2):
divorcedDay = element.get_value()
divorcedDay = divorcedDay.split(" ")
divorcedDay = datetime.date(int(divorcedDay[2]),
int(months[divorcedDay[1]]),
int(divorcedDay[0]))
famDict[famTag].Set_divorced(divorcedDay)
lines_dict[str(f"{famTag}: Divorced date set")] = line_count
isDivorced = False
if (element.get_level() == 1 and element.get_tag() == "HUSB"):
husbStr = element.to_gedcom_string()
husbStr = husbStr.replace('@', '').strip().split(" ")[2]
famDict[famTag].Set_husbandID(husbStr)
lines_dict[str(f"{famTag}: Husband ID set")] = line_count
famDict[famTag].Set_husbandName(indiDict[husbStr].Get_name())
lines_dict[str(f"{famTag}: Husband name set")] = line_count
if (element.get_level() == 1 and element.get_tag() == "WIFE"):
wifeStr = element.to_gedcom_string()
wifeStr = wifeStr.replace('@', '').strip().split(" ")[2]
famDict[famTag].Set_wifeID(wifeStr)
lines_dict[str(f"{famTag}: Wife ID set")] = line_count
famDict[famTag].Set_wifeName(indiDict[wifeStr].Get_name())
lines_dict[str(f"{famTag}: Wife name set")] = line_count
if (element.get_level() == 1 and element.get_tag() == "CHIL"):
child = element.to_gedcom_string()
child = child.replace('@', '').strip().split(" ")[2]
famDict[famTag].Set_children(child)
lines_dict[str(
f"{famTag}: {child} added to children")] = line_count
return indiDict, famDict, lines_dict
