yggdrasil-map/web/updateGraph.py

#!/usr/bin/env python
from flask import Config
from database import NodeDB
import graphPlotter


def generate_graph(time_limit=60*60*3):
	nodes, edges = load_graph_from_db(time_limit)
	print '%d nodes, %d edges' % (len(nodes), len(edges))

	graph = graphPlotter.position_nodes(nodes, edges)
	json = graphPlotter.get_graph_json(graph)

	with open('static/graph.json', 'w') as f:
		f.write(json)


def load_graph_from_db(time_limit):
	config = Config('./')
	config.from_pyfile('web_config.cfg')

	with NodeDB(config) as db:
		nodes = db.get_nodes(time_limit)
		edges = db.get_edges(nodes, 60*60*24*7)
		return (nodes, edges)


if __name__ == '__main__':
	generate_graph()