Make DiskDevice.shrink() return the new disk size

[snf-image-creator] / image_creator / disk.py

diff --git a/image_creator/disk.py b/image_creator/disk.py
index e186174..6f219bf 100644 (file)
--- 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 :