mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 12:30:08 -06:00
35 lines
793 B
Python
35 lines
793 B
Python
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
import sys
|
|
|
|
from pre_commit import color
|
|
|
|
|
|
LOG_LEVEL_COLORS = {
|
|
'DEBUG': '',
|
|
'INFO': '',
|
|
'WARNING': color.YELLOW,
|
|
'ERROR': color.RED,
|
|
}
|
|
|
|
|
|
class LoggingHandler(logging.Handler):
|
|
def __init__(self, use_color, write=sys.stdout.write):
|
|
super(LoggingHandler, self).__init__()
|
|
self.use_color = use_color
|
|
self.__write = write
|
|
|
|
def emit(self, record):
|
|
self.__write(
|
|
'{}{}\n'.format(
|
|
color.format_color(
|
|
'[{}]'.format(record.levelname),
|
|
LOG_LEVEL_COLORS[record.levelname],
|
|
self.use_color,
|
|
) + ' ',
|
|
record.getMessage(),
|
|
)
|
|
)
|
|
sys.stdout.flush()
|