ffmap-backend/alfred.py

import subprocess
import json


def _fetch(data_type):
    output = subprocess.check_output(
        ["alfred-json", "-z", "-f", "json", "-r", str(data_type)])
    return json.loads(output.decode("utf-8")).values()


def nodeinfo():
    return _fetch(158)


def statistics():
    return _fetch(159)


def vis():
    return _fetch(160)


def aliases():
    alias = {}
    for node in nodeinfo():
        node_alias = {}

        try:
            # TODO: better pass lat, lng as a tuple?
            node_alias['gps'] = "{lat}\x20{lng}".\
                format(lat=node['location']['latitude'],
                       lng=node['location']['longitude'])
        except KeyError:
            pass

        try:
            node_alias['firmware'] = node['software']['firmware']['release']
        except KeyError:
            pass

        try:
            node_alias['id'] = node['network']['mac']
        except KeyError:
            pass

        if 'hostname' in node:
            node_alias['name'] = node['hostname']
        elif 'name' in node:
            node_alias['name'] = node['name']
        if len(node_alias):
            alias[node['network']['mac']] = node_alias

    return alias