third-party/pre-commit
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit.git synced 2026-04-15 01:51:46 +04:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Convert LoggingHandler to use sys.stdout.write

Browse source
This commit is contained in:
Anthony Sottile 2014-04-13 18:36:14 -07:00
parent 3d5a360df4
commit 374ef75835
2 changed files with 7 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pre_commit/logging_handler.py
View file

@ -1,6 +1,5 @@
from __future__ import print_function
import logging import logging
import sys
from pre_commit import color from pre_commit import color
@ -14,14 +13,14 @@ LOG_LEVEL_COLORS = {
class LoggingHandler(logging.Handler): class LoggingHandler(logging.Handler):
def __init__(self, use_color, print_fn=print): def __init__(self, use_color, write=sys.stdout.write):
logging.Handler.__init__(self) logging.Handler.__init__(self)
self.use_color = use_color self.use_color = use_color
self.__print_fn = print_fn self.__write = write
def emit(self, record): def emit(self, record):
self.__print_fn( self.__write(
u'{0}{1}'.format( u'{0}{1}\n'.format(
color.format_color( color.format_color(
'[{0}]'.format(record.levelname), '[{0}]'.format(record.levelname),
LOG_LEVEL_COLORS[record.levelname], LOG_LEVEL_COLORS[record.levelname],

4
tests/logging_handler_test.py
View file

@ -19,7 +19,7 @@ def test_logging_handler_color():
handler = LoggingHandler(True, print_mock) handler = LoggingHandler(True, print_mock)
handler.emit(FakeLogRecord('hi', 'WARNING', 30)) handler.emit(FakeLogRecord('hi', 'WARNING', 30))
print_mock.assert_called_once_with( print_mock.assert_called_once_with(
color.YELLOW + '[WARNING]' + color.NORMAL + ' hi', color.YELLOW + '[WARNING]' + color.NORMAL + ' hi\n',
) )
@ -28,5 +28,5 @@ def test_logging_handler_no_color():
handler = LoggingHandler(False, print_mock) handler = LoggingHandler(False, print_mock)
handler.emit(FakeLogRecord('hi', 'WARNING', 30)) handler.emit(FakeLogRecord('hi', 'WARNING', 30))
print_mock.assert_called_once_with( print_mock.assert_called_once_with(
'[WARNING] hi', '[WARNING] hi\n',
) )