diff --git a/common.py b/common.py index 0dfb172..2275391 100644 --- a/common.py +++ b/common.py @@ -40,4 +40,5 @@ mapper_data = { "ai/an": "American Indian/Alaskan Native", "None": "Unknown", "nh/opi": "Native Hawaiian/Other Pacific Islander", + "i": "Intern", } diff --git a/interndata.py b/interndata.py new file mode 100644 index 0000000..f855c4c --- /dev/null +++ b/interndata.py @@ -0,0 +1,27 @@ +import json + +intern_data_file = "data/staffrecord.json" + +def get_staff_records(): + with open(intern_data_file, "r") as fd: + return json.loads(fd.read()) + +def map_name_to_record(name, data, first=True): + for i in data["payload"]: + if first: + if i["name"].split()[0] == name: + return i["innerpayload"] + else: + if i["name"] == name: + return i["innerpayload"] + + return None + +def return_intern_data_from_present_array(array, data): + output = [] + for i in array: + record = map_name_to_record(i, data) + if record: + output.append(record) + + return output