使用memory_profiler和docopt进行冲突

时间:2013-04-08 16:42:31

标签: python profiling

我有一个使用docopt的Python程序,似乎可以很好地解析命令行参数。但是,当我尝试在调用中包含memory_profiler时(使用-m memory_profiler),docopt无法解析命令并打印用法语句。这是一个示例程序:

"""
Usage:
  ids_transform [options]
  ids_transform --block BLOCK_MSG
  ids_transform --unblock
  ids_transform --version

Examples:
    ids_transform.py --block '2013-03-15 IM#123456 database down'
    ids_transform.py -c ../shared/etc/ids_transform.yaml

Options:
  -h --help                 show this help message and exit
  -c CONFIG --config=CONFIG config file
  -d --debug                begin debugging
  --force                   override block file; force run
  --profile                 use cProfile to gather statistics on process
"""

from docopt import docopt


if __name__ == '__main__':
    arguments = docopt(__doc__, version='1.0.0rc2')
    print(arguments)

这是一个成功的调用:

$ python ids.py -d --force -c foo.yml
{'--block': False,
 '--config': 'foo.yml',
 '--debug': True,
 '--force': True,
 '--help': False,
 '--profile': False,
 '--unblock': False,
 '--version': False,
 'BLOCK_MSG': None}

使用memory_profiler时出现错误:

$ python -m memory_profiler ids.py -d --force -c foo.yml
Usage:
  ids_transform [options]
  ids_transform --block BLOCK_MSG
  ids_transform --unblock
  ids_transform --version

我错过了什么?

1 个答案:

答案 0 :(得分:1)

似乎memory_profiler并没有将自己从sys.argv中删除,所以黑客(我猜)会自己做:

if sys.argv[0].endswith('memory_profiler.py'):
    del sys.argv[0]