import xmlrpclib
from PLC.Slices import Slices
-from PLC.SliceTags import SliceTags
+from PLC.SliceTags import SliceTags, SliceTag
+from PLC.TagTypes import TagTypes
from PLC.Nodes import Nodes
from PLC.Config import Config
from pyaspects.meta import MetaAspect
def __init__(self):
self.config = Config("/etc/planetlab/plc_config")
- self.log = open("/var/log/omf/plc_slice_calls.log", "a")
+ self.slice = None
+
+ # this was only for debugging, no need to log all method calls here -baris
+ # self.log = open("/var/log/omf/plc_slice_calls.log", "a")
+ self.log = None
+
def logit(self, call, args, kwargs, data, slice):
+ if not self.log: return
+
self.log.write("%s : args: %s kwargs: %s\n" % (call, args, kwargs))
self.log.write("data: %s\n" % data)
self.log.write("%s\n\n" % slice)
# return None
return slice
- def get_node_hostname(self, api, node_id):
- node_filter = {'node_id': node_id }
+ def get_node_hostname(self, api, node_id_or_hostname):
+ node_filter = {}
+ try:
+ node_filter['node_id'] = int(str(node_id_or_hostname))
+ except ValueError:
+ # we have a hostname
+ node_filter['hostname'] = node_id_or_hostname
+
try:
node = Nodes(api, node_filter = node_filter)[0]
return node['hostname']
def get_slice_tags(self, api, slice_id):
return SliceTags(api, slice_tag_filter = {'slice_id': slice_id})
+ def get_tag_type(self, api, tagname):
+ try:
+ tag = TagTypes(api, {'tagname':tagname})[0]
+ return tag
+ except IndexError:
+ return None
+
def create_slice(self, slice):
pass
slice_name_or_id = None
node_ids = None
- # DeleteSlice shall be handled before the actual method call;
- # after the call we won't be able to acess the slice.
- if api_method_name == "DeleteSlice":
+ if api_method_name == "AddSlice":
+ slice_name_or_id = args[1]['name']
+ elif api_method_name == "AddSliceToNodes" or api_method_name == "DeleteSliceFromNodes":
+ slice_name_or_id = args[1]
+ node_ids = args[2]
+ elif api_method_name == "AddSliceTag":
+ slice_name_or_id = args[1]
+ elif api_method_name == "DeleteSlice":
slice_name_or_id = args[1]
else: # ignore the rest
+ #self.logit(wobj.name, args, kwargs, data, "SLICE")
+ self.slice = None
return
- slice = self.get_slice(wobj.api, slice_name_or_id)
- if not slice:
- return
-
- if api_method_name == "DeleteSlice":
- self.delete_slice(slice['name'])
+ self.slice = self.get_slice(wobj.api, slice_name_or_id)
self.logit(wobj.name, args, kwargs, data, slice)
# aspect method
def after(self, wobj, data, *args, **kwargs):
- api_method_name = wobj.name
- slice_name_or_id = None
- node_ids = None
- if api_method_name == "AddSlice":
- slice_name_or_id = args[1]['name']
- elif api_method_name == "AddSliceToNodes" or api_method_name == "DeleteSliceFromNodes":
- slice_name_or_id = args[1]
- node_ids = args[2]
- else: # ignore the rest
- #self.logit(wobj.name, args, kwargs, data, "SLICE")
+ if not self.slice:
return
- slice = self.get_slice(wobj.api, slice_name_or_id)
- if not slice:
+ if data.has_key("method_return_value") and data['method_return_value'] == 1:
+ # return value 1 means that API call was successful, we can go on.
+ pass
+ else:
return
+ api_method_name = wobj.name
+
if api_method_name == "AddSlice":
self.create_slice(slice['name'])
elif api_method_name == "AddSliceToNodes":
for node_id in node_ids:
node_hostname = self.get_node_hostname(wobj.api, node_id)
self.add_resource(slice['name'], node_hostname)
+ elif api_method_name == "DeleteSlice":
+ self.delete_slice(slice['name'])
elif api_method_name == "DeleteSliceFromNodes":
for node_id in node_ids:
node_hostname = self.get_node_hostname(wobj.api, node_id)
self.delete_resource(slice['name'], node_hostname)
+ elif api_method_name == "AddSliceTag":
+ # OMF slices need to have dotsshmount vsys tag set to be
+ # able to access users' public keys.
+ tag_type_id_or_name = args[2]
+ omf_tag = self.get_tag_type(wobj.api, "omf_control")
+ vsys_tag = self.get_tag_type(wobj.api, "vsys")
+ if tag_type_id_or_name in (omf_tag['tagname'], omf_tag['tag_type_id']):
+ slice_tag = SliceTag(wobj.api)
+ slice_tag['slice_id'] = slice['slice_id']
+ slice_tag['tag_type_id'] = vsys_tag['tag_type_id']
+ slice_tag['value'] = u'dotsshmount'
+ slice_tag.sync()
self.logit(wobj.name, args, kwargs, data, slice)
BaseOMF.__init__(self)
slicemgr_url = self.config.PLC_OMF_SLICEMGR_URL
- self.server = xmlrpclib.ServerProxy(slicemgr_url)
+ self.server = xmlrpclib.ServerProxy(slicemgr_url, allow_none = 1)
def create_slice(self, slice):
self.server.createSlice(slice)