ffmap-backend/d3mapbuilder.py

import json

class D3MapBuilder:
  def __init__(self, db):
    self._db = db

  def build(self):
    output = dict()

    nodes = self._db.get_nodes()

    output['nodes'] = [{'group': x.group, 'name': x.name, 'id': x.id,
                        'macs': ', '.join(x.macs)
                       } for x in nodes if x.online]
    output['links'] = [{'source': x.pair[0], 'target': x.pair[1],
                        'distance': x.distance,
                        'strength': x.strength,
                        'quality': x.quality,
                        'id': "-".join(nodes[i].id for i in x.pair)
                       } for x in self._db.get_links()]

    return json.dumps(output)