X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Futil%2Fapi.py;h=39d6538254d2fa04ba85a52d29d6fcf38ec61351;hb=8d1e057a4da2ae30f93b3fd4e5ac7b05d0c42bdf;hp=bcbea6f0803ad84e9f57335a39a18ff6aaa4b045;hpb=5751f6cf1d6a5f1b373d048c114df91f035dfcf2;p=sfa.git diff --git a/sfa/util/api.py b/sfa/util/api.py index bcbea6f0..39d65382 100644 --- a/sfa/util/api.py +++ b/sfa/util/api.py @@ -1,8 +1,5 @@ # -# Geniwrapper XML-RPC and SOAP interfaces -# -### $Id: api.py 15596 2009-10-31 21:42:05Z anil $ -### $URL: https://svn.planet-lab.org/svn/sfa/trunk/sfa/plc/api.py $ +# SFA XML-RPC and SOAP interfaces # import sys @@ -11,14 +8,17 @@ import traceback import string import xmlrpclib -from sfa.trust.auth import Auth -from sfa.util.config import * from sfa.util.faults import * -from sfa.util.debug import * +from sfa.util.config import * +import sfa.util.xmlrpcprotocol as xmlrpcprotocol +from sfa.util.sfalogging import logger +from sfa.trust.auth import Auth +from sfa.util.cache import Cache from sfa.trust.credential import * from sfa.trust.certificate import * -from sfa.util.misc import * -from sfa.util.sfalogging import * + +# this is wrong all right, but temporary +from sfa.managers.import_manager import import_manager # See "2.2 Characters" in the XML specification: # @@ -95,35 +95,91 @@ def import_deep(name): mod = getattr(mod, comp) return mod +class ManagerWrapper: + """ + This class acts as a wrapper around an SFA interface manager module, but + can be used with any python module. The purpose of this class is raise a + SfaNotImplemented exception if someone attempts to use an attribute + (could be a callable) thats not available in the library by checking the + library using hasattr. This helps to communicate better errors messages + to the users and developers in the event that a specifiec operation + is not implemented by a libarary and will generally be more helpful than + the standard AttributeError + """ + def __init__(self, manager, interface): + self.manager = manager + self.interface = interface + + def __getattr__(self, method): + if not hasattr(self.manager, method): + raise SfaNotImplemented(method, self.interface) + return getattr(self.manager, method) + class BaseAPI: - def __init__(self, config = "/etc/sfa/sfa_config", encoding = "utf-8", methods='sfa.methods', - peer_cert = None, interface = None, key_file = None, cert_file = None): + protocol = None + + def __init__(self, config = "/etc/sfa/sfa_config.py", encoding = "utf-8", + methods='sfa.methods', peer_cert = None, interface = None, + key_file = None, cert_file = None, cache = None): self.encoding = encoding # flat list of method names - self.methods_module = methods_module = __import__(methods, fromlist=[methods]) self.methods = methods_module.all # Better just be documenting the API if config is None: return - # Load configuration self.config = Config(config) self.auth = Auth(peer_cert) + self.hrn = self.config.SFA_INTERFACE_HRN self.interface = interface self.key_file = key_file self.key = Keypair(filename=self.key_file) self.cert_file = cert_file self.cert = Certificate(filename=self.cert_file) + self.cache = cache + if self.cache is None: + self.cache = Cache() self.credential = None - + self.source = None self.time_format = "%Y-%m-%d %H:%M:%S" - self.logger=get_sfa_logger() + self.logger = logger + + # load registries + from sfa.server.registry import Registries + self.registries = Registries() + # load aggregates + from sfa.server.aggregate import Aggregates + self.aggregates = Aggregates() + + + def get_interface_manager(self, manager_base = 'sfa.managers'): + """ + Returns the appropriate manager module for this interface. + Modules are usually found in sfa/managers/ + """ + manager=None + if self.interface in ['registry']: + manager=import_manager ("registry", self.config.SFA_REGISTRY_TYPE) + elif self.interface in ['aggregate']: + manager=import_manager ("aggregate", self.config.SFA_AGGREGATE_TYPE) + elif self.interface in ['slicemgr', 'sm']: + manager=import_manager ("slice", self.config.SFA_SM_TYPE) + elif self.interface in ['component', 'cm']: + manager=import_manager ("component", self.config.SFA_CM_TYPE) + if not manager: + raise SfaAPIError("No manager for interface: %s" % self.interface) + + # this isnt necessary but will help to produce better error messages + # if someone tries to access an operation this manager doesn't implement + manager = ManagerWrapper(manager, self.interface) + + return manager def callable(self, method): """ @@ -131,7 +187,7 @@ class BaseAPI: """ # Look up method if method not in self.methods: - raise GeniInvalidAPIMethod, method + raise SfaInvalidAPIMethod, method # Get new instance of method try: @@ -140,7 +196,7 @@ class BaseAPI: callablemethod = getattr(module, classname)(self) return getattr(module, classname)(self) except ImportError, AttributeError: - raise GeniInvalidAPIMethod, method + raise SfaInvalidAPIMethod, method def call(self, source, method, *args): """ @@ -149,19 +205,26 @@ class BaseAPI: """ function = self.callable(method) function.source = source + self.source = source return function(*args) - def handle(self, source, data): + + def handle(self, source, data, method_map): """ Handle an XML-RPC or SOAP request from the specified source. """ # Parse request into method name and arguments try: interface = xmlrpclib + self.protocol = 'xmlrpclib' (args, method) = xmlrpclib.loads(data) + if method_map.has_key(method): + method = method_map[method] methodresponse = True + except Exception, e: if SOAPpy is not None: + self.protocol = 'soap' interface = SOAPpy (r, header, body, attrs) = parseSOAPRPC(data, header = 1, body = 1, attrs = 1) method = r._name @@ -172,26 +235,45 @@ class BaseAPI: try: result = self.call(source, method, *args) + except SfaFault, fault: + result = fault except Exception, fault: - traceback.print_exc(file = log) - # Handle expected faults - if interface == xmlrpclib: - result = fault - methodresponse = None - elif interface == SOAPpy: - result = faultParameter(NS.ENV_T + ":Server", "Method Failed", method) - result._setDetail("Fault %d: %s" % (fault.faultCode, fault.faultString)) - else: - raise + logger.log_exc("BaseAPI.handle has caught Exception") + result = SfaAPIError(fault) + # Return result - if interface == xmlrpclib: - if not isinstance(result, GeniFault): + response = self.prepare_response(result, method) + return response + + def prepare_response(self, result, method=""): + """ + convert result to a valid xmlrpc or soap response + """ + + if self.protocol == 'xmlrpclib': + if not isinstance(result, SfaFault): result = (result,) + response = xmlrpclib.dumps(result, methodresponse = True, encoding = self.encoding, allow_none = 1) + elif self.protocol == 'soap': + if isinstance(result, Exception): + result = faultParameter(NS.ENV_T + ":Server", "Method Failed", method) + result._setDetail("Fault %d: %s" % (result.faultCode, result.faultString)) + else: + response = buildSOAP(kw = {'%sResponse' % method: {'Result': result}}, encoding = self.encoding) + else: + if isinstance(result, Exception): + raise result + + return response - data = xmlrpclib.dumps(result, methodresponse = True, encoding = self.encoding, allow_none = 1) - elif interface == SOAPpy: - data = buildSOAP(kw = {'%sResponse' % method: {'Result': result}}, encoding = self.encoding) - - return data - + def get_cached_server_version(self, server): + cache_key = server.url + "-version" + server_version = None + if self.cache: + server_version = self.cache.get(cache_key) + if not server_version: + server_version = server.GetVersion() + # cache version for 24 hours + self.cache.add(cache_key, server_version, ttl= 60*60*24) + return server_version