def print_info(message, *args):
"""Print an info message to the terminal """
print_type = "-I-"
if len(args) > 0:
for arg in args:
message += arg + ", "
color_message = None
if sys.stdout.isatty():
msg_upper = message.upper()
if ":PASS" in msg_upper:
color_message = message[:msg_upper.index(":PASS") + 1] + "\033[1;32m" + "PASS" +\
"\033[0m" + message[msg_upper.index(":PASS") + 5:]
elif ":FAIL" in msg_upper:
color_message = message[:msg_upper.index(":FAIL") + 1] + "\033[1;31m" + "FAIL" +\
"\033[0m" + message[msg_upper.index(":FAIL") + 5:]
elif ":EXCEPTION" in msg_upper:
color_message = message[:msg_upper.index(":EXCEPTION") + 1] + "\033[1;31m" + "EXCEPTION" +\
"\033[0m" + message[msg_upper.index(":EXCEPTION") + 10:]
elif ":ERROR" in msg_upper:
color_message = message[:msg_upper.index(":ERROR") + 1] + "\033[1;31m" + "ERROR" +\
"\033[0m" + message[msg_upper.index(":ERROR") + 6:]
elif ":SKIPPED" in msg_upper:
color_message = message[:msg_upper.index(":SKIPPED") + 1] + "\033[1;33m" + "SKIPPED" +\
"\033[0m" + message[msg_upper.index(":SKIPPED") + 9:]
print_main(message, print_type, color_message)
return message
def print_debug(message, *args):
"""Print a debug message to the terminal """
print_type = "-D-"
if len(args) > 0:
for arg in args:
message += arg + ", "
print_main(message, print_type)
return message
def print_without_logging(message, *args):
"""Prints without writing to log file"""
print_type = "-N-"
if len(args) > 0:
for arg in args:
message += arg + ", "
print_main(message, print_type, logging=False)
return message
def print_normal(message, *args):
"""Prints with out print type(-I-,-E-),with color cyan in bold TEXT """
print_type = ""
if len(args) > 0:
for arg in args:
message += arg + ", "
print_main(message, print_type)
return message
def print_notype(message, *args):
"""Prints with out print type(-I-,-E-),with color cyan in bold TEXT """
print_type = ""
if len(args) > 0:
for arg in args:
message += arg + ", "
color_message = None
if sys.stdout.isatty():
color_message = "\033[0;34m" + str(message) + "\033[0m "
print_main(message, print_type, color_message)
return message
def print_error(message, *args):
"""Prints an error message to the terminal """
print_type = "-E-"
if len(args) > 0:
for arg in args:
message += arg + ", "
color_message = None
if sys.stdout.isatty():
print_type = "\033[1;31m" + "-E-" + "\033[0m"
color_message = "\033[1;31m" + str(message) + "\033[0m"
print_main(message, print_type, color_message)
return message
def print_sub(message, *args):
"""Substitutes the string with *args tuple provided
User has to provide the place holder for substitution in the message
as {0}, {1}, {2} ..etc and the corresponding values in the *args.
Eg: print_sub("My name is {0} {1}, I live in {2}", 'John', 'Doe', 'Texas')
Output: My Name is John Doe, I live in Texas
"""
print_type = "-I-"
message = message.format(*args)
print_main(message, print_type)
return message
