unify handling calls in before and after aspects. check method's return value before...

[plcapi.git] / aspects / omfaspects.py

diff --git a/aspects/omfaspects.py b/aspects/omfaspects.py
index 1735490..32eb364 100644 (file)
--- a/aspects/omfaspects.py
+++ b/aspects/omfaspects.py
@@ -4,7 +4,8 @@ import os
 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
@@ -14,6 +15,8 @@ class BaseOMF(object):
 
     def __init__(self):
         self.config = Config("/etc/planetlab/plc_config")
+        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
@@ -64,6 +67,13 @@ class BaseOMF(object):
     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
 
@@ -82,50 +92,61 @@ class BaseOMF(object):
         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)