diff --git a/image_creator/util.py b/image_creator/util.py index bb57b5fe0dc90fe14ecdd62791867a89f877db0f..a0d44ba8fb91e4aa9a154d3496243f6e22b701d8 100644 --- a/image_creator/util.py +++ b/image_creator/util.py @@ -49,16 +49,16 @@ def get_command(command): return find_sbin_command(command, e) -def error(msg): - puts_err(colored.red("Error: %s\n" % msg)) +def error(msg, new_line=True): + puts_err(colored.red("Error: %s\n" % msg), new_line) -def warn(msg): - puts_err(colored.yellow("Warning: %s" % msg)) +def warn(msg, new_line=True): + puts_err(colored.yellow("Warning: %s" % msg), new_line) -def success(msg): - puts(colored.green(msg)) +def success(msg, new_line=True): + puts(colored.green(msg), new_line) def progress_generator(label='', n=100):