import functools


def with_color(color_code: str):
    """Coloring decorator

    Arguments:
        color_code {str} -- e.g.: '\033[91m'
    """
    def wrapper(func):
        @functools.wraps(func)
        def inner(args):
            result = func(f'{color_code}{args}\033[0m')
            return result
        return inner
    return wrapper