KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make human_size() a little nicer

Browse Source
This commit is contained in:
Kovid Goyal 2021-09-12 13:28:07 +05:30
parent 3f3179db0c
commit 1136c64d49
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
kittens/tui/utils.py
View File

@ -29,7 +29,7 @@ def get_key_press(allowed: str, default: str) -> str:
return response
def human_size(size: int, sep: str = ' ') -> str:
def human_size(size: int, sep: str = ' ', max_num_of_decimals: int = 2) -> str:
""" Convert a size in bytes into a human readable form """
divisor, suffix = 1, "B"
for i, candidate in enumerate(('B', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB')):
@ -37,8 +37,7 @@ def human_size(size: int, sep: str = ' ') -> str:
divisor, suffix = (1 << (i * 10)), candidate
break
ans = str(size / divisor)
if ans.find(".") > -1:
ans = ans[:ans.find(".")+2]
if ans.endswith('.0'):
ans = ans[:-2]
return ans + sep + suffix
pos = ans.find('.')
if pos > -1:
ans = ans[:pos + max_num_of_decimals + 1]
return ans.rstrip('0').rstrip('.') + sep + suffix