filesize: more cleanups

python-humanize · Aug 19, 2023 · a0bb67a · a0bb67a
1 parent 4551ca0
commit a0bb67a
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/src/humanize/filesize.py b/src/humanize/filesize.py
@@ -89,15 +89,11 @@ def naturalsize(
     if abs_bytes == 1 and not gnu:
         return "%d Byte" % bytes_
 
-    if abs_bytes < base and not gnu:
-        return "%d Bytes" % bytes_
-
-    if abs_bytes < base and gnu:
-        return "%dB" % bytes_
-
-    for i, s in enumerate(suffix):
-        unit = base ** (i + 2)
+    if abs_bytes < base:
+        return ("%dB" if gnu else "%d Bytes") % bytes_
 
+    for i, s in enumerate(suffix, 2):
+        unit = base**i
         if abs_bytes < unit:
             break