python optparse的异常处理

Question

嗨，伙计们。

我正在使用cmd和optparse为一组已有功能的类(CDContainer、CD等)开发一个CLI.py。以下是代码的一些部分。我这里有个问题。当存在异常(输入类型错误或缺少值)时，optparse将退出整个程序，而不是特定的命令方法。

import cmd
class CLI(cmd.Cmd):

    def do_addcd(self, line):
        args=line.split()
        parser = OptionParser()
        parser.add_option("-t", "--track", dest="track_number", type="int",
            help="track number")
        parser.add_option("-n", "--cdname", dest="cd_name", type="string",
            help="CD name")
        (options, positional_args) = parser.parse_args(args[0:])
        cd_obj= CD()
        cd_obj.addCD(options.track_number, options.cd_name)

在">python“下，如果我输入CLI.py，那么我就会有(Cmd)，所以我可以像这样输入命令"(Cmd)addcd -t 3 -n thriller”。但是如果我输入"addcd -t r -n 3"，那么optparse将终止整个CLI.py并退出。这对我不好。我想提醒用户每个方法，而不是终止整个程序。

然而，optparse文档说“整个程序都退出了”。所以我不能“盲目”地使用optparse。我能做什么?

Answer 1

optparse文档中这样写道：

如果optparse的默认错误处理行为不能满足您的需要，您需要子类化OptionParser并覆盖它的

()和/或error()方法。

理想情况下，您应该定义一种新的异常类型，子类optparse，在已覆盖的exit()或error()方法中引发异常，然后捕获它并根据需要处理它。

不过，你可以作弊。如果您希望打印错误消息，但又不希望程序退出，那么您可以捕获SystemExit异常，以捕获optparse试图退出的位置并停止它。

所以，举个例子：

try:    
    (options, positional_args) = parser.parse_args(args[0:])
except SystemExit:
    return

cd_obj= CD()
cd_obj.addCD(options.track_number, options.cd_name)

或者重写该方法：

import optparse

class OptionParsingError(RuntimeError):
    def __init__(self, msg):
        self.msg = msg

class OptionParsingExit(Exception):
    def __init__(self, status, msg):
        self.msg = msg
        self.status = status

class ModifiedOptionParser(optparse.OptionParser):
    def error(self, msg):
        raise OptionParsingError(msg)

    def exit(self, status=0, msg=None):
        raise OptionParsingExit(status, msg)

然后：

try:
    parser = ModifiedOptionParser()
    parser.add_option("-t", "--track", dest="track_number", type="int",
        help="track number")
    (options, positional_args) = parser.parse_args(args[0:])
except OptionParsingError, e:
    print 'There was a parsing error: %s' % e.msg
    return
except OptionParsingExit, e:
    print 'The option parser exited with message %s and result code %s' % (e.msg, e.status)
    return

cd_obj= CD()
cd_obj.addCD(options.track_number, options.cd_name)

Answer 2

这可能与您传递给CD类的类型有关:在看不到它的情况下，它很可能会在那里失败。在创建该对象并传递参数之前，清理该数据、确保其类型正确并执行您认为合理的任何其他检查是一个非常好的主意。

Answer 3

    try:
    parser = ModifiedOptionParser()
    parser.add_option("-t", "--track", dest="track_number", type="int",
        help="track number")
    (options, positional_args) = parser.parse_args(args[0:])
except OptionParsingError, e:
    print 'There was a parsing error: %s' % e.msg
    return
except OptionParsingExit, e:
    print 'The option parser exited with message %s and result code %s' % (e.msg, e.status)
    return

自定义异常类仍然无法处理显示默认帮助上下文的'-h -- help‘选项。

我所做的就是在尝试之前使用sys.argv块来处理帮助函数。

if sys.argv[1] == '-h' or sys.argv[1] == '--help':
    raise Exception('help')

parser = ModifiedOptionParser()
...
except Exception as value:
    status = str(value)
    if status is 'help':
         parser.print_help()
         return -1    # I need to set different return value

顺便说一下，谢谢你的建议。