added: missing units (fixes #11)

Schluggi · Sep 9, 2021 · 40b0675 · 40b0675
1 parent c01c63d
commit 40b0675
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils.py b/utils.py
@@ -15,9 +15,9 @@ def imaperror_decode(e):
 
 
 def beautysized(b, factor=1000, precision=1):
-    units = ['B', 'KB', 'MB', 'GB']
+    units = ['B', 'KB', 'MB', 'GB', 'TB', 'PB']
     if factor == 1024:
-        units = ['B', 'KiB', 'MiB', 'GiB']
+        units = ['B', 'KiB', 'MiB', 'GiB', 'TiB', 'PiB']
     template = '{:.?f} {}'.replace('?', str(precision))
     rv = 0