sfa/generic/__init__.py

   1 from sfa.util.sfalogging import logger
   2 from sfa.util.config import Config
   3
   4 from sfa.managers.managerwrapper import ManagerWrapper
   5
   6 # a bundle is the combination of
   7 # (*) an api that reacts on the incoming requests to trigger the API methods
   8 # (*) a manager that implements the function of the service,
   9 #     either aggregate or registry
  10 # (*) a driver that controls the underlying testbed
  11 #
  12 #
  13 # The Generic class is a utility that uses the configuration to figure out
  14 # which combination of these pieces need to be put together
  15 # from config.
  16 # this extra indirection is needed to adapt to the current naming scheme
  17 # where we have 'pl' and 'plc' and components and the like, that does not
  18 # yet follow a sensible scheme
  19
  20 # needs refinements to cache more efficiently, esp. wrt the config
  21
  22
  23 class Generic:
  24
  25     def __init__(self, flavour, config):
  26         self.flavour = flavour
  27         self.config = config
  28
  29     # proof of concept
  30     # example flavour='pl' -> sfa.generic.pl.pl()
  31     @staticmethod
  32     def the_flavour(flavour=None, config=None):
  33         if config is None:
  34             config = Config()
  35         if flavour is None:
  36             flavour = config.SFA_GENERIC_FLAVOUR
  37         flavour = flavour.lower()
  38         # mixed = flavour.capitalize()
  39         module_path = "sfa.generic.%s" % flavour
  40         classname = "%s" % flavour
  41         logger.debug("Generic.the_flavour with flavour=%s" % flavour)
  42         try:
  43             module = __import__(module_path, globals(), locals(), [classname])
  44             return getattr(module, classname)(flavour, config)
  45         except:
  46             logger.log_exc(
  47                 "Cannot locate generic instance with flavour=%s" % flavour)
  48
  49     # provide default for importer_class
  50     def importer_class(self):
  51         return None
  52
  53     # in the simplest case these can be redefined to the class/module objects to be used
  54     # see pl.py for an example
  55     # some descendant of SfaApi
  56     def api_class(self): pass
  57     # the python classes to use to build up the context
  58
  59     def registry_class(self): pass
  60
  61     def aggregate_class(self): pass
  62
  63     # build an API object
  64     # insert a manager instance
  65     def make_api(self, *args, **kwargs):
  66         # interface is a required arg
  67         if not 'interface' in kwargs:
  68             logger.critical("Generic.make_api: no interface found")
  69         api = self.api_class()(*args, **kwargs)
  70         # xxx can probably drop support for managers implemented as modules
  71         # which makes it a bit awkward
  72         manager_class_or_module = self.make_manager(api.interface)
  73         driver = self.make_driver(api)
  74         # arrange stuff together
  75         # add a manager wrapper
  76         manager_wrap = ManagerWrapper(
  77             manager_class_or_module, api.interface, api.config)
  78         api.manager = manager_wrap
  79         # add it in api as well; driver.api is set too as part of make_driver
  80         api.driver = driver
  81         return api
  82
  83     def make_manager(self, interface):
  84         """
  85         interface expected in ['registry', 'aggregate']
  86         flavour is e.g. 'pl' or 'max' or whatever
  87         """
  88         flavour = self.flavour
  89         message = "Generic.make_manager for interface=%s and flavour=%s" % (
  90             interface, flavour)
  91
  92         classname = "%s_manager_class" % interface
  93         try:
  94             module_or_class = getattr(self, classname)()
  95             logger.debug("%s : %s" % (message, module_or_class))
  96             # this gets passed to ManagerWrapper that will
  97             # call the class constructor if it's a class,
  98             # or use the module as is if it's a module
  99             # so bottom line is, don't try the constructor here
 100             return module_or_class
 101         except Exception:
 102             logger.log_exc(message)
 103             exit(1)
 104
 105     # need interface to select the right driver
 106     def make_driver(self, api):
 107         config = api.config
 108         interface = api.interface
 109         flavour = self.flavour
 110         message = "Generic.make_driver for flavour=%s and interface=%s" % (
 111             flavour, interface)
 112
 113         classname = "driver_class"
 114         try:
 115             class_obj = getattr(self, classname)()
 116             logger.debug("%s : %s" % (message, class_obj))
 117             return class_obj(api)
 118         except Exception:
 119             logger.log_exc(message)
 120             exit(1)