yalies_csv_kalla.py

import yalies
import os
import csv

api = yalies.API(os.environ['YALIES_API_KEY'])

people = api.people(filters={'school_code': 'YC'})

columns = [
    'first_name', 'last_name', 'email', 'college', 'college_code', 'year', 'leave', 'major', 'address',
]
with open('undergrads.csv', 'w') as csv_file:
    writer = csv.writer(csv_file)
    writer.writerow(columns)
    for person in people:
        writer.writerow([person.raw[key] for key in columns])