diff --git a/archivist/utils/strings.py b/archivist/utils/strings.py index 6d84e9f..408e7af 100644 --- a/archivist/utils/strings.py +++ b/archivist/utils/strings.py @@ -1,6 +1,5 @@ - -import re +import math, re def get_urls(string:str = None) -> list: """ @@ -19,3 +18,20 @@ def get_urls(string:str = None) -> list: return [x[0] for x in re.findall(regex, string)] +def convert_size(size_bytes: int) -> str: + """ + A function that converts the given size in bytes to a human-readable format. + + Parameters: + size_bytes (int): An integer representing the size in bytes to be converted. + + Returns: + A string representing the converted size with the appropriate unit (B, KiB, MiB, GiB, etc.). + """ + if size_bytes == 0: + return "0B" + size_name = ("B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB", "YiB") + i = int(math.floor(math.log(size_bytes, 1024))) + p = math.pow(1024, i) + s = round(size_bytes / p, 2) + return f"{s} {size_name[i]}"