diff --git a/sprites/garbage_collector.py b/sprites/garbage_collector.py index 579202d..e1e11af 100644 --- a/sprites/garbage_collector.py +++ b/sprites/garbage_collector.py @@ -1,6 +1,6 @@ import pygame from sprites.cell import Cell, CELL_SIZE -from +from sprites.house import House from config import PLAY_HEIGHT, PLAY_WIDTH class Garbage_collector(Cell): @@ -29,9 +29,7 @@ class Garbage_collector(Cell): elif(destination is 'y'): self.y = value self.update() -PLASTIC = 0 # blue -GLASS = 1 # green -METAL = 2 # yellow + def collect_trash(self, house): rubbish = house.get_rubbish_data() to_collect = rubbish