ffmap-backend/bat2nodes.py

83 lines
2.2 KiB
Python
Raw Normal View History

2012-06-05 17:40:21 +02:00
#!/usr/bin/env python3
2012-02-17 02:09:21 +01:00
2012-04-28 23:06:34 +02:00
import json
2012-02-17 02:09:21 +01:00
import fileinput
import argparse
import os
import time
2012-02-17 02:09:21 +01:00
from batman import batman
from alfred import alfred
2014-08-17 19:32:13 +02:00
from rrddb import rrd
from nodedb import NodeDB
from d3mapbuilder import D3MapBuilder
2012-05-11 14:12:41 +02:00
2012-06-15 18:27:51 +02:00
# Force encoding to UTF-8
import locale # Ensures that subsequent open()s
locale.getpreferredencoding = lambda _=None: 'UTF-8' # are UTF-8 encoded.
2012-06-15 18:27:51 +02:00
import sys
2013-03-11 09:24:57 +01:00
#sys.stdin = open('/dev/stdin', 'r')
#sys.stdout = open('/dev/stdout', 'w')
#sys.stderr = open('/dev/stderr', 'w')
2012-06-15 18:27:51 +02:00
2012-02-17 02:09:21 +01:00
parser = argparse.ArgumentParser()
parser.add_argument('-a', '--aliases',
help='read aliases from FILE',
action='append',
2012-02-17 02:09:21 +01:00
metavar='FILE')
parser.add_argument('-m', '--mesh', action='append',
2014-09-21 22:18:31 +02:00
default=["bat0"],
help='batman mesh interface')
2012-02-17 02:09:21 +01:00
parser.add_argument('-A', '--alfred', action='store_true',
help='retrieve aliases from alfred')
parser.add_argument('-d', '--destination-directory', action='store',
help='destination directory for generated files',required=True)
2012-02-17 02:09:21 +01:00
args = parser.parse_args()
options = vars(args)
db = NodeDB(int(time.time()))
2014-09-21 22:18:31 +02:00
for mesh_interface in options['mesh']:
bm = batman(mesh_interface)
db.parse_vis_data(bm.vis_data(options['alfred']))
for gw in bm.gateway_list():
2014-09-21 22:18:31 +02:00
db.mark_gateway(gw)
2012-02-17 02:09:21 +01:00
if options['aliases']:
for aliases in options['aliases']:
db.import_aliases(json.load(open(aliases)))
2012-02-17 02:09:21 +01:00
if options['alfred']:
af = alfred()
db.import_aliases(af.aliases())
db.load_state("state.json")
# remove nodes that have been offline for more than 30 days
db.prune_offline(time.time() - 30*86400)
db.dump_state("state.json")
scriptdir = os.path.dirname(os.path.realpath(__file__))
m = D3MapBuilder(db)
2012-02-17 02:09:21 +01:00
#Write nodes json
nodes_json = open(options['destination_directory'] + '/nodes.json.new','w')
nodes_json.write(m.build())
nodes_json.close()
#Move to destination
os.rename(options['destination_directory'] + '/nodes.json.new',options['destination_directory'] + '/nodes.json')
rrd = rrd(scriptdir + "/nodedb/", options['destination_directory'] + "/nodes")
rrd.update_database(db)
rrd.update_images()