Implement AbstractSandbox as a context manager.

setuptools/sandbox.py

@@ -249,11 +249,9 @@ def run_setup(setup_script, args):
                 setup_script.encode(sys.getfilesystemencoding())
             )
 
-            def runner():
+            with DirectorySandbox(setup_dir):
                 ns = dict(__file__=dunder_file, __name__='__main__')
                 _execfile(setup_script, ns)
-
-            DirectorySandbox(setup_dir).run(runner)
         except SystemExit as v:
             if v.args and v.args[0]:
                 raise
@@ -275,22 +273,25 @@ class AbstractSandbox:
         for name in self._attrs:
             setattr(os, name, getattr(source, name))
 
-    def run(self, func):
-        """Run 'func' under os sandboxing"""
-        try:
+    def __enter__(self):
         self._copy(self)
         if _file:
             builtins.file = self._file
         builtins.open = self._open
         self._active = True
-            return func()
-        finally:
+
+    def __exit__(self, exc_type, exc_value, traceback):
         self._active = False
         if _file:
             builtins.file = _file
         builtins.open = _open
         self._copy(_os)
 
+    def run(self, func):
+        """Run 'func' under os sandboxing"""
+        with self:
+            return func()
+
     def _mk_dual_path_wrapper(name):
         original = getattr(_os, name)
 

setuptools/tests/test_sandbox.py

@@ -12,8 +12,8 @@ from setuptools.sandbox import DirectorySandbox
 
 class TestSandbox:
     def test_devnull(self, tmpdir):
-        sandbox = DirectorySandbox(str(tmpdir))
-        sandbox.run(self._file_writer(os.devnull))
+        with DirectorySandbox(str(tmpdir)):
+            self._file_writer(os.devnull)
 
     @staticmethod
     def _file_writer(path):