X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=api_calls.py;h=4a5558e5ad368f14a9d1ead195522833f3e3b477;hb=973e601b8663412a3c31860b2a77fa7b87c8c360;hp=a17539177b3c4f66de7537561bc4ddc502262e8c;hpb=164e7fc96baccd6ae5caa57b794fb9966167eca2;p=nodemanager.git diff --git a/api_calls.py b/api_calls.py index a175391..4a5558e 100644 --- a/api_calls.py +++ b/api_calls.py @@ -1,6 +1,4 @@ -# $Id$ -# $URL$ - +# """Sliver manager API. This module exposes an XMLRPC interface that allows PlanetLab users to @@ -20,6 +18,7 @@ import socket import struct import threading import xmlrpclib +import sliver_lxc try: from PLC.Parameter import Parameter, Mixed @@ -31,14 +30,14 @@ except: import accounts import logger -# TODO: These try/excepts are a hack to allow doc/DocBookLocal.py to -# import this file in order to extract the documentation from each -# exported function. +# TODO: These try/excepts are a hack to allow doc/DocBookLocal.py to +# import this file in order to extract the documentation from each +# exported function. # A better approach will involve more extensive code splitting, I think. try: import database except: import logger as database -try: import sliver_vs -except: import logger as sliver_vs +#try: import sliver_vs +#except: import logger as sliver_vs import ticket as ticket_module import tools @@ -94,27 +93,27 @@ def export_to_docbook(**kwargs): # accepts, # returns -@export_to_docbook(roles=['self'], - accepts=[], +@export_to_docbook(roles=['self'], + accepts=[], returns=Parameter([], 'A list of supported functions')) @export_to_api(0) def Help(): """Get a list of functions currently supported by the Node Manager API""" names=api_method_dict.keys() names.sort() - return ''.join(['**** ' + api_method_dict[name].__name__ + '\n' + api_method_dict[name].__doc__ + '\n' + return ''.join(['**** ' + api_method_dict[name].__name__ + '\n' + api_method_dict[name].__doc__ + '\n' for name in names]) -@export_to_docbook(roles=['self'], - accepts=[Parameter(str, 'A ticket returned from GetSliceTicket()')], +@export_to_docbook(roles=['self'], + accepts=[Parameter(str, 'A ticket returned from GetSliceTicket()')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def Ticket(ticket): """The Node Manager periodically polls the PLC API for a list of all - slices that are allowed to exist on the given node. Before + slices that are allowed to exist on the given node. Before actions are performed on a delegated slice (such as creation), - a controller slice must deliver a valid slice ticket to NM. - + a controller slice must deliver a valid slice ticket to NM. + This ticket is the value retured by PLC's GetSliceTicket() API call.""" try: data = ticket_module.verify(ticket) @@ -126,8 +125,8 @@ def Ticket(ticket): except Exception, err: raise xmlrpclib.Fault(102, 'Ticket error: ' + str(err)) -@export_to_docbook(roles=['self'], - accepts=[Parameter(str, 'A ticket returned from GetSlivers()')], +@export_to_docbook(roles=['self'], + accepts=[Parameter(str, 'A ticket returned from GetSlivers()')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def AdminTicket(ticket): @@ -144,15 +143,15 @@ def AdminTicket(ticket): @export_to_docbook(roles=['self'], - accepts=[], + accepts=[], returns={'sliver_name' : Parameter(int, 'the associated xid')}) @export_to_api(0) def GetXIDs(): """Return an dictionary mapping Slice names to XIDs""" - return dict([(pwent[0], pwent[2]) for pwent in pwd.getpwall() if pwent[6] == sliver_vs.Sliver_VS.SHELL]) + return dict([(pwent[0], pwent[2]) for pwent in pwd.getpwall() if pwent[6] == sliver_lxc.Sliver_LXC.SHELL]) @export_to_docbook(roles=['self'], - accepts=[], + accepts=[], returns={ 'sliver_name' : Parameter(str, 'the associated SSHKey')}) @export_to_api(0) def GetSSHKeys(): @@ -164,36 +163,36 @@ def GetSSHKeys(): return keydict -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def Create(sliver_name): """Create a non-PLC-instantiated sliver""" rec = sliver_name - if rec['instantiation'] == 'delegated': + if rec['instantiation'] == 'delegated': accounts.get(rec['name']).ensure_created(rec) logger.log("api_calls: Create %s"%rec['name']) - else: + else: raise Exception, "Only PLC can create non delegated slivers." -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def Destroy(sliver_name): """Destroy a non-PLC-instantiated sliver""" - rec = sliver_name - if rec['instantiation'] == 'delegated': + rec = sliver_name + if rec['instantiation'] == 'delegated': accounts.get(rec['name']).ensure_destroyed() logger.log("api_calls: Destroy %s"%rec['name']) - else: + else: raise Exception, "Only PLC can destroy non delegated slivers." -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def Start(sliver_name): @@ -203,8 +202,8 @@ def Start(sliver_name): logger.log("api_calls: Start %s"%rec['name']) -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def Stop(sliver_name): @@ -214,8 +213,8 @@ def Stop(sliver_name): logger.log("api_calls: Stop %s"%rec['name']) -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=Parameter(int, '1 if successful')) @export_to_api(1) def ReCreate(sliver_name): @@ -226,8 +225,8 @@ def ReCreate(sliver_name): accounts.get(rec['name']).start(rec) logger.log("api_calls: ReCreate %s"%rec['name']) -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=Parameter(dict, "A resource specification")) @export_to_api(1) def GetEffectiveRSpec(sliver_name): @@ -236,8 +235,8 @@ def GetEffectiveRSpec(sliver_name): return rec.get('_rspec', {}).copy() -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns={"resource name" : Parameter(int, "amount")}) @export_to_api(1) def GetRSpec(sliver_name): @@ -246,8 +245,8 @@ def GetRSpec(sliver_name): return rec.get('rspec', {}).copy() -@export_to_docbook(roles=['nm-controller', 'self'], - accepts=[Parameter(str, 'A sliver/slice name.')], +@export_to_docbook(roles=['nm-controller', 'self'], + accepts=[Parameter(str, 'A sliver/slice name.')], returns=[Mixed(Parameter(str, 'recipient slice name'), Parameter(str, 'resource name'), Parameter(int, 'resource amount'))]) @@ -260,13 +259,13 @@ def GetLoans(sliver_name): def validate_loans(loans): """Check that is a list of valid loan specifications.""" - def validate_loan(loan): + def validate_loan(loan): return (type(loan)==list or type(loan)==tuple) and len(loan)==3 \ and type(loan[0])==str and type(loan[1])==str and loan[1] in database.LOANABLE_RESOURCES and type(loan[2])==int and loan[2]>=0 return type(loans)==list and False not in [validate_loan(load) for loan in loans] -@export_to_docbook(roles=['nm-controller', 'self'], +@export_to_docbook(roles=['nm-controller', 'self'], accepts=[ Parameter(str, 'A sliver/slice name.'), [Mixed(Parameter(str, 'recipient slice name'), Parameter(str, 'resource name'), @@ -282,12 +281,12 @@ def SetLoans(sliver_name, loans): in the future. As well, there is currently no asynchronous notification of loss of resources.""" rec = sliver_name - if not validate_loans(loans): + if not validate_loans(loans): raise xmlrpclib.Fault(102, 'Invalid argument: the second argument must be a well-formed loan specification') rec['_loans'] = loans database.db.sync() -@export_to_docbook(roles=['nm-controller', 'self'], +@export_to_docbook(roles=['nm-controller', 'self'], returns=Parameter(dict, 'Record dictionary')) @export_to_api(0) def GetRecord(sliver_name):