Update disk setup step to use context manager

Signed-off-by: brianherrera <briher@amazon.com>

scripts/build/bootstrap/incremental_build_util.py

@@ -325,24 +325,22 @@ def mount_volume_to_device(created):
         # Verify drive is in an offline state.
         # Some Windows configs will automatically set new drives as online causing diskpart setup script to fail.
         offline_drive()
-        
+
-        f = tempfile.NamedTemporaryFile(delete=False)
+        with tempfile.NamedTemporaryFile(delete=False) as f:
-        f.write("""
+            f.write("""
-      select disk 1
+            select disk 1
-      online disk
+            online disk
-      attribute disk clear readonly
+            attribute disk clear readonly
-      """.encode('utf-8'))  # assume disk # for now
+            """.encode('utf-8'))  # assume disk # for now
 
         if created:
             print('Creating filesystem on new volume')
             f.write("""create partition primary
-          select partition 1
+            select partition 1
-          format quick fs=ntfs
+            format quick fs=ntfs
-          assign
+            assign
-          active
+            active
-          """.encode('utf-8'))
+            """.encode('utf-8'))
-
-        f.close()
 
         subprocess.call(['diskpart', '/s', f.name])