SZI2019SmieciarzWmi/DataModels/GC.py

from DataModels.Cell import Cell
from DataModels.Road import Road


class GC(Cell):
    def __init__(self, x, y, max_rubbish, yellow=0, green=0, blue=0):
        Cell.__init__(self, x, y, max_rubbish, yellow, green, blue)

    def move(self, direction, environment):
        x, y = [self.x, self.y]
        movement = {
            "right": (x + 1, y) if environment[x + 1][y] == Road else (x, y),
            "left": (x - 1, y) if environment[x - 1][y] == Road else (x, y),
            "up": (x, y + 1) if environment[x][y + 1] == Road else (x, y),
            "down": (x, y - 1) if environment[x][y - 1] == Road else (x, y)
        }
        self.x, self.y = movement[direction]
        print(self.x, self.y)