Ada banyak tanggapan. Tapi tidak ada yang berbicara tentang dekorator. Jadi ini milikku.
Karena itu jauh lebih sederhana.
Tidak perlu mengimpor apa pun, atau menulis subkelas apa pun:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import logging
NO_COLOR = "\33[m"
RED, GREEN, ORANGE, BLUE, PURPLE, LBLUE, GREY = \
map("\33[%dm".__mod__, range(31, 38))
logging.basicConfig(format="%(message)s", level=logging.DEBUG)
logger = logging.getLogger(__name__)
# the decorator to apply on the logger methods info, warn, ...
def add_color(logger_method, color):
def wrapper(message, *args, **kwargs):
return logger_method(
# the coloring is applied here.
color+message+NO_COLOR,
*args, **kwargs
)
return wrapper
for level, color in zip((
"info", "warn", "error", "debug"), (
GREEN, ORANGE, RED, BLUE
)):
setattr(logger, level, add_color(getattr(logger, level), color))
# this is displayed in red.
logger.error("Launching %s." % __file__)
Ini mengatur kesalahan dalam warna merah, men-debug pesan dengan warna biru, dan sebagainya. Seperti ditanyakan dalam pertanyaan.
Kami bahkan dapat mengadaptasi pembungkus untuk mengambil colorargumen agar secara dinamis mengatur warna pesan menggunakanlogger.debug("message", color=GREY)
EDIT: Jadi inilah dekorator yang disesuaikan untuk mengatur warna saat runtime:
def add_color(logger_method, _color):
def wrapper(message, *args, **kwargs):
color = kwargs.pop("color", _color)
if isinstance(color, int):
color = "\33[%dm" % color
return logger_method(
# the coloring is applied here.
color+message+NO_COLOR,
*args, **kwargs
)
return wrapper
# blah blah, apply the decorator...
# this is displayed in red.
logger.error("Launching %s." % __file__)
# this is displayed in blue
logger.error("Launching %s." % __file__, color=34)
# and this, in grey
logger.error("Launching %s." % __file__, color=GREY)
