swcarpentry sorting hat

sample_responses.csv

sortinghat.py

import sys, csv
"""
Usage: sortinghat.py [num_rooms] < survey_results.csv
    num_rooms   - number of rooms to divide participants up into
"""

answers = {
    8 : {
        "" : 0,
        "I have never programmed." : 0,
        'I program less than one a year.' : 1,
        "I program several times a year." : 2,
        "I program once a month." : 3,
        "I program once a week or more." : 4
    },
    10 : {
        "" : 0,
        'I could not complete this task.' : 0,
        'I could complete the task with documentation or search engine help.' : 1,
        'I could complete the task with little or no documentation or search engine help.' : 2
    },
    11 : {
        "" : 0,
        'I am not familiar with Git.' : 0,
        'I am familiar only with the name Git.' : 1,
        'I am familiar with Git but have never used it.' : 2,
        'I am familiar with Git because I have used or am using it.' : 3
    },
    13 : {
        "" : 0,
        'I am not familiar with\xc2\xa0unit testing or code coverage.' : 0,
        'I am familiar only with the terms "unit testing" and "code coverage".' : 1,
        'I am familiar with\xc2\xa0unit testing or code coverage\xc2\xa0but have never used it.' : 2,
        'I am familiar with\xc2\xa0unit testing or code coverage\xc2\xa0because I have used or am using them.' : 3
    },
    17 : {
        "" : 0,
        'I am not familiar with\xc2\xa0the command line.' : 0,
        'I am familiar only with the\xc2\xa0term "command line".' : 1,
        'I am familiar with\xc2\xa0the command line\xc2\xa0but have never used it.' : 2,
        'I am familiar with\xc2\xa0the command line\xc2\xa0because I have used or am using it.' : 3
    }
}

reader = csv.reader(sys.stdin)


try:
    people = []
    for j, row in enumerate( reader ):
        if j == 0: continue
        total = 0
        for i in answers.keys():
            total += answers[i][row[i].strip()]
        people.append({'score': total, 'name': row[1], 'email': row[2]})

    people.sort(key = lambda x: x['score'])
    if len(sys.argv) > 1:
        rooms = int(sys.argv[1])
        room_size = len(people)//rooms
        for room in xrange(0,len(people), room_size):
            members = people[room:room+room_size]
            print "Room {0}: scores {1[score]}-{2[score]}".format(room, members[0], members[-1])
            print ", ".join(map(lambda x: x['email'], members))
            print
    else: 
        for person in people:
            print "{score} {name} {email}".format(**person)

except IndexError, e:
    print >> sys.stderr, 'INDEX', row, '::', i
except KeyError, e:
    print >> sys.stderr, 'KEY', e, row, '::', answers[i].keys()