diff --git a/u5/kindergarten_garden.py b/u5/kindergarten_garden.py new file mode 100644 index 0000000..10f0dc8 --- /dev/null +++ b/u5/kindergarten_garden.py @@ -0,0 +1,22 @@ +class NoSuchStudentException(Exception): + pass + + +class Garden: + STUDENTS = ["Alena", "Boris", "Cyril", "Daniel", "Eva", "František", + "Gitka", "Helena", "Ivana", "Jozef", "Karol", "Ladislav"] + PLANTS = {"G": "Tráva", "C": "Ďatelina", "R": "Reďkovky", "V": "Fialky"} + + def __init__(self, diagram, students=None): + self.rows = diagram.split("\n") + self.students = sorted(students) if students else self.STUDENTS + + def plants(self, student): + if student not in self.students: + raise NoSuchStudentException(student) + i = self.students.index(student) * 2 + letters = self.rows[0][i:i+2] + self.rows[1][i:i+2] + result = [] + for letter in letters: + result.append(self.PLANTS[letter]) + return result \ No newline at end of file