mirror of
https://github.com/Xevion/exercism.git
synced 2025-12-06 11:15:00 -06:00
16 lines
541 B
Python
16 lines
541 B
Python
class School(object):
|
|
def __init__(self):
|
|
self.data = {}
|
|
|
|
# Just an easier way of handling self.grade & self.add_student, less typing
|
|
def get(self, grade):
|
|
return self.data[grade] if grade in self.data.keys() else []
|
|
|
|
def add_student(self, name, grade):
|
|
self.data[grade] = self.get(grade) + [name]
|
|
|
|
def roster(self):
|
|
return sum([sorted(seq[1]) for seq in sorted(self.data.items(), key=lambda item : item[0])], [])
|
|
|
|
def grade(self, grade):
|
|
return sorted(self.get(grade)) |