applied the except and raise fixers to the master branch to close the gap with py3
[nepi.git] / src / nepi / resources / ns3 / ns3errormodel.py
1 #
2 #    NEPI, a framework to manage network experiments
3 #    Copyright (C) 2014 INRIA
4 #
5 #    This program is free software: you can redistribute it and/or modify
6 #    it under the terms of the GNU General Public License version 2 as
7 #    published by the Free Software Foundation;
8 #
9 #    This program is distributed in the hope that it will be useful,
10 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
11 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 #    GNU General Public License for more details.
13 #
14 #    You should have received a copy of the GNU General Public License
15 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
16 #
17 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
18
19 from nepi.execution.resource import clsinit_copy
20 from nepi.resources.ns3.ns3base import NS3Base
21
22 @clsinit_copy
23 class NS3BaseErrorModel(NS3Base):
24     _rtype = "abstract::ns3::ErrorModel"
25
26     @property
27     def device(self):
28         from nepi.resources.ns3.ns3netdevice import NS3BaseNetDevice
29         devices = self.get_connected(NS3BaseNetDevice.get_rtype())
30
31         if not devices: 
32             msg = "ErrorModel not connected to device"
33             self.error(msg)
34             raise RuntimeError(msg)
35
36         return devices[0]
37
38     @property
39     def _rms_to_wait(self):
40         rms = set()
41         rms.add(self.device)
42         return rms
43
44     def _connect_object(self):
45         device = self.device
46         if device.uuid not in self.connected:
47             self.simulation.invoke(device.uuid, "SetReceiveErrorModel", self.uuid)
48             self._connected.add(device.uuid)
49