From 22cec876366721d70a9c0b5fcd33c4335e6f61b2 Mon Sep 17 00:00:00 2001 From: tcely Date: Fri, 24 Jan 2025 11:37:51 -0500 Subject: [PATCH] Add fixB filter --- tubesync/sync/templatetags/filters.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) 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)