49 lines
1.1 KiB
Python
Executable File
49 lines
1.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import fileinput
|
|
import argparse
|
|
|
|
from nodedb import NodeDB
|
|
from d3mapbuilder import D3MapBuilder
|
|
|
|
# Force encoding to UTF-8
|
|
import locale # Ensures that subsequent open()s
|
|
locale.getpreferredencoding = lambda: 'UTF-8' # are UTF-8 encoded.
|
|
|
|
import sys
|
|
sys.stdin = open('/dev/stdin', 'r')
|
|
sys.stdout = open('/dev/stdout', 'w')
|
|
sys.stderr = open('/dev/stderr', 'w')
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('-a', '--aliases',
|
|
help='read aliases from FILE',
|
|
action='append',
|
|
metavar='FILE')
|
|
|
|
parser.add_argument('-g', '--gateway', action='append',
|
|
help='MAC of a gateway')
|
|
|
|
parser.add_argument('batmanjson', help='output of batman vd json')
|
|
|
|
args = parser.parse_args()
|
|
|
|
options = vars(args)
|
|
|
|
db = NodeDB()
|
|
print(options)
|
|
db.import_batman(list(fileinput.input(options['batmanjson'])))
|
|
|
|
if options['aliases']:
|
|
for aliases in options['aliases']:
|
|
db.import_aliases(json.load(open(aliases)))
|
|
|
|
if options['gateway']:
|
|
db.mark_gateways(options['gateway'])
|
|
|
|
m = D3MapBuilder(db)
|
|
|
|
print(m.build())
|