Machine_learning_2023/domain/entities/vacuum.py

from domain.entities.entity import Entity
from domain.world import World


class Vacuum(Entity):
    def __init__(self, x: int, y: int):
        super().__init__(x, y, "VACUUM")
        self.direction = (1, 0)
        self.battery = 100
        self.cleaning_detergent = 100
        self.container_filling = 0

    def increase_container_filling(self) -> None:
        self.container_filling += 5

    def dump_trash(self) -> None:
        self.container_filling = 0

    def get_container_filling(self):
        return self.container_filling

    # TODO VACUUM: add more properties