mirror of
https://github.com/cwinfo/yggdrasil-map
synced 2024-11-22 12:50:27 +00:00
30 lines
670 B
Python
Executable File
30 lines
670 B
Python
Executable File
#!/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()
|