Question

我正在使用外部python模块，这不是我写的，因此无法更改。此模块称为magnum（http://micromagnum.informatik.uni-hamburg.de），处理所有可选的命令行参数。这里举例说明行为：

script1.py

#!/usr/bin/env python
import magnum

执行脚本会产生：

>>> ./script1.py -h
[WARNING] - Python Imaging Library not found!
[WARNING] - -> This means that the ImageShapeCreator and related classes are not available!
[   INFO] - Imported FFTW wisdom from file
Usage: scipt1.py [options]

Options:
  --version             show program's version number and exit
  -h, --help            show this help message and exit

  Hardware options:
    Options that control which hardware is used.

    -g GPU_ID           enable GPU processing (using 32-bit accuracy) on cuda
                    device GPU_ID. The simulator will fall back to CPU
                    mode if it was not compiled with CUDA support or when
                    no CUDA capable graphics cards were detected.
    -G GPU_ID           enable GPU processing (using 64-bit accuracy) on cuda
                    device GPU_ID. TODO: Describe fallback behaviour.
    -t NUM_THREADS, --threads=NUM_THREADS
                    enable CPU multithreading with NUM_THREADS (1..64)
                    threads. This parameter instructs the fftw library to
                    use NUM_THREADS threads for computing FFTs.

  Logging options:
    Options related to logging and benchmarking.

    -l LEVEL, --loglevel=LEVEL
                    set log level (0:Debug, 1:Info, 2:Warn, 4:Error,
                    5:Critical), default is Debug (0).
    --prof              Log profiling info at program exit.

  Parameter sweep options:
    These options have only an effect when the simulation script uses a
    Controller object to sweep through a parameter range.

    -p RANGE, --param-range=RANGE
                    select parameter set to run, e.g. --param-range=0,64
                    to run sets 0 to 63.
    --print-num-params  print number of sweep parameters to stdout and exit.
    --print-all-params  print all sweep parameters to stdout and exit.

  Miscellanous options:
    --on_io_error=MODE  Specifies what to do when an i/o error occurs when
                    writing an .omf/.vtk file. 0: Abort (default), 1:
                    Retry a few times, then abort, 2: Retry a few times,
                    then pause and ask for user intervention

现在我想编写一个小脚本，它接受自己的命令行参数，然后使用magnum模块执行一些小的计算。我想用argparse来解析参数。但是，argparse的优先级似乎低于此外部模块的参数处理，并且我自己的参数无法识别：

script2.py

#!/usr/bin/env python
import argparse
import magnum
parser = argparse.ArgumentParser(
        description='TEST')
parser.add_argument('--x',
        help='test_arg')
args = parser.parse_args()
print args.x

致电：

>>>./scrip2.py --x 3
[WARNING] - Python Imaging Library not found!
[WARNING] - -> This means that the ImageShapeCreator and related classes are not available!
[   INFO] - Imported FFTW wisdom from file
Usage: test.py [options]

test.py: error: no such option: --x

如果我在magnum之前或之后导入argparse并不重要。如果我不输入大量的话，argparse就可以了：

script3.py

#!/usr/bin/env python
import argparse
parser = argparse.ArgumentParser(
        description='TEST')
parser.add_argument('--x',
        help='test_arg')
args = parser.parse_args()
print args.x

执行它会产生：

>>>./scrip2.py --x 3
3

我的问题是：如何在不编辑magnum的情况下停止处理我的命令行参数？

Answer 1

虽然我不认为有什么好的＆＃39;解决方案，你可以monkeypatch argparse让它成为nop：

class EmptyParser():
    def parse_args():
        return
    ... (more redirects for add_argument)
argparse.ArgumentParser = EmptyParser
import magnum

或者，你可以在一个函数中包装导入magnum，并通过该函数创建一个接口，并在magnum解析之前创建参数。

def magnum(param1, param2):
    sys.argv = [param1, '--x', param2]
    import magnum

但是，这两种方法都是超级黑客。

Answer 2

查看github源代码https://github.com/MicroMagnum

import magnum

导入包（magnum/__init__.py）。 init中的关键操作是

config.cfg.initialize(sys.argv)

定义并创建cfg = MagnumConfig()。检查环境。它还定义并运行optparse解析器。

因此没有明显的方法绕过其解析器并仍然设置计算环境。在sys.argv之前进行自己的解析和调整import magnum似乎是唯一的解决方案。

Answer 3

为了将来参考，这里是基于上述评论/答案的工作解决方案：

script3.py

#!/usr/bin/env python
import argparse
import sys
parser = argparse.ArgumentParser(
    description='TEST')
parser.add_argument('--x',
    help='test_arg')
args = parser.parse_args()
sys.argv = [sys.argv[0]]
import magnum
print args.x

执行脚本表明正确导入了magnum，但是参数已被解析并存储在args变量中：

>>>./script3.py --x 3
[WARNING] - Python Imaging Library not found!
[WARNING] - -> This means that the ImageShapeCreator and related classes are not available!
[   INFO] - Imported FFTW wisdom from file
[   INFO] - ----------------------------------------------------------------------
[   INFO] - MicroMagnum 0.2rc3
[   INFO] - Copyright (C) 2012 by the MicroMagnum team.
[   INFO] - This program comes with ABSOLUTELY NO WARRANTY.
[   INFO] - This is free software, and you are welcome to redistribute it under
[   INFO] - certain conditions; see the file COPYING in the distribution package.
[   INFO] - ----------------------------------------------------------------------
[   INFO] - FFTW using 1 threads from now on
[   INFO] - CUDA GPU support: no
3

此解决方案不会将任何参数转发给magnum，这是我想要实现的目标。

python：停止导入的模块解析命令行参数

3 个答案: