from __future__ import absolute_import
from __future__ import print_function
from __future__ import division

import numpy as np

from ds_min_priority_queue import MinPriorityQueue


def dijkstra(weighted_graph_d, start_vertex):
    """Dijkstra algorithm for "weighted" graph.

    Finds shortest path in a weighted graph from a particular node 
    to all vertices that are reachable from it.
    """
    min_pq = MinPriorityQueue()

    distance_d = {v: np.inf for v in weighted_graph_d.keys()}
    visited_d = {v: False for v in weighted_graph_d.keys()}
    previous_d = {v: None for v in weighted_graph_d.keys()}
    
    min_pq.insert(start_vertex)
    distance_d[start_vertex] = 0
    visited_d[start_vertex] = True

    while min_pq.size > 0:
        v = min_pq.extract_min()
        visited_d[v] = True

        for v_neighbor in weighted_graph_d[v].keys():
            if (not visited_d[v_neighbor] and 
                distance_d[v_neighbor] > 
                    distance_d[v] + weighted_graph_d[v][v_neighbor]):
                pass


def main():
    weighted_graph_d = {
        's': {'a': 2, 'b': 6},
        'a': {'b': 3, 'c': 1},
        'b': {'a': 5, 'd': 2},
        'c': {'b': 1, 'e': 4, 'f': 2},
        'd': {'c': 3, 'f': 2},
        'e': {},
        'f': {'e': 1}
    }
    start_vertex = 's'
    print('weighted_graph_d: {}'.format(weighted_graph_d))
    print('Dijkstra shortest path from {}:'.format(start_vertex))
    

if __name__ == '__main__':
    main()

from __future__ import absolute_import
from __future__ import print_function
from __future__ import division

import numpy as np

from ds_min_priority_queue import MinPriorityQueue


def dijkstra(w_graph_d, start_vertex):
    """Dijkstra algorithm for "weighted" graph.

    Finds shortest path in a weighted graph from a particular node 
    to all vertices that are reachable from it.
    """
    min_pq = MinPriorityQueue()

    distance_d = {v: np.inf for v in w_graph_d.keys()}
    visited_d = {v: False for v in w_graph_d.keys()}
    previous_d = {v: None for v in w_graph_d.keys()}
    
    min_pq.insert(start_vertex)
    distance_d[start_vertex] = 0
    visited_d[start_vertex] = True

    while min_pq.size > 0:
        v = min_pq.extract_min()
        visited_d[v] = True

        for v_neighbor in w_graph_d[v].keys():
            if (not visited_d[v_neighbor] and 
                distance_d[v_neighbor] > 
                    distance_d[v] + w_graph_d[v][v_neighbor]):
                distance_d[v_neighbor] = distance_d[v] + w_graph_d[v][v_neighbor]
                previous_d[v_neighbor] = v
                min_pq.insert(v_neighbor)

    return distance_d, previous_d, visited_d


def main():
    w_graph_d = {
        's': {'a': 2, 'b': 6},
        'a': {'b': 3, 'c': 1},
        'b': {'a': 5, 'd': 2},
        'c': {'b': 1, 'e': 4, 'f': 2},
        'd': {'c': 3, 'f': 2},
        'e': {},
        'f': {'e': 1}
    }
    start_vertex = 's'
    print('w_graph_d: {}'.format(w_graph_d))
    print('Dijkstra shortest path from {}:'.format(start_vertex))
    
    distance_d, previous_d, visited_d = (
        dijkstra(w_graph_d, start_vertex))

    print('distance_d: {}'.format(distance_d))
    print('previous_d: {}'.format(distance_d))
    print('visited_d: {}'.format(visited_d))
   

if __name__ == '__main__':
    main()

Complete Dijkstra alg, TODO: revise min priority queue to tuple