print_colors.py

def color_wrap(msg, color='blue'):
    end_color = '\033[0m'
    color_dict = {
        'blue': '\033[94m',
        'green': '\033[92m',
        'yellow': '\033[93m',
        'red': '\033[91m',
        'purple':'\033[95m'
    }
    color = color_dict.get(color)
    return color + msg + end_color
    
  print(color_wrap("hello!", color='green'))