Update disk setup step to use context manager

Signed-off-by: brianherrera <briher@amazon.com>
monroegm-disable-blank-issue-2
brianherrera 4 years ago
parent 56f0ea68a6
commit ac8201a2fa
No known key found for this signature in database
GPG Key ID: FC67BB5F533BAE04

@ -326,7 +326,7 @@ def mount_volume_to_device(created):
# 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("""
select disk 1
online disk
@ -342,8 +342,6 @@ def mount_volume_to_device(created):
active
""".encode('utf-8'))
f.close()
subprocess.call(['diskpart', '/s', f.name])
time.sleep(5)

Loading…
Cancel
Save