getAllNeighbors.py

def getAllNeighbors(origin, steps = 1):
    seen = {origin}
    worklist = [origin]
    while steps:
        new_worklist = []
        for next in worklist:
            seen.add(next)
            for neighbor in next.get_all_neighbors(15):
                if neighbor not in seen: new_worklist.append(neighbor)
        worklist = new_worklist
        steps -= 1
    return seen