diff --git a/tubesync/sync/templatetags/filters.py b/tubesync/sync/templatetags/filters.py index 54f72d98..54f17ed1 100644 --- a/tubesync/sync/templatetags/filters.py +++ b/tubesync/sync/templatetags/filters.py @@ -1,10 +1,20 @@ from django import template -from django.template.defaultfilters import filesizeformat +from django.template.defaultfilters import filesizeformat, stringfilter + register = template.Library() + +@register.filter(is_safe=True) +@stringfilter +def fixB(input): + return input.replace('B', 'iB') + + @register.filter(is_safe=True) def bytesformat(input): - intermediate = filesizeformat(input) - return intermediate.replace('B', 'iB') + output = filesizeformat(input) + if not (output and output.endswith('B')): + return output + return fixB(output)