ffmap-backend/d3mapbuilder.py

37 lines
1,001 B
Python
Raw Normal View History

import json
2013-03-12 20:09:48 +01:00
import datetime
class D3MapBuilder:
def __init__(self, db):
2012-05-12 14:57:25 +02:00
self._db = db
def build(self):
output = dict()
2013-03-12 20:54:14 +01:00
now = datetime.datetime.utcnow().replace(microsecond=0)
2013-03-12 20:52:20 +01:00
2012-06-06 03:34:52 +02:00
nodes = self._db.get_nodes()
output['nodes'] = [{'name': x.name, 'id': x.id,
2012-06-07 00:02:45 +02:00
'macs': ', '.join(x.macs),
2013-01-17 22:13:20 +01:00
'geo': [float(x) for x in x.gps.split(" ")] if x.gps else None,
'firmware': x.firmware,
'flags': x.flags,
'clientcount': x.clientcount
2012-06-12 00:37:07 +02:00
} for x in nodes]
links = self._db.get_links()
output['links'] = [{'source': x.source.id, 'target': x.target.id,
2012-06-06 03:34:52 +02:00
'quality': x.quality,
'type': x.type,
'id': x.id
} for x in links]
2013-03-12 20:09:48 +01:00
output['meta'] = {
2013-03-12 20:52:20 +01:00
'timestamp': now.isoformat()
2013-03-12 20:09:48 +01:00
}
return json.dumps(output)