Merge pull request #1 from NeoRaider/master

Python3

aliases.json

@@ -2,7 +2,7 @@
    "04:11:6b:98:08:21" : {
       "name" : "sellars"
    },
-   "da:7b:6f:c1:63:d2" : {
+   "00:25:86:e6:f1:bf" : {
       "name" : "krtek"
    },
    "b0:48:7a:e7:d3:64" : {
@@ -12,8 +12,7 @@
       "name" : "holstentor"
    },
    "56:47:05:ab:00:2b" : {
-      "name" : "aftermath",
-      "group" : 1
+      "name" : "aftermath"
    },
    "ca:96:05:3c:54:f9" : {
       "name" : "prometheus"
@@ -35,5 +34,8 @@
    },
    "ca:a8:ca:2d:c8:b2" : {
       "name" : "prometheus"
+   },
+   "fa:d1:11:80:41:d4" : {
+      "name" : "confusion"
    }
 }

bat2nodes.py

@@ -4,7 +4,7 @@
 # Gatewayliste
 # aliases.json
 
-import simplejson as json
+import json
 import fileinput
 import argparse
 
@@ -128,7 +128,8 @@ if options['aliases']:
     if 'group' in alias:
       node.group = alias['group']
 
-for gateway in options['gateway']:
+if options['gateway']:
+  for gateway in options['gateway']:
     try:
       node = maybe_node_by_mac(nodes, (gateway, ))
       node.group = 2
@@ -149,16 +150,16 @@ def map_link(nodes, pair):
   return link
 
 
-links = map(lambda x: map_link(nodes, x), links)
+links = [map_link(nodes, x) for x in links]
 
 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)
-                                }, nodes)
-output['links'] = map(lambda x: {'source': x.pair[0], 'target': x.pair[1],
+                   } for x in nodes]
+output['links'] = [{'source': x.pair[0], 'target': x.pair[1],
                     'distance': x.distance,
                     'strength': x.strength
-                                }, links)
+                   } for x in links]
 
-print json.dumps(output)
+print(json.dumps(output))