Source code for extorch.utils.logger

# -*- coding:utf-8 -*-

import os
import sys
import logging

__all__ = ["logger", "getLogger"]


LEVEL = "info"
LEVEL = getattr(logging, LEVEL.upper())
LOG_FORMAT = "%(asctime)s %(name)-16s %(levelname)7s: %(message)s"


logging.basicConfig(stream = sys.stdout, level = LEVEL, format = LOG_FORMAT, datefmt = "%m/%d %I:%M:%S %p")
logger = logging.getLogger()


def addFile(self, filename: str) -> None:
    handler = logging.FileHandler(filename)
    handler.setFormatter(logging.Formatter(LOG_FORMAT))
    self.addHandler(handler)


logging.Logger.addFile = addFile


[docs]def getLogger(name: str) -> logging.Logger:
    return logger.getChild(name)