Add Garden class implementation for kindergarten garden

This commit is contained in:
Maksym Kurets 2026-10-13 19:34:53 +02:00
parent 63a4107fd4
commit 5b720afa92

35
kindergarten_garden.py Normal file
View File

@ -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]