web-dev-qa-db-ja.com

pythonロギングモジュールで現在のログレベルを取得する方法

外部パッケージなしで色付けされる基本的なロガーを作成しようとしていますが、

_# these have to be the first functions so I can use it in the logger settings
def create_log_name(log_path="{}/log", filename="zeus-log-{}.log"):
    if not os.path.exists(log_path.format(os.getcwd())):
        os.mkdir(log_path.format(os.getcwd()))
    find_file_amount = len(os.listdir(log_path.format(os.getcwd())))
    full_log_path = "{}/{}".format(log_path.format(os.getcwd()), filename.format(find_file_amount + 1))
    return full_log_path


def set_color_value(levelname):
    log_set = {
        "INFO": "\033[92m{}\033[0m",
        "WARNING": "\033[93m{}\033[0m",
        "DEBUG": "\033[94m{}\033[0m",
        "ERROR": "\033[91m{}\033[0m",
        "CRITICAL": "\033[91m{}\033[0m"
    }
    return log_set[levelname].format(levelname)

logger = logging.getLogger("zeus-log")
logger.setLevel(logging.DEBUG)
file_handler = logging.FileHandler(
    filename=create_log_name(), mode="a+"
)
file_handler.setLevel(logging.DEBUG)
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.DEBUG)
file_format = logging.Formatter(
    '%(asctime)s;%(name)s;%(levelname)s;%(message)s'
)
console_format = logging.Formatter(
    "[%(asctime)s {}] %(message)s".format(set_color_value()), "%H:%M:%S"
)
file_handler.setFormatter(file_format)
console_handler.setFormatter(console_format)
logger.addHandler(console_handler)
logger.addHandler(file_handler)
_

したがって、現時点では、_logging.Formatter_に設定される現在のログレベルを取得して、それを私の小さな関数に送信するだけです。

_console_format = logging.Formatter(
    "[%(asctime)s {}] %(message)s".format(set_color_value()), "%H:%M:%S"
)
_

ロギングパッケージから現在のログレベルを取得することは可能ですか?


たとえば、logger.INFO("test")を渡したとしましょう。そのINFO部分を文字列として取得する方法が必要です。そこから、set_color_value("INFO")は次を返す必要があります。

enter image description here

5
wahwahwah

私はこれを別の方法で行い、レベル番号を使用して文字列自体に色を追加することにしました。

def set_color(org_string, level=None):
    color_levels = {
        10: "\033[36m{}\033[0m",       # DEBUG
        20: "\033[32m{}\033[0m",       # INFO
        30: "\033[33m{}\033[0m",       # WARNING
        40: "\033[31m{}\033[0m",       # ERROR
        50: "\033[7;31;31m{}\033[0m"   # FATAL/CRITICAL/EXCEPTION
    }
    if level is None:
        return color_levels[20].format(org_string)
    else:
        return color_levels[int(level)].format(org_string)

したがって、たとえば:

logger.info(set_color("test"))
logger.debug(set_color("test", level=10))
logger.warning(set_color("test", level=30))
logger.error(set_color("test", level=40))
logger.fatal(set_color("test", level=50))

出力します:

enter image description here

5
wahwahwah

はい、次の方法でロガーレベルを確認できます

level = logger.level
9

ルートロガーを使用している場合、たとえばlogging.basicConfig()を呼び出したため、次を使用できます。

import logging
logging.root.level

例えば

if logging.DEBUG >= logging.root.level:
    # Do something
6
bcattle