Changeset 10377 in genesis for tools

Timestamp:

Apr 7, 2012, 11:55:17 AM (13 years ago)

Author:

rick

Message:

Allow the network kml to be more flexible.

File:

• tools/make-network-kml.py (modified) (4 diffs)

tools/make-network-kml.py

@@ -13 +13 @@
 import yaml
 
-OUTFILE = os.path.join(os.getcwd(),'network.kml')
 
 try:
@@ -51 +50 @@
 </kml>
 '''
-def make_graph():
+def make_graph(debug=False):
   poel = {}
   link_type = {}
   link_data = {}
   hosts = {}
+
   try:
     for host in gformat.get_hostlist():
-      print "## Processing host", host
+      if debug: print "## Processing host", host
       datadump = gformat.get_yaml(host)
       nodename = datadump['nodename']
@@ -94 +94 @@
               uptime = store['uptime'][nodename]
               t_kb = float(b_in + b_out) / 1024
-              print "# INFO: Found %s kB in %s seconds" % (t_kb, INTERVAL)
+              if debug: print "# INFO: Found %s kB in %s seconds" % (t_kb, INTERVAL)
               retval = ((t_kb) / uptime) * INTERVAL
               link_data[addr] = retval
 
-          print "### %s [%s] is of type %s" % (gformat.showaddr(addr), iface_key, link_type[addr])
+          if debug: print "### %s [%s] is of type %s" % (gformat.showaddr(addr), iface_key, link_type[addr])
   except (KeyError, ValueError), e:
-    print "[FOUT] in '%s' interface '%s'" % (host,iface_key)
-    print e
+    if debug: print "[FOUT] in '%s' interface '%s'" % (host,iface_key)
+    if debug: print e
     sys.exit(1)
 
-  f = open(OUTFILE,'w')
-  sys.stderr.write("# Building KML file\n")
-  print >> f, HEADER
+  output = ''
+  if debug: print "# Building KML file"
+  output += HEADER
   for nodename, datadump in hosts.iteritems():
-    print >> f, POINT % datadump
+    output += POINT % datadump
 
   for addr,leden in poel.iteritems():
@@ -129 +129 @@
         for key, value in hosts[lid].iteritems():
            keys[key + 'B'] = value
-        print >> f, LINE % keys
-  print >> f, FOOTER
-  f.flush()
-  sys.stderr.write("# COMPLETED find your output in %s\n" % OUTFILE)
+        output += LINE % keys
+  output += FOOTER
+  return output
 
 if __name__ == "__main__":
-  make_graph()
+  OUTFILE = os.path.join(os.getcwd(),'network.kml')
+  kml_data = make_graph(debug=True)
+  f = open(OUTFILE,'w')
+  f.write(kml_data)
+  f.close()
+  print "# COMPLETED find your output in %s\n" % OUTFILE