X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;ds=sidebyside;f=bin%2Fplc-kml.py;h=5096a799c16bf50f19187596e0fe466b13a649cf;hb=d954c58ba83857391a19b8f4e109af4869634ae7;hp=f091b82df77b7f730d6d930672a74f5b50faf091;hpb=b466ae5477434db790d03928349f0d0aa18798d4;p=myplc.git
diff --git a/bin/plc-kml.py b/bin/plc-kml.py
index f091b82..5096a79 100755
--- a/bin/plc-kml.py
+++ b/bin/plc-kml.py
@@ -17,6 +17,12 @@ default_foreign_icon = "sites/google-foreign.png"
default_local_builtin = "palette-4.png"
default_foreign_builtin = "palette-3.png"
+# cosmetic - peername cannot be easily changed on the PLC-PLE link...
+def render_public_name (peername):
+ if peername=='PlanetLab': return "PlanetLab Central"
+ elif peername == 'PlanetLabEurope': return "PlanetLab Europe"
+ else: return peername
+
class KmlMap:
def __init__ (self,outputname,options):
@@ -34,32 +40,19 @@ class KmlMap:
def write(self,string):
self.output.write(string.encode("UTF-8"))
- # mention local last
- @staticmethod
- def site_compare (s1,s2):
- p1 = p2 = 0
- if s1['peer_id']: p1=s1['peer_id']
- if s2['peer_id']: p2=s2['peer_id']
- return p2-p1
-
- def refresh (self):
- self.open()
- self.write_header()
- # cache peers
- peers = GetPeers()
- all_sites = GetSites({'enabled':True,'is_public':True})
- all_sites.sort(KmlMap.site_compare)
- for site in all_sites:
- self.write_site(site,peers)
- self.write_footer()
- self.close()
-
# initial placement is for europe - dunno how to tune that yet
def write_header (self):
+ if not self.options.nodegroup:
+ local_peername=render_public_name(api.config.PLC_NAME)
+ title="%s sites"%local_peername
+ detailed="All the sites known to the %s testbed"%local_peername
+ else:
+ title="Nodegroup %s"%self.options.nodegroup
+ detailed="All sites involved in nodegroup %s"%self.options.nodegroup
self.write("""
- PlanetLab Sites
+ %(title)s
9.180821112577378
44.43275321178062
@@ -68,8 +61,8 @@ class KmlMap:
0
-7.767386340832667
- All the sites known to the PlanetLab testbed.
-""")
+ %(detailed)s.
+"""%locals())
def write_footer (self):
self.write("""
@@ -77,13 +70,69 @@ class KmlMap:
def peer_info (self,site, peers):
if not site['peer_id']:
- return (api.config.PLC_NAME, "http://%s/"%api.config.PLC_API_HOST,)
+ return (render_public_name(api.config.PLC_NAME), "http://%s/"%api.config.PLC_API_HOST,)
for peer in peers:
if peer['peer_id'] == site['peer_id']:
- return (peer['peername'],peer['peer_url'].replace("PLCAPI/",""),)
+ return (render_public_name(peer['peername']),peer['peer_url'].replace("PLCAPI/",""),)
return "Unknown peer_name"
- def write_site (self, site, peers):
+ # mention local last
+ @staticmethod
+ def site_compare (s1,s2):
+ p1 = p2 = 0
+ if s1['peer_id']: p1=s1['peer_id']
+ if s2['peer_id']: p2=s2['peer_id']
+ return p2-p1
+
+ ####################
+ def refresh (self):
+ if self.options.nodegroup:
+ self.refresh_nodegroup()
+ else:
+ self.refresh_all_sites()
+
+ def refresh_all_sites(self):
+ self.open()
+ self.write_header()
+ # cache peers
+ peers = GetPeers()
+ all_sites = GetSites({'enabled':True,'is_public':True})
+ all_sites.sort(KmlMap.site_compare)
+ for site in all_sites:
+ self.write_site(site,peers)
+ self.write_footer()
+ self.close()
+
+ def refresh_nodegroup(self):
+ try:
+ nodegroup=GetNodeGroups({'groupname':self.options.nodegroup})[0]
+ except:
+ print "No such nodegroup %s - ignored"%self.options.nodegroup
+ return
+ nodegroup_node_ids=nodegroup['node_ids']
+ if len(nodegroup_node_ids)==0:
+ print "Empty nodegroup %s - ignored"%self.options.nodegroup
+ return
+ # let's go
+ self.open()
+ self.write_header()
+ # cache peers
+ peers = GetPeers()
+ nodes=GetNodes(nodegroup_node_ids)
+ global_node_hash = dict ( [ (n['node_id'],n) for n in nodes ] )
+ site_ids = [ node['site_id'] for node in nodes]
+ # remove any duplicate
+ site_ids = list(set(site_ids))
+ sites = GetSites (site_ids)
+ # patch sites so that 'node_ids' only contains the nodes in the nodegroup
+ for site in sites:
+ site['node_ids'] = [ node_id for node_id in site['node_ids'] if node_id in nodegroup_node_ids ]
+ node_hash = dict ( [ (node_id, global_node_hash[node_id]) for node_id in site['node_ids'] ] )
+ self.write_site(site,peers,nodegroup_id=nodegroup['nodegroup_id'], node_hash=node_hash)
+ self.write_footer()
+ self.close()
+
+ def write_site (self, site, peers, nodegroup_id=False, node_hash={}):
# discard sites with missing lat or lon
if not site['latitude'] or not site['longitude']:
return
@@ -150,9 +199,22 @@ class KmlMap:
description += ""
description += ""
+ # nodegroup direct link
+ if self.options.nodegroup:
+ nodegroup=self.options.nodegroup
+ description += "
"
+ description += "Nodegroup | "
+ description += ""
+ description += " %(nodegroup)s "%locals()
+ description += " | "
+ description += "
"
+
+
# Usage area
+ if not nodegroup_id: title="Usage"
+ else: title="Nodes"
description += ""
- description += "Usage | "
+ description += "%(title)s | "%locals()
# encapsulate usage in a table of its own
description += ""
@@ -160,25 +222,37 @@ class KmlMap:
description += " |
"
# NODES
- description += ""
- description += ""%locals()
- description += " | "
- if nb_nodes:
- description += "%(nb_nodes)d node(s)"%locals()
- #description += " (in Comon)"%locals()
+ # regular all-sites mode
+ if not nodegroup_id:
+ description += " |
"
+ description += ""%locals()
+ description += " | "
+ if nb_nodes:
+ description += "%(nb_nodes)d node(s)"%locals()
+ else:
+ description += "No node"
+ description += " |
"
+ # nodegroup mode : show all nodes
else:
- description += "No node"
- description += ""
+ for node_id in site['node_ids']:
+ node=node_hash[node_id]
+ hostname=node['hostname']
+ description += ""
+ description += ""%locals()
+ description += " | "
+ description += "%(hostname)s "%locals()
+ description += " |
"
#SLICES
- description += ""
- description += ""%locals()
- description += " | "
- if nb_slices:
- description += "%(nb_slices)d slice(s)"%locals()
- else:
- description += "No slice"
- description += " |
"
+ if not nodegroup_id:
+ description += ""
+ description += ""%locals()
+ description += " | "
+ if nb_slices:
+ description += "%(nb_slices)d slice(s)"%locals()
+ else:
+ description += "No slice"
+ description += " |
"
# close usage table
description += ""
@@ -219,6 +293,10 @@ def main () :
parser.add_option("-n","--no-label",action="store_false",dest="labels",
default=True,
help="outputs only geographic positions, no labels")
+
+ parser.add_option("-g","--nodegroup",action='store',dest='nodegroup',default=None,
+ help="outputs a kml file for a given nodegroup only")
+
# default - for private depls. - is to use google-provided icons like palette-3
parser.add_option("-c","--custom",action="store_true",dest="use_custom_icons",
default=False,