Device = ""
File = ""
-class ColorFormatter(logging.Formatter):
+class DebugFormatter(logging.Formatter):
+ def format(self, record):
+ if record.levelno == logging.DEBUG:
+ record.debuginfo = "[{}:{}] ".format(os.path.basename(record.pathname), record.lineno)
+ else:
+ record.debuginfo = ''
+ return logging.Formatter.format(self, record)
+
+class ColorFormatter(DebugFormatter):
_levelToColor = {
logging.CRITICAL: "\x1b[35;1m",
logging.ERROR: "\x1b[33;1m",
def format(self, record):
record.levelcolor = self._levelToColor[record.levelno]
record.msg = record.msg
- return logging.Formatter.format(self, record)
+ return super().format(record)
class ColorStreamHandler(logging.StreamHandler):
def __init__(self, stream=None, format=None, datefmt=None, style='%', cformat=None):
self.formatter = ColorFormatter(cformat, datefmt, style)
self.terminator = "\x1b[0m\n"
else:
- self.formatter = logging.Formatter(format, datefmt, style)
+ self.formatter = DebugFormatter(format, datefmt, style)
class Partition:
def __init__(self, name, size, start=None, ptype=None, fstype="raw", bootable=False):
print(f" {k:6} {v.long_name}")
sys.exit(0)
- conh = ColorStreamHandler(format='%(asctime)s %(levelname)s:%(message)s',
- cformat='%(asctime)s: %(levelcolor)s%(message)s',
- datefmt='%Y-%m-%dT%H:%M:%S')
+ conh = ColorStreamHandler(format='%(asctime)s.%(msecs)s %(debuginfo)%(levelname)-8s %(message)s',
+ cformat='%(asctime)s.%(msecs)s %(debuginfo)s%(levelcolor)s%(message)s',
+ datefmt='%Y-%m-%dT%H:%M:%S')
log_handlers = [conh]
- logging.basicConfig(format='%(asctime)s %(levelname)s:%(message)s',
+ logging.basicConfig(format='%(asctime)s %(levelname)-8s %(message)s',
handlers=log_handlers,
level=args.log_level.upper())