Source code for roocs_utils.utils.common

import re

from dask.utils import byte_sizes


def parse_size(size):
    """

    Parse size string into number of bytes.

    :param size: (str) size to parse in any unit
    :return: (int) number of bytes
    """
    n, suffix = re.match(r"^(\d+\.?\d*)([a-zA-Z]+)$", size).groups()

    try:

        multiplier = byte_sizes[suffix.lower()]

        size_in_bytes = multiplier * float(n)
    except KeyError as e:
        raise ValueError(f"Could not interpret '{suffix}' as a byte unit")

    return size_in_bytes