woodman

from random import choice


class TinWoodman:
    wood_data = {'Молодая сосна': {'blows': 1, 'oil': 10, 'income': 2},
                 'Ель': {'blows': 2, 'oil': 7.5, 'income': 3},
                 'Кедр': {'blows': 3, 'oil': 6, 'income': 4},
                 'Дуб': {'blows': 5, 'oil': 8, 'income': 10}}
    oil = [100, 0]
    overall_wood = 0
    wood_planks = 0
    tree_health = None
    cur_tree = None
    die = False
    main_script = "Выберите действие:\n1 - поиск дерева;\n2 - пополнить масло;\n3 - статистика;\n4 - вернуться домой.\n"
    tree_script = "Выберите действие:\n1 - срубить дерево;\n2 - пополнить масло;\n3 - статистика;\n4 - вернуться домой.\n"

    def main_work(self):
        request = 1
        print('Дровосек начинает работу в лесу')
        while request != 4 and not self.die:
            if self.cur_tree == None:
                print(self.main_script)
            else:
                print(self.tree_script)
            request = int(input('Ответ:'))
            if self.cur_tree == None:
                if request == 1:
                    self.search_tree()
                elif request == 2:
                    self.fill_oil()
                elif request == 3:
                    self.stat()
            else:
                if request == 1:
                    self.cut_tree()
                elif request == 2:
                    self.fill_oil()
                elif request == 3:
                    self.stat()
        if self.die:
            print('О нет! Масло закончилось!')
            print('Дровосек заржавел, игра окончена.')
            print()
            self.stat()
        else:
            self.end_work()

    def search_tree(self):
        self.cur_tree = choice(list(self.wood_data.keys()))
        self.tree_health = 0
        print('Дровосек находит', self.cur_tree + '!')

    def cut_tree(self):
        self.tree_health += 1
        self.lower_oil()
        print('Дровосек делает удар по', self.cur_tree,
              '(' + str(self.tree_health) + '/' + str(self.wood_data[self.cur_tree]['blows']) + ')')
        print('Масло', '(' + str(self.oil[0]) + '/100)')
        if self.die:
            return
        if self.tree_health == self.wood_data[self.cur_tree]['blows']:
            self.overall_wood += self.wood_data[self.cur_tree]['income']
            self.wood_planks += 1
            print('Дерево упало, дровосек получил', self.wood_data[self.cur_tree]['income'], 'доски.')
            self.cur_tree = None

    def lower_oil(self):
        self.oil[0] -= self.wood_data[self.cur_tree]['oil']
        if self.oil[0] < 0:
            self.die = True

    def fill_oil(self):
        print('Дровосек достает маслёнку')
        self.oil[1] += 100 - self.oil[0]
        self.oil[0] = 100
        print('Масло 100/100')

    def stat(self):
        print('Статистика:')
        print('Масло:', str(self.oil[0]) + '/100')
        print('Израсходовано всего масла:', self.oil[1])
        print('Доски:', self.overall_wood)
        print('Срублено деревьев:', self.wood_planks)

    def end_work(self):
        print('После тяжелого трудового дня дровосек возвращается домой')
        self.stat()


tw = TinWoodman()
tw.main_work()