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