From 5b720afa9274f3f28b2d3e278cc61ff3e249a677 Mon Sep 17 00:00:00 2001 From: Maksym Kurets Date: Tue, 13 Oct 2026 19:34:53 +0200 Subject: [PATCH] Add Garden class implementation for kindergarten garden --- kindergarten_garden.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 kindergarten_garden.py diff --git a/kindergarten_garden.py b/kindergarten_garden.py new file mode 100644 index 0000000..f688b33 --- /dev/null +++ b/kindergarten_garden.py @@ -0,0 +1,35 @@ +class NoSuchStudentException(Exception): + pass + +class Garden: + def __init__(self, diagram, students=None): + self.diagram = diagram + if students is None: + self.students = [ + "Alena", "Boris", "Cyril", "Daniel", "Eva", "František", + "Gitka", "Helena", "Ivana", "Jozef", "Karol", "Ladislav" + ] + else: + self.students = sorted(students) + + self.plant_names = { + "G": "Tráva", + "C": "Ďatelina", + "R": "Reďkovky", + "V": "Fialky" + } + + def plants(self, student): + if student not in self.students: + raise NoSuchStudentException("Student not found") + + index = self.students.index(student) + start = index * 2 + + rows = self.diagram.splitlines() + row1 = rows[0] + row2 = rows[1] + + letters = row1[start:start+2] + row2[start:start+2] + + return [self.plant_names[char] for char in letters]