From efe1ac1a31a6c1bc3b0f16a24c58c8262d43ea0d Mon Sep 17 00:00:00 2001 From: Sylvain Corlay Date: Mon, 4 Apr 2016 00:13:31 -0400 Subject: [PATCH] Use context manager --- setup.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/setup.py b/setup.py index ebb3f1c..a466e26 100644 --- a/setup.py +++ b/setup.py @@ -17,11 +17,8 @@ def has_flag(compiler, flagname): """ import tempfile fd, fname = tempfile.mkstemp('.cpp', 'main', text=True) - f = os.fdopen(fd, 'w') - try: + with os.fdopen(fd, 'w') as f: f.write('int main (int argc, char **argv) { return 0; }') - finally: - f.close() try: compiler.compile([fname], extra_postargs=[flagname]) except setuptools.distutils.errors.CompileError: