simple_planner.py

import json
import networkx as nx
from itertools import izip
from celery import Celery

celery = Celery('tut1',broker='redis://',backend='redis://')

graph = nx.MultiGraph()

data_str = open('routes.json').read()
data = json.loads(data_str)

for item in data:
	route_name = item['name']
	stops = item['stops']
	stop_names = [stop['stop']['name'] for stop in stops]
	if stop_names!=[]:
		graph.add_star(stop_names,route=route_name)


@celery.task
def get_routes(start,end):
    paths =  nx.all_shortest_paths(graph,start,end)
    return [path for path in paths]


@celery.task
def get_full_routes(start,end):
    paths = nx.all_shortest_paths(graph,start,end)
    routes=[]
    for path in paths:
        for k,v in zip(path,path[1:]):
            edges = graph[k][v]
            route=[]
            for edge in edges.values():
                route.append((k,v,edge['route']))
            routes.append(route)
    return routes
            
#print nx.shortest_path(graph,'Dilsukhnagar Bus station','Patancheru Bus Stop')
#print nx.shortest_path(graph,'Patancheru Bus Stop','Hayath Nagar Bus Stop')
#for path in nx.all_shortest_paths(graph,'Dilsukhnagar Bus station','Patancheru Bus Stop'):
#	print path
	
#all_paths =  nx.all_pairs_shortest_path(graph)
#path =  all_paths['Nagole Bus Stop']['Ziaguda']
#for k,v in zip(path,path[1:]):
#	edges = graph[k][v]
#	for edge in edges.values():
#		print k,v,edge['route']