#
# Faiyaz Ahmed <faiyaza@cs.princeton.edu
#
-# $Id: plc.py,v 1.1 2006/11/14 19:27:09 faiyaza Exp $
+# $Id: plc.py,v 1.16 2007/07/03 19:59:02 soltesz Exp $
#
from emailTxt import *
import logging
import auth
import time
-import config
+from config import config,XMLRPC_SERVER
logger = logging.getLogger("monitor")
-XMLRPC_SERVER = 'https://www.planet-lab.org/PLCAPI/'
+#XMLRPC_SERVER = config.XMLRPC_SERVER
+
+config = config()
'''
Returns list of nodes in dbg as reported by PLC
if nodeinfo['pcu_ids']:
sitepcu = api.GetPCUs(auth.auth, nodeinfo['pcu_ids'])[0]
sitepcu[nodename] = nodeinfo["ports"][0]
- return False
+ return sitepcu
else:
logger.info("%s doesn't have PCU" % nodename)
- return sitepcu
-
+ return False
'''
Returns all site nodes for site id (loginbase).
logger.info("getSiteNodes: %s" % exc)
return nodelist
+def getSites(filter=None):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ sites = []
+ anon = {'AuthMethod': "anonymous"}
+ try:
+ sites = api.GetSites(anon, filter, None)
+ except Exception, exc:
+ print "getSiteNodes2: %s" % exc
+ logger.info("getSiteNodes2: %s" % exc)
+ return sites
+
+def getSiteNodes2(loginbase):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
+ nodelist = []
+ anon = {'AuthMethod': "anonymous"}
+ try:
+ nodeids = api.GetSites(anon, {"login_base": loginbase})[0]['node_ids']
+ nodelist += getNodes({'node_id':nodeids})
+ except Exception, exc:
+ logger.info("getSiteNodes2: %s" % exc)
+ return nodelist
+
+def getNodeNetworks(filter=None):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ nodenetworks = api.GetNodeNetworks(auth.auth, filter, None)
+ return nodenetworks
+
+def getNodes(filter=None):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ nodes = api.GetNodes(auth.auth, filter, None) #['boot_state', 'hostname',
+ #'site_id', 'date_created', 'node_id', 'version', 'nodenetwork_ids',
+ #'last_updated', 'peer_node_id', 'ssh_rsa_key' ])
+ return nodes
'''
Sets boot state of a node.
except Exception, exc:
logger.info("suspendSlices: %s" % exc)
+def enableSlices(nodename):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ for slice in slices(siteId(nodename)):
+ logger.info("Enabling slices %s" % slice)
+ try:
+ if not config.debug:
+ slice_list = api.GetSlices(auth.auth, {'name': slice}, None)
+ if len(slice_list) == 0:
+ return
+ slice_id = slice_list[0]['slice_id']
+ l_attr = api.GetSliceAttributes(auth.auth, {'slice_id': slice_id}, None)
+ for attr in l_attr:
+ if "enabled" == attr['name'] and attr['value'] == "0":
+ logger.info("Deleted enable=0 attribute from slice %s" % slice)
+ api.DeleteSliceAttribute(auth.auth, attr['slice_attribute_id'])
+ except Exception, exc:
+ logger.info("enableSlices: %s" % exc)
+ print "exception: %s" % exc
#I'm commenting this because this really should be a manual process.
#'''
# logger.info("Suspending slice %s" % slice)
# api.SliceAttributeAdd(auth.auth, slice, "plc_slice_state", {"state" : "suspended"})
#
+def enableSliceCreation(nodename):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ try:
+ loginbase = siteId(nodename)
+ logger.info("Enabling slice creation for site %s" % loginbase)
+ if not config.debug:
+ logger.info("\tcalling UpdateSite(%s, enabled=True)" % loginbase)
+ api.UpdateSite(auth.auth, loginbase, {'enabled': True})
+ except Exception, exc:
+ print "ERROR: enableSliceCreation: %s" % exc
+ logger.info("ERROR: enableSliceCreation: %s" % exc)
'''
Removes ability to create slices. Returns previous max_slices
api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
try:
loginbase = siteId(nodename)
- numslices = api.GetSites(auth.auth, {"login_base": loginbase},
- ["max_slices"])[0]['max_slices']
+ #numslices = api.GetSites(auth.auth, {"login_base": loginbase},
+ # ["max_slices"])[0]['max_slices']
logger.info("Removing slice creation for site %s" % loginbase)
if not config.debug:
- api.UpdateSite(auth.auth, siteid, {'max_slices': 0})
+ #api.UpdateSite(auth.auth, loginbase, {'max_slices': 0})
+ api.UpdateSite(auth.auth, loginbase, {'enabled': False})
except Exception, exc:
logger.info("removeSliceCreation: %s" % exc)
'''
QED
'''
-def enableSliceCreation(nodename, maxslices):
- api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
- anon = {'AuthMethod': "anonymous"}
- siteid = api.AnonAdmQuerySite (anon, {"node_hostname": nodename})
- if len(siteid) == 1:
- logger.info("Enabling slice creation for site %s" % siteId(nodename))
- try:
- if not config.debug:
- api.AdmUpdateSite(auth.auth, siteid[0], {"max_slices" : maxslices})
- except Exception, exc:
- logger.info("API: %s" % exc)
- else:
- logger.debug("Cant find site for %s. Cannot enable creation." % nodename)
+#def enableSliceCreation(nodename, maxslices):
+# api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
+# anon = {'AuthMethod': "anonymous"}
+# siteid = api.AnonAdmQuerySite (anon, {"node_hostname": nodename})
+# if len(siteid) == 1:
+# logger.info("Enabling slice creation for site %s" % siteId(nodename))
+# try:
+# if not config.debug:
+# api.AdmUpdateSite(auth.auth, siteid[0], {"max_slices" : maxslices})
+# except Exception, exc:
+# logger.info("API: %s" % exc)
+# else:
+# logger.debug("Cant find site for %s. Cannot enable creation." % nodename)
def main():
logger.setLevel(logging.DEBUG)