From e3aac3f935c62d3c7433deaaa9e10dff9e724f7d Mon Sep 17 00:00:00 2001
From: Nikos Skalkotos <skalkoto@grnet.gr>
Date: Tue, 28 Feb 2012 18:26:26 +0200
Subject: [PATCH] Make DiskDevice.shrink() return the new disk size

---
 image_creator/disk.py | 2 ++
 1 file changed, 2 insertions(+)

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 :
-- 
GitLab