From 260417bcd0461bf4b72ba0760579d8b705eca5a0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Th=C3=A9ophile=20Bastian?= Date: Mon, 9 Mar 2020 12:24:37 +0100 Subject: [PATCH] Network: handle object cleanup --- src/network.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) 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()