#
# Faiyaz Ahmed <faiyaza@cs.princeton.edu
#
-# $Id: plc.py,v 1.16 2007/07/03 19:59:02 soltesz Exp $
+# $Id: plc.py,v 1.18 2007/08/29 17:26:50 soltesz Exp $
#
-from emailTxt import *
import xml, xmlrpclib
import logging
import auth
#XMLRPC_SERVER = config.XMLRPC_SERVER
config = config()
+api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+
+def getAPI(url):
+ api = xmlrpclib.Server(url, verbose=False, allow_none=True)
+ return api
+
+class PLC:
+ def __init__(self, auth, url):
+ self.auth = auth
+ self.url = url
+ self.api = xmlrpclib.Server(self.url, verbose=False, allow_none=True)
+
+ def __getattr__(self, name):
+ method = getattr(self.api, name)
+ if method is None:
+ raise AssertionError("method does not exist")
+
+ return lambda *params : method(self.auth, *params)
+
+ def __repr__(self):
+ return self.api.__repr__()
'''
Returns list of nodes in dbg as reported by PLC
logger.info("%s doesn't have PCU" % nodename)
return False
+def GetPCUs(filter=None, fields=None):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ pcu_list = api.GetPCUs(auth.auth, filter, fields)
+ return pcu_list
+
'''
Returns all site nodes for site id (loginbase).
'''
-def getSiteNodes(loginbase):
+def getSiteNodes(loginbase, fields=None):
api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
nodelist = []
anon = {'AuthMethod': "anonymous"}
try:
- nodeids = api.GetSites(anon, {"login_base": loginbase})[0]['node_ids']
- for node in api.GetNodes(anon, {"node_id": nodeids}):
+ nodeids = api.GetSites(anon, {"login_base": loginbase}, fields)[0]['node_ids']
+ for node in api.GetNodes(anon, {"node_id": nodeids}, ['hostname']):
nodelist.append(node['hostname'])
except Exception, exc:
logger.info("getSiteNodes: %s" % exc)
+ print "getSiteNodes: %s" % exc
return nodelist
-def getSites(filter=None):
+def getPersons(filter=None, fields=None):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
+ persons = []
+ try:
+ persons = api.GetPersons(auth.auth, filter, fields)
+ except Exception, exc:
+ print "getPersons: %s" % exc
+ logger.info("getPersons: %s" % exc)
+ return persons
+
+def getSites(filter=None, fields=None):
api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
sites = []
anon = {'AuthMethod': "anonymous"}
try:
- sites = api.GetSites(anon, filter, None)
+ sites = api.GetSites(anon, filter, fields)
except Exception, exc:
- print "getSiteNodes2: %s" % exc
- logger.info("getSiteNodes2: %s" % exc)
+ print "getSites: %s" % exc
+ logger.info("getSites: %s" % exc)
return sites
def getSiteNodes2(loginbase):
nodenetworks = api.GetNodeNetworks(auth.auth, filter, None)
return nodenetworks
-def getNodes(filter=None):
+def getNodes(filter=None, fields=None):
api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False, allow_none=True)
- nodes = api.GetNodes(auth.auth, filter, None) #['boot_state', 'hostname',
+ nodes = api.GetNodes(auth.auth, filter, fields)
+ #['boot_state', 'hostname',
#'site_id', 'date_created', 'node_id', 'version', 'nodenetwork_ids',
#'last_updated', 'peer_node_id', 'ssh_rsa_key' ])
return nodes
except Exception, exc:
logger.info("nodeBootState: %s" % exc)
+def updateNodeKey(nodename, key):
+ api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
+ try:
+ return api.UpdateNode(auth.auth, nodename, {'key': key})
+ except Exception, exc:
+ logger.info("updateNodeKey: %s" % exc)
+
'''
Sends Ping Of Death to node.
'''