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

Play with human size some more

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

20
kittens/tui/utils.py
View File

@ -4,6 +4,7 @@
import sys
from contextlib import suppress
from typing import Tuple
from .operations import raw_mode, set_cursor_visible
@ -29,15 +30,18 @@ def get_key_press(allowed: str, default: str) -> str:
return response
def human_size(size: int, sep: str = ' ', max_num_of_decimals: int = 2) -> str:
def human_size(
size: int, sep: str = ' ',
max_num_of_decimals: int = 2,
unit_list: Tuple[str, ...] = ('B', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB')
) -> 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')):
if size < (1 << ((i + 1) * 10)):
divisor, suffix = (1 << (i * 10)), candidate
break
ans = str(size / divisor)
if size < 2:
return f'{size}{sep}{unit_list[0]}'
from math import log
exponent = min(int(log(size, 1024)), len(unit_list) - 1)
ans = str(size / 1024**exponent)
pos = ans.find('.')
if pos > -1:
ans = ans[:pos + max_num_of_decimals + 1]
return ans.rstrip('0').rstrip('.') + sep + suffix
return ans.rstrip('0').rstrip('.') + sep + unit_list[exponent]