Gallery-Archivist/archivist/utils/strings.py

import math, re

def get_urls(string:str = None) -> list:
    """
    Extracts URLs from a given string using regular expressions.

    Parameters:
    string (str): The input string from which URLs need to be extracted.

    Returns:
    list: A list of URLs extracted from the input string.
    """
    regex = re.compile(
        r'((([A-Za-z]{3,9}:(?:\/\/)?)(?:[-;:&=\+\$,\w]+@)?[A-Za-z0-9.-]+|(?:www.|[-;:&=\+\$,\w]+@)[A-Za-z0-9.-]+)((?:\/[\+~%\/.\w_]*)?\??(?:[-\+=&;%@.\w_]*)#?(?:[.\!\/\\w]*))?)',
        re.IGNORECASE)

    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]}"