aspects/omfaspects.py

   1 # Baris Metin <tmetin@sophia.inria.fr>
   2
   3 import os
   4 import xmlrpclib
   5
   6 from PLC.Slices import Slices
   7 from PLC.SliceTags import SliceTags
   8 from PLC.Nodes import Nodes
   9 from PLC.Config import Config
  10 from pyaspects.meta import MetaAspect
  11
  12
  13 class BaseOMF(object):
  14
  15     def __init__(self):
  16         self.config = Config("/etc/planetlab/plc_config")
  17         self.log = open("/var/log/omf/plc_slice_calls.log", "a")
  18
  19     def logit(self, call, args, kwargs, data, slice):
  20         self.log.write("%s : args: %s  kwargs: %s\n" % (call, args, kwargs))
  21         self.log.write("data: %s\n" % data)
  22         self.log.write("%s\n\n" % slice)
  23         self.log.flush()
  24
  25
  26     def get_slice(self, api, id_or_name):
  27         slice_filter = {}
  28         try: # if integer
  29             slice_filter['slice_id'] = int(str(id_or_name))
  30         except ValueError:
  31             # we have a string
  32             slice_filter['name'] = id_or_name
  33         slice = Slices(api, slice_filter = slice_filter)[0]
  34 # don't bother to check for slice tags for the moment. we'll only
  35 # create XMPP pubsub groups for all slices
  36 #
  37 #         slice_tags = SliceTags(api, slice_tag_filter = { 'slice_id': slice['slice_id'] })
  38 #         omf_tag = [tag for tag in slice_tags if tag['name'] == 'omf']
  39 #         if omf_tag and omf_tag['value'] not in ('false','0','no'):
  40 #             # OK, slice has the "omf" tag set.
  41 #             return slice
  42 #         return None
  43         return slice
  44
  45     def get_node_hostname(self, api, node_id):
  46         node_filter = {'node_id': node_id }
  47         try:
  48             node = Nodes(api, node_filter = node_filter)[0]
  49             return node['hostname']
  50         except IndexError:
  51             return None
  52
  53     def get_slice_tags(self, api, slice_id):
  54         return SliceTags(api, slice_tag_filter = {'slice_id': slice_id})
  55
  56     def create_slice(self, slice):
  57         pass
  58
  59     def add_resource(self, slice, resource):
  60         pass
  61
  62     def delete_slice(self, slice):
  63         pass
  64
  65     def delete_resource(self, slice, resource):
  66         pass
  67
  68     # aspect method
  69     def before(self, wobj, data, *args, **kwargs):
  70         api_method_name = wobj.name
  71         slice_name_or_id = None
  72         node_ids = None
  73
  74         # DeleteSlice shall be handled before the actual method call;
  75         # after the call we won't be able to acess the slice.
  76         if api_method_name == "DeleteSlice":
  77             slice_name_or_id = args[1]
  78         else: # ignore the rest
  79             return
  80
  81         slice = self.get_slice(wobj.api, slice_name_or_id)
  82         if not slice:
  83             return
  84
  85         if api_method_name == "DeleteSlice":
  86             self.delete_slice(slice['name'])
  87
  88         self.logit(wobj.name, args, kwargs, data, slice)
  89
  90     # aspect method
  91     def after(self, wobj, data, *args, **kwargs):
  92         api_method_name = wobj.name
  93         slice_name_or_id = None
  94         node_ids = None
  95         if api_method_name == "AddSlice":
  96             slice_name_or_id = args[1]['name']
  97         elif api_method_name == "AddSliceToNodes" or api_method_name == "DeleteSliceFromNodes":
  98             slice_name_or_id = args[1]
  99             node_ids = args[2]
 100         else: # ignore the rest
 101             #self.logit(wobj.name, args, kwargs, data, "SLICE")
 102             return
 103
 104         slice = self.get_slice(wobj.api, slice_name_or_id)
 105         if not slice:
 106             return
 107
 108         if api_method_name == "AddSlice":
 109             self.create_slice(slice['name'])
 110         elif api_method_name == "AddSliceToNodes":
 111             for node_id in node_ids:
 112                 node_hostname = self.get_node_hostname(wobj.api, node_id)
 113                 self.add_resource(slice['name'], node_hostname)
 114         elif api_method_name == "DeleteSliceFromNodes":
 115             for node_id in node_ids:
 116                 node_hostname = self.get_node_hostname(wobj.api, node_id)
 117                 self.delete_resource(slice['name'], node_hostname)
 118
 119         self.logit(wobj.name, args, kwargs, data, slice)
 120
 121
 122
 123 class OMFAspect_xmlrpc(BaseOMF):
 124     __metaclass__ = MetaAspect
 125     name = "omfaspect_xmlrpc"
 126
 127     def __init__(self):
 128         BaseOMF.__init__(self)
 129
 130         slicemgr_url = self.config.PLC_OMF_SLICEMGR_URL
 131         self.server = xmlrpclib.ServerProxy(slicemgr_url)
 132
 133     def create_slice(self, slice):
 134         self.server.createSlice(slice)
 135
 136     def add_resource(self, slice, resource):
 137         self.server.addResource(slice, resource)
 138
 139     def delete_slice(self, slice):
 140         self.server.deleteSlice(slice)
 141
 142     def delete_resource(self, slice, resource):
 143         self.server.removeResource(slice, resource)
 144
 145     def before(self, wobj, data, *args, **kwargs):
 146         BaseOMF.before(self, wobj, data, *args, **kwargs)
 147
 148     def after(self, wobj, data, *args, **kwargs):
 149         BaseOMF.after(self, wobj, data, *args, **kwargs)
 150
 151
 152
 153 OMFAspect = OMFAspect_xmlrpc
 154