Merge pull request #11 from toppoint/obscure-clients

Fixes and nodes.json consistency
This commit is contained in:
Nils Schneider 2014-01-04 00:32:57 -08:00
commit 64918cb092
2 changed files with 8 additions and 7 deletions

View file

@ -67,6 +67,10 @@ rrd.update_images()
m = D3MapBuilder(db) m = D3MapBuilder(db)
nodes_json = open(options['destination_directory'] + '/nodes.json','w') #Write nodes json
nodes_json = open(options['destination_directory'] + '/nodes.json.new','w')
nodes_json.write(m.build()) nodes_json.write(m.build())
nodes_json.close() nodes_json.close()
#Move to destination
os.rename(options['destination_directory'] + '/nodes.json.new',options['destination_directory'] + '/nodes.json')

View file

@ -34,14 +34,14 @@ class NodeDB:
if mac.lower() in node.macs: if mac.lower() in node.macs:
return node return node
raise raise KeyError
def maybe_node_by_id(self, mac): def maybe_node_by_id(self, mac):
for node in self._nodes: for node in self._nodes:
if mac.lower() == node.id: if mac.lower() == node.id:
return node return node
raise raise KeyError
def parse_vis_data(self,vis_data): def parse_vis_data(self,vis_data):
for x in vis_data: for x in vis_data:
@ -221,7 +221,6 @@ class NodeDB:
if node.flags['client']: if node.flags['client']:
node.macs = set() node.macs = set()
clientIds[node.id] = None clientIds[node.id] = None
sys.stderr.write("client:" + node.id)
for link in self._links: for link in self._links:
ids = link.source.interface ids = link.source.interface
@ -241,7 +240,6 @@ class NodeDB:
globalIdCounter += 1 globalIdCounter += 1
elif ids in clientIds: elif ids in clientIds:
sys.stderr.write("passed ids")
newId = generateId(idt) newId = generateId(idt)
clientIds[ids] = newId clientIds[ids] = newId
ids = newId ids = newId
@ -250,7 +248,6 @@ class NodeDB:
node_source.id = ids; node_source.id = ids;
elif idt in clientIds: elif idt in clientIds:
sys.stderr.write("passed idt")
newId = generateId(ids,nodeCounters) newId = generateId(ids,nodeCounters)
clientIds[idt] = newId clientIds[idt] = newId
idt = newId idt = newId
@ -261,7 +258,7 @@ class NodeDB:
link.id = ids + "-" + idt link.id = ids + "-" + idt
except: except:
raise pass
# extends node id by incremented node counter # extends node id by incremented node counter
def generateId(nodeId,nodeCounters): def generateId(nodeId,nodeCounters):