From de476aac46f685551983b4516289b1f5758945b1 Mon Sep 17 00:00:00 2001 From: Ricter Z Date: Wed, 19 Oct 2016 13:00:59 +0800 Subject: [PATCH] qwq --- nhentai/logger.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/nhentai/logger.py b/nhentai/logger.py index ccd68ce..9a1fb19 100644 --- a/nhentai/logger.py +++ b/nhentai/logger.py @@ -49,6 +49,28 @@ class ColorizingStreamHandler(logging.StreamHandler): isatty = getattr(self.stream, 'isatty', None) return isatty and isatty() and not self.disable_coloring + def emit(self, record): + try: + message = self.format(record) + stream = self.stream + + if not self.is_tty: + if message and message[0] == "\r": + message = message[1:] + stream.write(message) + else: + self.output_colorized(message) + stream.write(getattr(self, 'terminator', '\n')) + + self.flush() + except (KeyboardInterrupt, SystemExit): + raise + except IOError: + pass + except: + self.handleError(record) + + if not platform.system() == 'Windows': def output_colorized(self, message): self.stream.write(message)