sortinghat.py

class House:
	name = None
	ability = None
	students = None

class Student:
	name = None
	ability = None

# houses

gryffindor = House()
gryffindor.name = 'Gryffindor'
gryffindor.ability = 'bravery'

slytherin = House()
slytherin.name = 'Slytherin'
slytherin.ability = 'cunning'

# students

harry = Student()
harry.name = 'Harry Potter'
harry.ability = 'bravery'

draco = Student()
draco.name = 'Draco Malfoy'
draco.ability = 'cunning'

all_houses = [gryffindor, slytherin]
all_students = [harry, draco]

def sort(student):
	for house in all_houses:
		if house.ability == student.ability:
			if house.students is None:
				house.students = []

			house.students.append(student)
			print student.name + ' goes to ' + house.name + '!'