SI_projekt_smieciarka/trash.py

import random
from house import is_house


class Trash:
    def __init__(self, grid_w, grid_h, grid_size):
        self.grid_w = grid_w
        self.grid_h = grid_h
        self.size = grid_size

    def new_pos(self, truck_pos, houses, multi):
        while True:
            self.pos = [random.randrange(0, self.grid_w, self.size),
                        random.randrange(0, self.grid_h, self.size)]
            if self.pos != truck_pos and not is_house(self.pos, houses) and self not in multi:
                break

    def __eq__(self, other):
        if isinstance(self, Trash):
            return self.pos == other.pos
        return False