From: Nikos Skalkotos <skalkoto@grnet.gr>
Date: Fri, 28 Dec 2012 01:16:43 +0000 (+0200)
Subject: Remove temporary bundeled image during cleanup
X-Git-Tag: v0.2~26
X-Git-Url: https://code.grnet.gr/git/snf-image-creator/commitdiff_plain/69a4f8947311cfb39373524b0b740ec57e65f640

Remove temporary bundeled image during cleanup
---

diff --git a/image_creator/disk.py b/image_creator/disk.py
index b662c07..a8d5d66 100644
--- a/image_creator/disk.py
+++ b/image_creator/disk.py
@@ -82,7 +82,9 @@ class Disk(object):
     def _dir_to_disk(self):
         if self.source == '/':
             bundle = BundleVolume(self.out, self.meta)
-            return self._losetup(bundle.create_image())
+            image = bundle.create_image()
+            self._add_cleanup(os.unlink, image)
+            return self._losetup(image)
         raise FatalError("Using a directory as media source is supported")
 
     def cleanup(self):