def transform(string, year):
"""Where possible, replace all runner's with their database ID numbers.
May raise a key error."""
result_list = load(string)
schools = [(school, Regex("|".join(school.names())))
for school in Schools.select()]
for (school, pattern) in schools:
for item in result_list:
if pattern.match(item["school"]):
item["school_id"] = school.id
del item["school"]
join = INNERJOINOn(Runners, Affiliations,
Runners.q.id == Affiliations.q.runner)
for runner in Runners.select(Affiliations.q.year == year, join=join):
for name in runner.given_names:
last_first = r"%s,\s*%s" % (runner.surname, name)
first_last = r"%s\s*%s" % (name, runner.surname)
pattern = Regex(last_first + "|" + first_last, IGNORECASE)
for item in result_list:
if pattern.match(item["name"]):
item["runner_id"] = runner.id
del item["name"]
del item["school_id"]
return dump(result_list)
def main(options, arguments):
runners = irows(Runners.select())
for runner in Table(runners, column_seperator=" | "):
print runner
