Ticket #14002: bigger-filesizeformat.2.diff

django/template/defaultfilters.py

@@ -803 +803 @@
     except (TypeError,ValueError,UnicodeDecodeError):
         return u"0 bytes"
 
-    if bytes < 1024:
+    BYTE_UNITS = (
+        ('KB', 1024),
+        ('MB', 1024 * 1024),
+        ('GB', 1024 * 1024 * 1024),
+        ('TB', 1024 * 1024 * 1024 * 1024),
+        ('PB', 1024 * 1024 * 1024 * 1024 * 1024)
+    )
+
+    if bytes < BYTE_UNITS[0][1]:
         return ungettext("%(size)d byte", "%(size)d bytes", bytes) % {'size': bytes}
-    if bytes < 1024 * 1024:
-        return ugettext("%.1f KB") % (bytes / 1024)
-    if bytes < 1024 * 1024 * 1024:
-        return ugettext("%.1f MB") % (bytes / (1024 * 1024))
-    return ugettext("%.1f GB") % (bytes / (1024 * 1024 * 1024))
+    for index, (unit, unit_size) in enumerate(BYTE_UNITS):
+        if bytes < unit_size * 1024 or index == len(BYTE_UNITS) - 1:
+            return ugettext("%.1f %s") % (bytes / unit_size, unit)
 filesizeformat.is_safe = True
 
 def pluralize(value, arg=u's'):

tests/regressiontests/defaultfilters/tests.py

@@ -476 +476 @@
 >>> filesizeformat(1024*1024*1024)
 u'1.0 GB'
 
+>>> filesizeformat(1024*1024*1024*1024)
+u'1.0 TB'
+
+>>> filesizeformat(1024*1024*1024*1024*1024)
+u'1.0 PB'
+
+>>> filesizeformat(1024*1024*1024*1024*1024*2000)
+u'2000.0 PB'
+
 >>> filesizeformat(complex(1,-1))
 u'0 bytes'