From: Nikos Skalkotos <skalkoto@grnet.gr>
Date: Tue, 28 Feb 2012 16:26:26 +0000 (+0200)
Subject: Make DiskDevice.shrink() return the new disk size
X-Git-Tag: v0.1~151
X-Git-Url: https://code.grnet.gr/git/snf-image-creator/commitdiff_plain/e3aac3f935c62d3c7433deaaa9e10dff9e724f7d

Make DiskDevice.shrink() return the new disk size
---

diff --git a/image_creator/disk.py b/image_creator/disk.py
index e186174..6f219bf 100644
--- a/image_creator/disk.py
+++ b/image_creator/disk.py
@@ -164,6 +164,8 @@ class DiskDevice(object):
         start = last_partition['part_start'] / sector_size
         end = start + (block_size * block_cnt) / sector_size - 1
 
+        return (end + 1) * sector_size
+
 
 
 # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :