InteligentnySaper/classes/node.py

class Node:
    def __init__(self, parent, action, state_array):
        self.parent = parent
        self.position = state_array

        if parent:
            parent_pos = parent[1].get_position()
            diff = [state_array[0] - parent_pos[0], state_array[1] - parent_pos[1]]
            if diff[0]==1:
                action="E"
            elif diff[0]==-1:
                action="W"
            elif diff[1]==1:
                action="S"
            elif diff[1]==-1:
                action="N"
            self.action=action
        else:
            self.action=action

    def get_position(self):
        return self.position

    def get_action(self):
        return self.action

    def get_parent(self):
        return self.parent

    def set_parent(self, parent):
        self.parent = parent