Задача про Картошку

class Potato:
    states = {0: 'Отсутствует', 1: 'Росток', 2: 'Зелёная', 3: 'Зрелая'}
 
    def __init__(self, index):
        self.index = index
        self.state = 0
 
    def grow(self):
        if self.state < 3:
            self.state += 1
            self.print_stats()
 
    def is_ripe(self):
        if self.state == 3:
            return True
        return False
 
    def print_stats(self):
        print('Картошка {} сейчас {}'.format(
            self.index, Potato.states[self.state]
        ))
 
 
class PotatoGarden:
 
    def __init__(self, count):
        self.potatoes = [Potato(index) for index in range(1, count + 1)]
 
    def grow_all(self):
        print('Картошка прорастает!')
        for i_potato in self.potatoes:
            i_potato.grow()
 
    def are_all_ripe(self):
        if not all([i_potato.is_ripe() for i_potato in self.potatoes]):
            print('Картошка ещё не созрела!')
        else:
            print('Вся картошка созрела! Можно собирать\n')
 
 
class Gardener:
    def __init__(self, name, harvest):
        self.name, self.harvest = name, harvest
 
    def info_gardener(self):
        print(f'Имя садовника: {self.name}\nСобрал картошки: {self.harvest}')
 
    def action(worker, my_garden):
        if all([i_potato.is_ripe() for i_potato in my_garden.potatoes]):
            quest = int(input('Собрать картошку? \n1 - да, 2 - нет\n'))
            if quest == 1:
                potato_count = 0
                for i_potato in my_garden.potatoes:
                    worker.harvest += 1
                    potato_count += 1
                    i_potato.state = 0
                print(f'{worker.name} собрал {potato_count} картофелин!')
                worker.info_gardener()
        else:
            quest = int(input(f'Отправить {worker.name}а ухаживать за картошкой? \n 1 - да, 2 - нет\n'))
            if quest == 1:
                my_garden.grow_all()
                my_garden.are_all_ripe()
 
 
my_garden = PotatoGarden(5)
worker = Gardener('Ostin', 0)
while True:
    Gardener.action(worker, my_garden)

if quest == 1:
    # работаем по одной схеме
elif quest == 2:
    # работаем по другой схеме

if quest == 1:
    # работаем по одной схеме
if quest == 2:
    # работаем по другой схеме