cell.py

import svgwrite
from distances import Distances

class Cell:
    def __init__(self, row, column):
        self.row, self.column = row, column
        self.links = {}
        self.distance = 0

    def link(self, cell, bidi=True):
        self.links[cell] = True
        if bidi:
            cell.link(self, False)

    def unlink(self, cell, bidi = True):
        if cell in self.links:
            del self.links[cell]
        if bidi: cell.unlink(self, False)

    def show_links(self):
        print (self.links)

    def get_links(self):
        return self.links.keys()

    def get_links_full(self):
        return self.links

    def linked(self, cell):
        if cell in self.links:
            return True
        return False

    def neighbors(self):
        list = []
        if self.north: list.append(self.north)
        if self.south: list.append(self.south)
        if self.west: list.append(self.west)
        if self.east: list.append(self.east)
        return list

    def distances(self):
        distances = Distances(self)
        frontier = [self]

        while len(frontier) > 0:
            new_frontier = []

            for cell in frontier:
                for link in cell.get_links():
                    if not link in distances.cells:
                        distances.cells[link] = distances.cells[cell] + 1
                        new_frontier.append(link)

            frontier = new_frontier
        return distances

    def set_distance(self, distance):
        self.distance = distance