2 # -*- coding: utf-8 -*-
4 from constants import TESTBED_ID
5 from nepi.core import testbed_impl
8 class TestbedController(testbed_impl.TestbedController):
9 def __init__(self, testbed_version):
10 super(TestbedController, self).__init__(TESTBED_ID, testbed_version)
12 self._home_directory = None
16 def home_directory(self):
17 return self._home_directory
24 self._home_directory = self._attributes.\
25 get_attribute_value("homeDirectory")
26 self._netns = self._load_netns_module()
28 def set(self, time, guid, name, value):
29 super(TestbedController, self).set(time, guid, name, value)
31 # TODO: take on account schedule time for the task
32 element = self._elements.get(guid)
34 setattr(element, name, value)
36 def get(self, time, guid, name):
37 # TODO: take on account schedule time for the task
38 element = self._elements.get(guid)
41 if hasattr(element, name):
43 return getattr(element, name)
45 # Try design-time attributes
46 return self.box_get(time, guid, name)
47 except KeyError, AttributeError:
50 def get_route(self, guid, index, attribute):
51 # TODO: fetch real data from netns
53 return self.box_get_route(guid, int(index), attribute)
54 except KeyError, AttributeError:
57 def get_address(self, guid, index, attribute='Address'):
58 # TODO: fetch real data from netns
60 return self.box_get_address(guid, int(index), attribute)
61 except KeyError, AttributeError:
65 def action(self, time, guid, action):
66 raise NotImplementedError
69 for trace in self._traces.values():
71 for element in self._elements.values():
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))
78 def follow_trace(self, trace_id, trace):
79 self._traces[trace_id] = trace
81 def _load_netns_module(self):
82 # TODO: Do something with the configuration!!!
85 netns_mod = sys.modules["netns"]
87 enable_debug = self._attributes.get_attribute_value("enableDebug")
89 netns_mod.environ.set_log_level(netns_mod.environ.LOG_DEBUG)