pixano.utils.python

estimate_size(folder_path)

Estimate folder size and return it as a human-readable string

Parameters:

Name	Type	Description	Default
folder_path	Path	Folder path	required

Returns:

Type	Description
str	Folder size as a human-readable string

Source code in pixano/utils/python.py

def estimate_size(folder_path: Path) -> str:
    """Estimate folder size and return it as a human-readable string

    Args:
        folder_path (Path): Folder path

    Returns:
        str: Folder size as a human-readable string
    """

    # Estimate size
    total_size = 0
    for dirpath, _, filenames in os.walk(folder_path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            # skip if it is symbolic link
            if not os.path.islink(fp):
                total_size += os.path.getsize(fp)

    # Format size
    i = 0
    suffixes = ["B", "KB", "MB", "GB", "TB", "PB"]
    while total_size >= 1024 and i < len(suffixes) - 1:
        total_size /= 1024.0
        i += 1
    f = (f"{total_size:.2f}").rstrip("0").rstrip(".")
    readable_size = f"{f} {suffixes[i]}"

    return readable_size

natural_key(string)

Return key for string natural sort

Parameters:

Name	Type	Description	Default
string	str	Input string	required

Returns:

Type	Description
list	Sort key

Source code in pixano/utils/python.py

def natural_key(string: str) -> list:
    """Return key for string natural sort

    Args:
        string (str): Input string

    Returns:
        list: Sort key
    """

    return [int(s) if s.isdecimal() else s for s in re.split(r"(\d+)", string)]