def parse(gedcom):
"""Parse each line of the GEDCOM file"""
errors = [] # Collect errors here
valid_tags = [["INDI", "FAM", "HEAD", "TRLR", "NOTE"],
["NAME", "SEX", "BIRT", "DEAT", "FAMC","FAMS",
"MARR", "HUSB", "WIFE", "CHIL", "DIV"],
["DATE"]]
def get_tag(parts):
"""Get tag from list of line parts"""
if parts[0] == "0" and (parts[2] == "INDI" or parts[2] == "FAM"):
return parts[2]
return parts[1]
def is_valid(level, tag):
"""Determine whether tag in line is valid"""
if tag in valid_tags[int(level)]:
return "Y"
return "N"
def get_args(level, tag, parts):
"""Get rest of arguments from line"""
parts.remove(level)
parts.remove(tag)
return " ".join(parts)
def add_to_dict(cur_key, tag, args):
"""Add current line to appropiate dictionary"""
if cur_key[0] == "I":
if tag == "FAMS":
if tag not in individuals[cur_key]:
individuals[cur_key][tag] = [args]
else:
individuals[cur_key][tag].append(args)
else:
individuals[cur_key][tag] = args
else:
if tag == "CHIL":
if tag not in families[cur_key]:
families[cur_key][tag] = [args]
else:
families[cur_key][tag].append(args)
else:
families[cur_key][tag] = args
individuals = {}
families = {}
cur_key = ""
second_key = ""
for line in gedcom:
line_parts = line.rstrip().split(" ")
level = line_parts[0]
tag = get_tag(line_parts)
valid = is_valid(level, tag)
args = get_args(level, tag, line_parts)
if valid:
if level == "0":
if tag == "INDI":
cur_key = args
if args in individuals:
errors.append("Error US22: Multiple individuals found with same ID: {}".format(args))
if args not in individuals:
individuals[args] = {}
if tag == "FAM":
cur_key = args
if args in families:
errors.append("Error US22: Multiple families found with same ID: {}".format(args))
if args not in families:
families[args] = {}
elif level == "1":
if tag == "BIRT" or tag == "DEAT" or tag == "MARR" or tag == "DIV":
add_to_dict(cur_key, tag, "")
second_key = tag
else:
add_to_dict(cur_key, tag, args)
else:
if second_key == "":
add_to_dict(cur_key, tag, args)
if not valid_date(args):
errors.append("Error US42: Invalid Date ({})".format(args))
else:
add_to_dict(cur_key, second_key, args)
second_key = ""
if not valid_date(args):
errors.append("Error US42: Invalid Date ({})".format(args))
return (individuals, families, errors)
def test_invalid_leap_year(self):
"""Checks if date is invalid when not leap year"""
test = date_checker.valid_date('29 FEB 2017')
self.assertFalse(test)
def test_invalid_date(self):
"""Checks if date is invalid when obviously wrong"""
test = date_checker.valid_date('87 APR 2017')
self.assertFalse(test)
def test_valid_leap_year(self):
"""Checks if date is valid when leap year"""
test = date_checker.valid_date('29 FEB 2016')
self.assertTrue(test)
def test_valid_date(self):
"""Checks if date is valid"""
test = date_checker.valid_date('19 APR 2017')
self.assertTrue(test)