HW10_RobotCleaner.py

import time


class LowBattery(Exception):
    pass


class NoCharge(Exception):
    pass


class FullTrash(Exception):
    pass


class NoWater(Exception):
    pass


class RobotCleaner:
    max_energy, max_trash, max_water = 100, 500, 500
    min_energy, min_trash, min_water = 0, 0, 0
    work_energy = 5
    work_trash = 50
    work_water = 100

    def __init__(self, charge_battery, trash_can, water_quantity):
        self.charge_battery = float(charge_battery)
        self.trash_can = float(trash_can)
        self.water_quantity = float(water_quantity)

        if self.charge_battery < 0:
            self.charge_battery = 0.0
        elif self.trash_can < 0:
            self.trash_can = 0.0
        elif self.water_quantity < 0:
            self.water_quantity = 0.0

    def wash(self):
        if self.water_quantity > 0:
            self.water_quantity -= self.work_water
            print(f"I wash your floor! I have {self.water_quantity} ml of water in my stomach!")
        else:
            raise NoWater

    def vacuum_cleaner(self):
        if self.charge_battery > 10:
            self.charge_battery -= self.work_energy
            print(f"I move to clean your house! My charge is {self.charge_battery}!")
        elif 1 < self.charge_battery <= 10:
            self.charge_battery -= 1
            raise LowBattery
        else:
            raise NoCharge
        if self.trash_can < 500:
            self.trash_can += self.work_trash
            print(f"I clean your house from trash! I got {self.trash_can} ml trash in my stomach!")
        else:
            raise FullTrash

    def move(self):
        i = 0
        while True:
            print(f"//////{i}/////")
            try:
                self.vacuum_cleaner()
            except FullTrash:
                print("Help! I need to empty my stomach!")
                self.trash_can = 0
            except LowBattery:
                print(f"Hey, I'm running to charge myself! My charge is {self.charge_battery}!")
            except NoCharge:
                print("I can't get charging myself. Bring me to charging place!")
                exit()
            try:
                self.wash()
            except NoWater:
                print("My water is off. I only vacuum clean!")
            i += 1
            time.sleep(1)


# zhorick = RobotCleaner(100, 0, 1000)
#
# zhorick.move()