mirror of
https://github.com/cwinfo/yggdrasil-map
synced 2024-09-19 16:09:34 +00:00
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
import json
|
|
from database import NodeDB
|
|
from graph import Node, Edge
|
|
import traceback
|
|
import time
|
|
|
|
def insert_graph_data(config, data, mail, ip, version):
|
|
try:
|
|
graph_data = json.loads(data)
|
|
except ValueError:
|
|
return 'Invalid JSON'
|
|
|
|
log = '[%s] ip: %s, version: %d, mail: %r, nodes: %d, edges: %d' % (
|
|
time.strftime('%Y-%m-%d %H:%M:%S'), ip,
|
|
version, mail, len(graph_data['nodes']), len(graph_data['edges']))
|
|
|
|
with open(config['LOG'], 'a') as f:
|
|
f.write(log + '\n')
|
|
|
|
if mail == 'your@email.here':
|
|
return 'Please change email address in config.'
|
|
|
|
if version != 2:
|
|
return 'You are using outdated version of sendGraph script. Get new version from https://github.com/zielmicha/fc00.org/blob/master/scripts/sendGraph.py'
|
|
|
|
nodes = dict()
|
|
edges = []
|
|
|
|
try:
|
|
for n in graph_data['nodes']:
|
|
try:
|
|
node = Node(n['ip'], version=n['version'])
|
|
nodes[n['ip']] = node
|
|
except Exception:
|
|
pass
|
|
|
|
for e in graph_data['edges']:
|
|
try:
|
|
edge = Edge(nodes[e['a']], nodes[e['b']])
|
|
edges.append(edge)
|
|
except Exception:
|
|
pass
|
|
except Exception:
|
|
return 'Invalid JSON nodes'
|
|
|
|
print "Accepted %d nodes and %d links." % (len(nodes), len(edges))
|
|
|
|
if len(nodes) == 0 or len(edges) == 0:
|
|
return 'No valid nodes or edges'
|
|
|
|
uploaded_by = ip
|
|
|
|
try:
|
|
with NodeDB(config) as db:
|
|
db.insert_graph(nodes, edges, uploaded_by)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
return 'Database failure'
|
|
|
|
return None
|