There was an error while loading. Please reload this page.
1 parent 3d5a360 commit 374ef75Copy full SHA for 374ef75
2 files changed
pre_commit/logging_handler.py
@@ -1,6 +1,5 @@
1
-from __future__ import print_function
2
-
3
import logging
+import sys
4
5
from pre_commit import color
6
@@ -14,14 +13,14 @@
14
13
15
16
class LoggingHandler(logging.Handler):
17
- def __init__(self, use_color, print_fn=print):
+ def __init__(self, use_color, write=sys.stdout.write):
18
logging.Handler.__init__(self)
19
self.use_color = use_color
20
- self.__print_fn = print_fn
+ self.__write = write
21
22
def emit(self, record):
23
- self.__print_fn(
24
- u'{0}{1}'.format(
+ self.__write(
+ u'{0}{1}\n'.format(
25
color.format_color(
26
'[{0}]'.format(record.levelname),
27
LOG_LEVEL_COLORS[record.levelname],
tests/logging_handler_test.py
@@ -19,7 +19,7 @@ def test_logging_handler_color():
handler = LoggingHandler(True, print_mock)
handler.emit(FakeLogRecord('hi', 'WARNING', 30))
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():
28
handler = LoggingHandler(False, print_mock)
29
30
31
- '[WARNING] hi',
+ '[WARNING] hi\n',
32
0 commit comments