_context.py: Make Context class a Python context manager

This will allow reliable cleanup.

1 files changed, 14 insertions, 0 deletions

diff --git a/src/buildstream/_context.py b/src/buildstream/_context.py
index 286ed6d3d..1e45d776d 100644
--- a/src/buildstream/_context.py
+++ b/src/buildstream/_context.py
@@ -164,6 +164,20 @@ class Context():
         self._cascache = None
         self._casquota = None
 
+    # __enter__()
+    #
+    # Called when entering the with-statement context.
+    #
+    def __enter__(self):
+        return self
+
+    # __exit__()
+    #
+    # Called when exiting the with-statement context.
+    #
+    def __exit__(self, exc_type, exc_value, traceback):
+        return None
+
     # load()
     #
     # Loads the configuration files