Skip to content

Commit

Permalink
Fix logger for Python3
Browse files Browse the repository at this point in the history
  • Loading branch information
wkentaro committed May 16, 2023
1 parent 0bacc48 commit d987c44
Showing 1 changed file with 8 additions and 18 deletions.
26 changes: 8 additions & 18 deletions labelme/logger.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import datetime
import logging
import os
import sys

import termcolor

Expand Down Expand Up @@ -49,23 +50,12 @@ def colored(text):
return logging.Formatter.format(self, record)


class ColoredLogger(logging.Logger):

FORMAT = (
"[%(levelname2)s] %(module2)s:%(funcName2)s:%(lineno2)s - %(message2)s"
)

def __init__(self, name):
logging.Logger.__init__(self, name, logging.INFO)

color_formatter = ColoredFormatter(self.FORMAT)

console = logging.StreamHandler()
console.setFormatter(color_formatter)

self.addHandler(console)
return
logger = logging.getLogger(__appname__)

stream_handler = logging.StreamHandler(sys.stderr)
handler_format = ColoredFormatter(
"[%(levelname2)s] %(module2)s:%(funcName2)s:%(lineno2)s - %(message2)s"
)
stream_handler.setFormatter(handler_format)

logger = logging.getLogger(__appname__)
logger.__class__ = ColoredLogger
logger.addHandler(stream_handler)

0 comments on commit d987c44

Please sign in to comment.