Python 在多个模块中自定义记录器，无需传递额外的 arguments

Question

I have this custom Logger class which accepts 2 non-standard arguments test_name and start_time :

class S_Logger(Logger):

    def __init__(self,
             test_name: str,
             start_time: str,
             log_name: str = 'main',
             *args, **kwargs) -> None:
    
    Logger.__init__(self, log_name, *args, **kwargs)
    self.test_name = test_name
    self.setLevel(logging.DEBUG)
    self.addHandler(S_StreamHandler())
    self.addHandler(S_FileHandler(test_name, start_time))

@classmethod
def get_logger(cls) -> logging.Logger:
    # That doesn't work, I have to specify handlers one more time...
    return logging.getLogger('main')

The first time I use that logger is in module exampleA.py, but then I would like to reuse the same logger with those same handlers in another module (exampleB.py) WITHOUT passing those 2 additional arguments. Is this possible?

module exampleA.py:

from source import S_Logger

class exampleA(object):

    def run(self):
        log = S_Logger(self.test_name, self.start_time)
        log.info('module A')

module exampleB.py:

from source import S_Logger

logger = S_Logger.get_logger()

class exampleB(object):

    def something(self):
        logger.info('module B')

Example output that I would like to achieve:

2022-09-21 15:00:00 INFO  line:10 of exampleA.py   >> Module A
2022-09-21 15:00:00 INFO  line:15 of exampleB.py   >> Module B

Answer 1

Try to add default value as you do with non-standard argument log_name .

class S_Logger(Logger):

    def __init__(self,
             test_name: str = 'default name',
             start_time: str = 'default time',
             log_name: str = 'main',
             *args, **kwargs) -> None:
    
    Logger.__init__(self, log_name, *args, **kwargs)
    self.test_name = test_name
    self.setLevel(logging.DEBUG)
    self.addHandler(S_StreamHandler())
    self.addHandler(S_FileHandler(test_name, start_time))

@classmethod
def get_logger(cls) -> logging.Logger:
    # That doesn't work, I have to specify handlers one more time...
    return logging.getLogger('main')