summaryrefslogtreecommitdiff
path: root/rawdisk.write
diff options
context:
space:
mode:
authorDan Firth <dan.firth@codethink.co.uk>2013-10-10 16:07:37 +0000
committerDan Firth <dan.firth@codethink.co.uk>2013-10-14 12:00:27 +0000
commit1e57089d8aa73ffc49ef80622206d43e1bd15bf0 (patch)
tree493c8ec041a89afbb713869d9c89563e37745b5e /rawdisk.write
parentdc3d9cd1de7fcc4d0fed2ff4d958c73817415d9c (diff)
downloaddefinitions-1e57089d8aa73ffc49ef80622206d43e1bd15bf0.tar.gz
Deployment failures will now remove the disk image
Diffstat (limited to 'rawdisk.write')
-rwxr-xr-xrawdisk.write10
1 files changed, 8 insertions, 2 deletions
diff --git a/rawdisk.write b/rawdisk.write
index a74d6905..8723ac0c 100755
--- a/rawdisk.write
+++ b/rawdisk.write
@@ -46,8 +46,14 @@ class RawDiskWriteExtension(morphlib.writeexts.WriteExtension):
if os.path.isfile(location):
self.upgrade_local_system(location, temp_root)
else:
- self.create_local_system(temp_root, location)
- self.status(msg='Disk image has been created at %s' % location)
+ try:
+ self.create_local_system(temp_root, location)
+ self.status(msg='Disk image has been created at %s' % location)
+ except Exception:
+ os.remove(location)
+ self.status(msg='Failure to create disk image at %s' %
+ location)
+ raise
def upgrade_local_system(self, raw_disk, temp_root):
mp = self.mount(raw_disk)