diff --git a/src/network.py b/src/network.py index 171a83d..493066b 100644 --- a/src/network.py +++ b/src/network.py @@ -62,3 +62,17 @@ class Network(util.NumberedClass): ) self.lxc_network = self.conn.networkCreateXML(xml) + + def cleanup(self): + if self.lxc_network: + self.lxc_network.destroy() + + def __enter__(self): + self.create() + return self + + def __exit__(self, exc_type, exc_val, exc_tb): + self.cleanup() + + def __del__(self): + self.cleanup()