Python - Message avec code couleur

Sommaire

On a toujours besoin rapidement d’un peu de couleur pour visualiser le résultat de ces scripts. Ci dessous une fonction avec un code couleur selon le type de message.

Fonctions
#

Mise en forme textuel avec couleur - Type bannière
#

def format_message(level, message):
    if level == "INFO":
        return f"[\033[1;34mINFO\033[0m] {message}"
    elif level == "WARN":
        return f"[\033[1;33mWARN\033[0m] {message}"
    elif level == "ERROR" or level == "FAIL":
        return f"[\033[1;31mFAIL\033[0m] {message}"
    elif level == "OK":
        return f"[\033[1;32m OK \033[0m] {message}"
    elif level == "KO":
        return f"[\033[1;31m KO \033[0m] {message}"
    else:
        return message

Cas d’utilisation
#

Mini script de résolution DNS avec retour de message sur la réussite ou non de l’opération

import os


MyFile = "agent_secret.txt"

# Vérification de l'existence du fichier
if not os.path.isfile(MyFile):
    print(format_message("FAIL", "Fichier invalide ou introuvable: {}".format(MyFile)))
else:
    print(format_message("OK", "Fichier disponible: {}".format(MyFile)))