Make script Python3 compatible
parent
aabaabc67a
commit
cd7b9cf235
33
bat2nodes.py
33
bat2nodes.py
|
@ -4,7 +4,7 @@
|
||||||
# Gatewayliste
|
# Gatewayliste
|
||||||
# aliases.json
|
# aliases.json
|
||||||
|
|
||||||
import simplejson as json
|
import json
|
||||||
import fileinput
|
import fileinput
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
@ -128,12 +128,13 @@ if options['aliases']:
|
||||||
if 'group' in alias:
|
if 'group' in alias:
|
||||||
node.group = alias['group']
|
node.group = alias['group']
|
||||||
|
|
||||||
for gateway in options['gateway']:
|
if options['gateway']:
|
||||||
try:
|
for gateway in options['gateway']:
|
||||||
node = maybe_node_by_mac(nodes, (gateway, ))
|
try:
|
||||||
node.group = 2
|
node = maybe_node_by_mac(nodes, (gateway, ))
|
||||||
except:
|
node.group = 2
|
||||||
continue
|
except:
|
||||||
|
continue
|
||||||
|
|
||||||
def map_link(nodes, pair):
|
def map_link(nodes, pair):
|
||||||
distance = 80
|
distance = 80
|
||||||
|
@ -149,16 +150,16 @@ def map_link(nodes, pair):
|
||||||
return link
|
return link
|
||||||
|
|
||||||
|
|
||||||
links = map(lambda x: map_link(nodes, x), links)
|
links = [map_link(nodes, x) for x in links]
|
||||||
|
|
||||||
output = dict()
|
output = dict()
|
||||||
|
|
||||||
output['nodes'] = map(lambda x: {'group': x.group, 'name': x.name,
|
output['nodes'] = [{'group': x.group, 'name': x.name,
|
||||||
'macs': ', '.join(x.macs)
|
'macs': ', '.join(x.macs)
|
||||||
}, nodes)
|
} for x in nodes]
|
||||||
output['links'] = map(lambda x: {'source': x.pair[0], 'target': x.pair[1],
|
output['links'] = [{'source': x.pair[0], 'target': x.pair[1],
|
||||||
'distance': x.distance,
|
'distance': x.distance,
|
||||||
'strength': x.strength
|
'strength': x.strength
|
||||||
}, links)
|
} for x in links]
|
||||||
|
|
||||||
print json.dumps(output)
|
print(json.dumps(output))
|
||||||
|
|
Loading…
Reference in New Issue