X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=bin%2Fplc-kml.py;h=5096a799c16bf50f19187596e0fe466b13a649cf;hb=refs%2Fheads%2Fpython2;hp=6cd460db6e8bffd2845fca85aeb8050a39fd1d5e;hpb=a455b90d009a6a43cf2c66e99aba9f639223c548;p=myplc.git
diff --git a/bin/plc-kml.py b/bin/plc-kml.py
index 6cd460d..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
@@ -101,67 +150,6 @@ class KmlMap:
baseurl='http://%s'%api.config.PLC_WWW_HOST
peer_id=site['peer_id']
- # open description
- description = "
"
-
- # TESTBED
- description += " | | |
"
- description += ""
- description += "Testbed"
- description += " | "
- (peername,peerurl) = self.peer_info (site,peers)
- description += " %(peername)s "%locals()
- #description += "[description]"%locals()
- description += " | "
- if peer_id:
- description += ""%locals()
- else:
- description += ""%locals()
- description += " |
"
-
- # URL
- if site['url']:
- site_url=site['url']
- description += ""
- description += "Website"
- description += " | "
- description += " %(site_url)s "%locals()
- description += " | |
"
-
- description += " | | |
"
- description += "Usage details: | | |
"
- description += " | | |
"
-
- # NODES
- description += ""
- description += ""%locals()
- description += " | "
- if nb_nodes:
- description += "%(nb_nodes)d node(s)"%locals()
- #description += " (in Comon)"%locals()
- else:
- description += "No node"
- description += " | |
"
-
- #SLICES
- description += ""
- description += ""%locals()
- description += " | "
- if nb_slices:
- description += "%(nb_slices)d slice(s)"%locals()
- else:
- description += "No slice"
- description += " | |
"
-
- description += " | | |
"
-
- # close description
- description += "
"
-
- if not self.options.labels:
- name=""
- description=""
-
# STYLE
# the size for google icons
if not self.options.use_custom_icons:
@@ -185,6 +173,98 @@ class KmlMap:
iconspec="%(iconurl)s%(xyspec)s"%locals()
+ # open description
+ # can't seem to get classes to get through to the google maps API
+ # so have to use hard-wired settings
+ description = ""
+ description += ""
+
+ if not self.options.labels:
+ name=""
+ description=""
+
# set the camera 50km high
template="""
@@ -213,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,