cli.py: Use Context as context manager

1 files changed, 23 insertions, 23 deletions

diff --git a/src/buildstream/_frontend/cli.py b/src/buildstream/_frontend/cli.py
index 8fd834825..d4686e4fc 100644
--- a/src/buildstream/_frontend/cli.py
+++ b/src/buildstream/_frontend/cli.py
@@ -117,30 +117,30 @@ def complete_target(args, incomplete):
 
 def complete_artifact(orig_args, args, incomplete):
     from .._context import Context
-    ctx = Context()
-
-    config = None
-    if orig_args:
-        for i, arg in enumerate(orig_args):
-            if arg in ('-c', '--config'):
-                try:
-                    config = orig_args[i + 1]
-                except IndexError:
-                    pass
-    if args:
-        for i, arg in enumerate(args):
-            if arg in ('-c', '--config'):
-                try:
-                    config = args[i + 1]
-                except IndexError:
-                    pass
-    ctx.load(config)
-
-    # element targets are valid artifact names
-    complete_list = complete_target(args, incomplete)
-    complete_list.extend(ref for ref in ctx.artifactcache.list_artifacts() if ref.startswith(incomplete))
+    with Context() as ctx:
 
-    return complete_list
+        config = None
+        if orig_args:
+            for i, arg in enumerate(orig_args):
+                if arg in ('-c', '--config'):
+                    try:
+                        config = orig_args[i + 1]
+                    except IndexError:
+                        pass
+        if args:
+            for i, arg in enumerate(args):
+                if arg in ('-c', '--config'):
+                    try:
+                        config = args[i + 1]
+                    except IndexError:
+                        pass
+        ctx.load(config)
+
+        # element targets are valid artifact names
+        complete_list = complete_target(args, incomplete)
+        complete_list.extend(ref for ref in ctx.artifactcache.list_artifacts() if ref.startswith(incomplete))
+
+        return complete_list
 
 
 def override_completions(orig_args, cmd, cmd_param, args, incomplete):