Mercurial > hg > NetworkVis
changeset 6:aeef1fedd899
first version of ismi-python-neo4jrestclient. doesn't work yet.
| author | casties |
|---|---|
| date | Mon, 28 Sep 2015 18:15:35 +0200 |
| parents | fa1b4fa5b4f8 |
| children | 45dad9e38c82 |
| files | ismi-python-neo4jrestclient/ismi-vis.py ismi-python-neo4jrestclient/requirements.txt ismi-python-neo4jrestclient/static/index.html |
| diffstat | 3 files changed, 244 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/ismi-python-neo4jrestclient/ismi-vis.py Mon Sep 28 18:15:35 2015 +0200 @@ -0,0 +1,81 @@ +#!/usr/bin/env python +from json import dumps + +from flask import Flask, Response, request + +from neo4jrestclient.client import GraphDatabase, Node + +app = Flask(__name__, static_url_path='/static/') +gdb = GraphDatabase("http://localhost:7474", username="neo4j", password="neo5j") + + +@app.route("/") +def get_index(): + return app.send_static_file('index.html') + + +@app.route("/graph") +def get_graph(): + query = ("MATCH (m:Movie)<-[:ACTED_IN]-(a:Person) " + "RETURN m.title as movie, collect(a.name) as cast " + "LIMIT {limit}") + results = gdb.query(query, + params={"limit": request.args.get("limit", 100)}) + nodes = [] + rels = [] + i = 0 + for movie, cast in results: + nodes.append({"title": movie, "label": "movie"}) + target = i + i += 1 + for name in cast: + actor = {"title": name, "label": "actor"} + try: + source = nodes.index(actor) + except ValueError: + nodes.append(actor) + source = i + i += 1 + rels.append({"source": source, "target": target}) + return Response(dumps({"nodes": nodes, "links": rels}), + mimetype="application/json") + + +@app.route("/search") +def get_search(): + try: + q = request.args["q"] + except KeyError: + return [] + else: + query = ("MATCH (movie:Movie) " + "WHERE movie.title =~ {title} " + "RETURN movie") + results = gdb.query( + query, + returns=Node, + params={"title": "(?i).*" + q + ".*"} + ) + return Response(dumps([{"movie": row.properties} + for [row] in results]), + mimetype="application/json") + + +@app.route("/movie/<title>") +def get_movie(title): + query = ("MATCH (movie:Movie {title:{title}}) " + "OPTIONAL MATCH (movie)<-[r]-(person:Person) " + "RETURN movie.title as title," + "collect([person.name, " + " head(split(lower(type(r)), '_')), r.roles]) as cast " + "LIMIT 1") + results = gdb.query(query, params={"title": title}) + title, cast = results[0] + return Response(dumps({"title": title, + "cast": [dict(zip(("name", "job", "role"), member)) + for member in cast]}), + mimetype="application/json") + + +if __name__ == '__main__': + app.run(port=8080)
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/ismi-python-neo4jrestclient/requirements.txt Mon Sep 28 18:15:35 2015 +0200 @@ -0,0 +1,2 @@ +Flask==0.10.1 +neo4jrestclient==2.0.4
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/ismi-python-neo4jrestclient/static/index.html Mon Sep 28 18:15:35 2015 +0200 @@ -0,0 +1,161 @@ +<html> +<head> + <meta name="viewport" content="width=device-width, initial-scale=1.0"> + <link rel="stylesheet" href="http://neo4j-contrib.github.io/developer-resources/language-guides/assets/css/main.css"> + <title>Neo4j Movies</title> +</head> + +<body> +<div id="graph"> +</div> +<div role="navigation" class="navbar navbar-default navbar-static-top"> + <div class="container"> + <div class="row"> + <div class="col-sm-6 col-md-6"> + <ul class="nav navbar-nav"> + <li> + <form role="search" class="navbar-form" id="search"> + <div class="form-group"> + <input type="text" value="Matrix" placeholder="Search for Movie Title" class="form-control" name="search"> + </div> + <button class="btn btn-default" type="submit">Search</button> + </form> + </li> + </ul> + </div> + <div class="navbar-header col-sm-6 col-md-6"> + <div class="logo-well"> + <a href="http://neo4j.com/developer-resources"> + <img src="http://neo4j-contrib.github.io/developer-resources/language-guides/assets/img/logo-white.svg" alt="Neo4j World's Leading Graph Database" id="logo"> + </a> + </div> + <div class="navbar-brand"> + <div class="brand">ISMI Texts</div> + </div> + </div> + </div> + </div> +</div> + +<div class="row"> + <div class="col-md-5"> + <div class="panel panel-default"> + <div class="panel-heading">Search Results</div> + <table id="results" class="table table-striped table-hover"> + <thead> + <tr> + <th>Movie</th> + <th>Released</th> + <th>Tagline</th> + </tr> + </thead> + <tbody> + </tbody> + </table> + </div> + </div> + <div class="col-md-7"> + <div class="panel panel-default"> + <div class="panel-heading" id="title">Details</div> + <div class="row"> + <div class="col-sm-4 col-md-4"> + <img src="" class="well" id="poster"/> + </div> + <div class="col-md-8 col-sm-8"> + <h4>Crew</h4> + <ul id="crew"> + </ul> + </div> + </div> + </div> + </div> +</div> +<style type="text/css"> + .node { stroke: #222; stroke-width: 1.5px; } + .node.actor { fill: #888; } + .node.movie { fill: #BBB; } + .link { stroke: #999; stroke-opacity: .6; stroke-width: 1px; } +</style> + +<script type="text/javascript" src="//code.jquery.com/jquery-1.11.0.min.js"></script> +<script src="http://d3js.org/d3.v3.min.js" type="text/javascript"></script> +<script type="text/javascript"> + $(function () { + function showMovie(title) { + $.get("/movie/" + encodeURIComponent(title), + function (data) { + if (!data) return; + $("#title").text(data.title); + $("#poster").attr("src","http://neo4j-contrib.github.io/developer-resources/language-guides/assets/posters/"+encodeURIComponent(data.title)+".jpg"); + var $list = $("#crew").empty(); + data.cast.forEach(function (cast) { + $list.append($("<li>" + cast.name + " " +cast.job + (cast.job == "acted"?" as " + cast.role : "") + "</li>")); + }); + }, "json"); + return false; + } + function search() { + var query=$("#search").find("input[name=search]").val(); + $.get("/search?q=" + encodeURIComponent(query), + function (data) { + var t = $("table#results tbody").empty(); + if (!data || data.length == 0) return; + data.forEach(function (row) { + var movie = row.movie; + $("<tr><td class='movie'>" + movie.title + "</td><td>" + movie.released + "</td><td>" + movie.tagline + "</td></tr>").appendTo(t) + .click(function() { showMovie($(this).find("td.movie").text());}) + }); + showMovie(data[0].movie.title); + }, "json"); + return false; + } + + $("#search").submit(search); + search(); + }) +</script> + +<script type="text/javascript"> + var width = 800, height = 800; + + var force = d3.layout.force() + .charge(-200).linkDistance(30).size([width, height]); + + var svg = d3.select("#graph").append("svg") + .attr("width", "100%").attr("height", "100%") + .attr("pointer-events", "all"); + + d3.json("/graph", function(error, graph) { + if (error) return; + + force.nodes(graph.nodes).links(graph.links).start(); + + var link = svg.selectAll(".link") + .data(graph.links).enter() + .append("line").attr("class", "link"); + + var node = svg.selectAll(".node") + .data(graph.nodes).enter() + .append("circle") + .attr("class", function (d) { return "node "+d.label }) + .attr("r", 10) + .call(force.drag); + + // html title attribute + node.append("title") + .text(function (d) { return d.title; }) + + // force feed algo ticks + force.on("tick", function() { + link.attr("x1", function(d) { return d.source.x; }) + .attr("y1", function(d) { return d.source.y; }) + .attr("x2", function(d) { return d.target.x; }) + .attr("y2", function(d) { return d.target.y; }); + + node.attr("cx", function(d) { return d.x; }) + .attr("cy", function(d) { return d.y; }); + }); + }); +</script> +</body> +</html>