src/nepi/testbeds/netns/execute.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 from constants import TESTBED_ID
   5 from nepi.core import testbed_impl
   6 from nepi.util.constants import TESTBED_STATUS_CREATED
   7 import os
   8
   9 class TestbedController(testbed_impl.TestbedController):
  10     def __init__(self, testbed_version):
  11         super(TestbedController, self).__init__(TESTBED_ID, testbed_version)
  12         self._netns = None
  13         self._home_directory = None
  14         self._traces = dict()
  15
  16     @property
  17     def home_directory(self):
  18         return self._home_directory
  19
  20     @property
  21     def netns(self):
  22         return self._netns
  23
  24     def do_setup(self):
  25         self._home_directory = self._attributes.\
  26             get_attribute_value("homeDirectory")
  27         self._netns = self._load_netns_module()
  28         super(TestbedController, self).do_setup()
  29
  30     def set(self, time, guid, name, value):
  31         super(TestbedController, self).set(time, guid, name, value)
  32         # TODO: take on account schedule time for the task
  33         element = self._elements.get(guid)
  34         if element:
  35             setattr(element, name, value)
  36
  37     def get(self, time, guid, name):
  38         value = super(TestbedController, self).get(time, guid, name)
  39         # TODO: take on account schedule time for the task
  40         factory_id = self._create[guid]
  41         factory = self._factories[factory_id]
  42         if factory.box_attributes.is_attribute_design_only(name):
  43             return value
  44         element = self._elements.get(guid)
  45         try:
  46             return getattr(element, name)
  47         except KeyError, AttributeError:
  48             return value
  49
  50     def get_route(self, guid, index, attribute):
  51         # TODO: fetch real data from netns
  52         try:
  53             return self.box_get_route(guid, int(index), attribute)
  54         except KeyError, AttributeError:
  55             return None
  56
  57     def get_address(self, guid, index, attribute='Address'):
  58         # TODO: fetch real data from netns
  59         try:
  60             return self.box_get_address(guid, int(index), attribute)
  61         except KeyError, AttributeError:
  62             return None
  63
  64
  65     def action(self, time, guid, action):
  66         raise NotImplementedError
  67
  68     def shutdown(self):
  69         for trace in self._traces.values():
  70             trace.close()
  71         for element in self._elements.values():
  72             element.destroy()
  73
  74     def trace_filename(self, guid, trace_id):
  75         # TODO: Need to be defined inside a home!!!! with and experiment id_code
  76         return os.path.join(self.home_directory, "%d_%s" % (guid, trace_id))
  77
  78     def follow_trace(self, trace_id, trace):
  79         self._traces[trace_id] = trace
  80
  81     def _load_netns_module(self):
  82         # TODO: Do something with the configuration!!!
  83         import sys
  84         __import__("netns")
  85         netns_mod = sys.modules["netns"]
  86         # enable debug
  87         enable_debug = self._attributes.get_attribute_value("enableDebug")
  88         if enable_debug:
  89             netns_mod.environ.set_log_level(netns_mod.environ.LOG_DEBUG)
  90         return netns_mod
  91