Question

在文件中（比如parser.py）我有：

import argparse

def parse_cmdline(cmdline=None):
    parser = argparse.ArgumentParser()
    parser.add_argument('--first-param',help="Does foo.")
    parser.add_argument('--second-param',help="Does bar.")

    if cmdline is not None:
        args = parser.parse_args(cmdline)
    else:
        args = parser.parse_args()

    return vars(args)

if __name__=='__main__':
    print parse_cmdline()

果然，当从命令行调用时，它可以工作，并且给了我很多我期望的东西：

$ ./parser.py --first-param 123 --second-param 456
{'first_param': '123', 'second_param': '456'}

但是我想unittest它，因此我写了一个test_parser.py文件：

import unittest
from parser import parse_cmdline

class TestParser(unittest.TestCase):
    def test_parse_cmdline(self):
        parsed = parse_cmdline("--first-param 123 --second-param 456")

        self.assertEqual(parsed['first_param'],'123')
        self.assertEqual(parsed['second_param'],'456')

if __name__ == '__main__':
    unittest.main()

然后我收到以下错误：

usage: test_parser.py [-h] [--first-param FIRST_PARAM]
                      [--second-param SECOND_PARAM]
test_parser.py: error: unrecognized arguments: - - f i r s t - p a r a m   1 2 3   - - s e c o n d - p a r a m   4 5 6
E
======================================================================
ERROR: test_parse_cmdline (__main__.TestParser)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "./test_parser.py", line 8, in test_parse_cmdline
    parsed = parse_cmdline("--first-param 123 --second-param 456")
  File "/home/renan/test_argparse/parser.py", line 12, in parse_cmdline
    args = parser.parse_args(cmdline)
  File "/usr/lib/python2.7/argparse.py", line 1691, in parse_args
    self.error(msg % ' '.join(argv))
  File "/usr/lib/python2.7/argparse.py", line 2361, in error
    self.exit(2, _('%s: error: %s\n') % (self.prog, message))
  File "/usr/lib/python2.7/argparse.py", line 2349, in exit
    _sys.exit(status)
SystemExit: 2

----------------------------------------------------------------------
Ran 1 test in 0.004s

FAILED (errors=1)

可以看出，我指定的命令行（--first-param 123 --second-param 456）成了 - - f i r s t - p a r a m 1 2 3 - - s e c o n d - p a r a m 4 5 6（每个字符用空格分隔）。

我不明白为什么：我做错了什么？

Answer 1

argparse想要一个“参数向量” - 即一个单独的参数列表 - 而不是“命令行”字符串。

只是致电split无法解决问题。例如，来自shell的这个命令行：

python script.py --first-param '123 456' --second-param 789

...会正确地为您提供123 456和789，但这行代码：

parse_cmdline("--first-param '123 456' --second-param 789")

不会;它会为您'123和789提供额外的456'，但它不知道如何处理。

事实上，即使这是错误的：

parse_cmdline("--first-param '123' --second-param 789")

...因为您将获得'123'而不是123。

有两种方法可以解决这个问题。

首先，如果你知道你想要传递的是什么，你可以先把它作为一个列表而不是字符串传递给你，你不必担心引用和分割细节：

parse_cmdline(["--first-param", "123 456", "--second-param", "789"])

或者，如果您不确切知道自己要传递的内容，只需知道它在shell上的样子，就可以使用shlex：

if cmdline is not None:
    args = parser.parse_args(shlex.split(cmdline))

...现在Python将以与标准Unix shell相同的方式拆分命令行，将123 456作为单个参数。

Answer 2

回答自己（几分钟后我意识到自己的错误）：

我应该

if cmdline is not None:
    args = parser.parse_args(cmdline.split())
else:
    args = parser.parse_args()

现在测试正确通过了！

从unittest测试调用时，argparse失败

2 个答案: