pathfinder-model.py

import networkx as nx
from yhat import YhatModel, Yhat


graphdict = {
    "Houston": ["Austin", "Dallas", "Oklahoma City"],
    "Austin": ["New Orleans", "Dallas"],
    "New Orleans": ["Austin"],
    "Dallas": ["New York"],
    "Oklahoma City": ["Austin", "Dallas"],
    "New York": ["London", "Berlin", "Chicago", "San Francisco"],
    "San Francisco": ["Tokyo"]
}

graph = nx.Graph(graphdict)

def find_shortest_path(source, target):
    return nx.shortest_path(graph, source=source, target=target)

class Pathfinder(YhatModel):
    REQUIREMENTS = [
        "networkx"
    ]
    def execute(self, data):
        source = data['source']
        target = data['target']

        if source not in graph.nodes():
            return { "shortestPath": None, "error": "{} is not in our network".format(source)}
        if target not in graph.nodes():
            return { "shortestPath": None, "error": "{} is not in our network".format(target)}

        path = find_shortest_path(source, target)

        return { "shortestPath": path }


pf = Pathfinder()
print pf.execute({ "source": "Houston", "target": "Oklahoma City" })
print pf.execute({ "source": "Houston", "target": "Chicago" })
print pf.execute({ "source": "Houston", "target": "Tokyo" })

yh = Yhat(USERNAME, APIKEY, URL)
yh.deploy("Pathfinder", Pathfinder, globals())