Python - 在调用它的模块中的函数内部使用记录器

Question

假设我有这样的代码:

模块A.py:

import logging
logger = logging.getLogger('A')

def utility_func():
    logger.info('hi')
print utility_func()

模块B.py:

import logging
logger = logging.getLogger('B')
from ModuleA import utility_func

print utility_func()

当从 ModuleA.py 中调用 utility_func 时，我希望它使用“A”记录器，而当它从 ModuleB.py 中调用时，我希望它使用“B”记录器。

有没有办法做到这一点？或者更好的设置方式？

更新:

关于将 utility_func 更改为:

def utility_func():
    logging.info('hi')

这会冒泡到调用代码正在使用的日志吗？

Answer 1

这就是我最终的结果。但我仍然很想知道是否有更优雅的方式。

模块B.py:

import logging
import ModuleA

logger = logging.getLogger('B')
ModuleA.logger = logger

print ModuleA.utility_func()