python为特定函数记录不同级别的日志

Question

我试图减少napalm库发送到syslog的日志记录量，但也允许从代码的其他部分发送info日志。我将logging.basicConfig设置为INFO，但我希望napalm函数为WARNING及以上

我有这样的代码：

from napalm import get_network_driver
import logging
import getpass

logging.basicConfig(
    filename="/var/log/myscripts/script.log", level=logging.INFO, format="%(asctime)s %(message)s")

def napalm(device):
    logging.getLogger().setLevel(logging.WARNING)
    username = getpass.getuser()
    driver = get_network_driver("junos")
    router = driver(str(device), username, "", password="", timeout=120)
    router.open()
    return router

router = napalm('myrouter')
config = "hostname foobar"
router.load_merge_candidate(config=config)
show = router.compare_config()
logging.info(show)

问题是logging.info输出从未进入日志文件。如果我做了logging.warning(show)，它会做，但我希望这是info。我希望函数是WARNING的原因是它在info级别上生成了太多的其他日志记录，这只是噪声。所以尽量减少这一点

Answer 1

这是本书中的一个很好的技巧。看看这是否有助于你的处境

def napalm(count):
    for x in range(count):
        logger.info('useless log line')

@contextmanager
def debug_logging(level):
    logger = logging.getLogger()
    old_level = logger.getEffectiveLevel()
    logger.setLevel(level)
    try:
        yield
    finally:
        logger.setLevel(old_level)

napalm(5)
with debug_logging(logging.WARNING):
    napalm(5)
napalm(5)

Answer 2

通过在没有参数的情况下调用logging.getLogger()，您当前正在检索根记录器，并且覆盖它的级别也会影响所有其他记录器。 相反，您应该检索库的记录器，并仅覆盖该特定记录器的级别

napalm库在其__init__.py中执行以下操作：

logger = logging.getLogger("napalm")

即，库记录器的名称为“凝固汽油弹”。 因此，您应该能够通过在脚本中放入以下行来覆盖特定记录器的级别：

logging.getLogger("napalm").setLevel(logging.WARNING)

---

一般示例：

import logging

logging.basicConfig(level=logging.DEBUG, format="%(levelname)s: %(message)s")
A = logging.getLogger("A")
B = logging.getLogger("B")

A.debug("#1 from A gets printed")
B.debug("#1 from B gets printed")

logging.getLogger("A").setLevel(logging.CRITICAL)

A.debug("#2 from A doesn't get printed")  # because we've increased the level
B.debug("#2 from B gets printed")

输出：

DEBUG: #1 from A gets printed
DEBUG: #1 from B gets printed
DEBUG: #2 from B gets printed

---

编辑：

因为这对您来说不太好，可能是因为这个库中有很多其他的记录器：

$ grep -R 'getLogger' .
napalm/junos/junos.py:log = logging.getLogger(__file__)
napalm/base/helpers.py:logger = logging.getLogger(__name__)
napalm/base/clitools/cl_napalm.py:logger = logging.getLogger("napalm")
napalm/base/clitools/cl_napalm_validate.py:logger = logging.getLogger("cl_napalm_validate.py")
napalm/base/clitools/cl_napalm_test.py:logger = logging.getLogger("cl_napalm_test.py")
napalm/base/clitools/cl_napalm_configure.py:logger = logging.getLogger("cl-napalm-config.py")
napalm/__init__.py:logger = logging.getLogger("napalm")
napalm/pyIOSXR/iosxr.py:logger = logging.getLogger(__name__)
napalm/iosxr/iosxr.py:logger = logging.getLogger(__name__)

然后我会求助于这样的事情（相关：How to list all existing loggers using python.logging module）：

# increase level for all loggers
for name, logger in logging.root.manager.loggerDict.items():
    logger.setLevel(logging.WARNING)

或者，如果您只需打印各种记录器，识别出最吵闹的记录器，并逐一使其静音，就足够了