misc: Fix format_size from running out of sizes

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/buildservice/misc.py b/src/buildservice/misc.py
index 98dcdeb5d9fac341cd9dde414cb0589dc066b6fa..dec4dbc562aff2f35cfdb23de6cbf7eaa32c6507 100644 (file)
--- a/src/buildservice/misc.py
+++ b/src/buildservice/misc.py
@@ -46,14 +46,13 @@ def normalize(*args, iteration=1):
        return "-".join(s.split())
 
 def format_size(s):
-       units = ("B", "KiB", "MiB", "GiB", "TiB")
+       units = ("B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB")
 
-       i = 0
-       while s >= 1024 and i < len(units):
-               s /= 1024
-               i += 1
+       for unit in units:
+               if s < 1024:
+                       return "%d %s" % (round(s), unit)
 
-       return "%d %s" % (round(s), units[i])
+               s /= 1024
 
 def group(items, key):
        """