Use with clause to handle files.

1 files changed, 7 insertions, 3 deletions

diff --git a/utils/update_build_version.py b/utils/update_build_version.py
index fd600a1..b1a4b0b 100755
--- a/utils/update_build_version.py
+++ b/utils/update_build_version.py
@@ -110,9 +110,13 @@ def main():
         '"{}\\n"\n'.format(get_version_string(p, d))
         for (p, d) in zip(projects, sys.argv[1:])
     ])
-    if os.path.isfile(OUTFILE) and new_content == open(OUTFILE, 'r').read():
-        sys.exit(0)
-    open(OUTFILE, 'w').write(new_content)
+
+    if os.path.isfile(OUTFILE):
+        with open(OUTFILE, 'r') as f:
+            if new_content == f.read():
+                return
+    with open(OUTFILE, 'w') as f:
+        f.write(new_content)
 
 
 if __name__ == '__main__':