82

--verbose如果我从命令行启用并且脚本中有以下内容,我想添加一个调试打印语句测试。

logger.info("test")

我经历了以下问题,但无法得到答案......

4

7 回答 7

140

我发现--verbose(对于用户)和--debug(对于开发人员)都很有用。以下是我使用loggingand的方法argparse

import argparse
import logging

parser = argparse.ArgumentParser()
parser.add_argument(
    '-d', '--debug',
    help="Print lots of debugging statements",
    action="store_const", dest="loglevel", const=logging.DEBUG,
    default=logging.WARNING,
)
parser.add_argument(
    '-v', '--verbose',
    help="Be verbose",
    action="store_const", dest="loglevel", const=logging.INFO,
)
args = parser.parse_args()    
logging.basicConfig(level=args.loglevel)

因此,如果--debug设置了,则日志记录级别设置为DEBUG. 如果--verbose,日志记录设置为INFO。如果两者都没有,则缺少--debug将日志记录级别设置为默认值WARNING

于 2013-12-18T16:11:21.847 回答
95

您需要将Argparse 教程的智慧与Python 的 Logging HOWTO结合起来。这是一个例子......

> cat verbose.py 
#!/usr/bin/env python

import argparse
import logging

parser = argparse.ArgumentParser(
    description='A test script for http://stackoverflow.com/q/14097061/78845'
)
parser.add_argument("-v", "--verbose", help="increase output verbosity",
                    action="store_true")

args = parser.parse_args()
if args.verbose:
    logging.basicConfig(level=logging.DEBUG)

logging.debug('Only shown in debug mode')

运行帮助:

> ./verbose.py -h
usage: verbose.py [-h] [-v]

A test script for http://stackoverflow.com/q/14097061/78845

optional arguments:
  -h, --help     show this help message and exit
  -v, --verbose  increase output verbosity

以详细模式运行:

> ./verbose.py -v
DEBUG:root:Only shown in debug mode

默默运行:

> ./verbose.py   
> 
于 2012-12-31T06:31:14.480 回答
68

这是一个更简洁的方法,它进行边界检查,并将在帮助中列出有效值:

parser = argparse.ArgumentParser(description='This is a demo.')
parser.add_argument("-l", "--log", dest="logLevel", choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], help="Set the logging level")

args = parser.parse_args()
if args.logLevel:
    logging.basicConfig(level=getattr(logging, args.logLevel))

用法:

demo.py --log DEBUG
于 2015-02-19T16:42:19.623 回答
32

另一种变体是计算数量-v并将计数用作 a 的索引,list实际水平来自logging

import argparse
import logging

parser = argparse.ArgumentParser()
parser.add_argument('-v', '--verbose', action='count', default=0)
args = parser.parse_args()

levels = [logging.WARNING, logging.INFO, logging.DEBUG]
level = levels[min(len(levels)-1,args.verbose)]  # capped to number of levels

logging.basicConfig(level=level,
                    format="%(asctime)s %(levelname)s %(message)s")

logging.debug("a debug message")
logging.info("a info message")
logging.warning("a warning message")

这适用于-vvvv, -vvv, -vv, -v, -v -v, 等,如果没有,如果提供更多,则-v选择然后logging.WARNING-vINFODEBUG

于 2015-12-03T12:12:04.590 回答
8

-v您可以在标志之后将级别明确指定为整数:

parser = argparse.ArgumentParser()
parser.add_argument("-v", "--verbose", const=1, default=0, type=int, nargs="?",
                    help="increase verbosity: 0 = only warnings, 1 = info, 2 = debug. No number means info. Default is no verbosity.")
args = parser.parse_args()

logger = logging.getLogger()
if args.verbose == 0:
    logger.setLevel(logging.WARN) 
elif args.verbose == 1:
    logger.setLevel(logging.INFO) 
elif args.verbose == 2:
    logger.setLevel(logging.DEBUG) 
于 2014-02-24T07:11:14.690 回答
2

如果您想为不想(或不能)编辑的脚本启用 logging.DEBUG 级别,您可以自定义启动:

jcomeau@aspire:~$ python -c "import site; site._script()"
[snip]...
USER_BASE: '/home/jcomeau/.local' (exists)
USER_SITE: '/home/jcomeau/.local/lib/python2.7/site-packages' (exists)
ENABLE_USER_SITE: True
jcomeau@aspire:~$ mkdir -p ~/.local/lib/python2.7/site-packages
jcomeau@aspire:~$ vi ~/.local/lib/python2.7/site-packages/usercustomize.py

输入以下内容:

import os, logging
if os.getenv('DEBUGGING'):
    logging.basicConfig(level = logging.DEBUG)

那么你可以:

jcomeau@aspire:~$ mkdir -p /tmp/some/random/
jcomeau@aspire:~$ echo 'import logging; logging.debug("test")' >> /tmp/some/random/script.py
jcomeau@aspire:~$ DEBUGGING=1 python /tmp/some/random/script.py 
DEBUG:root:test

来自 Paul Ollis 在http://nedbatchelder.com/blog/201001/running_code_at_python_startup.html


2017-07-18:我已经改用另一种方法:

logging.basicConfig(level=logging.DEBUG if __debug__ else logging.INFO)

它的作用是,如果您在没有优化的情况下运行(如 中python script.py),您将获得 DEBUG 级别的内容,而如果您在运行时python -OO script.py则没有。没有要设置的环境变量。

于 2015-04-01T23:54:32.280 回答
2

这是另一个让 argparse 计算 -v 选项的方法,以将详细程度从默认的 WARNING 到 INFO (-v) 到 DEBUG (-vv) 增加两个级别。这不映射到定义的常量,logging而是直接计算值,限制输入:

print( "Verbosity / loglevel:", args.v )
logging.basicConfig( level=10*(3-max(0,min(args.v,3))) )
logging.debug("debug") # 10
logging.info("info") # 20
logging.warning("warning") # 30 - The default level is WARNING, which means that only events of this level and above will be tracked
logging.error("error") # 40
logging.critical("critical") # 50
于 2016-06-22T12:40:53.367 回答