python - 你如何处理不能一起使用的选项(使用 OptionParser)？

Question

我的 Python 脚本(用于待办事项列表)是从命令行启动的，如下所示:

todo [options] <command> [command-options]

有些选项不能一起使用，例如

todo add --pos=3 --end "Ask Stackoverflow"

将指定列表的第三个位置和末尾。同样

todo list --brief --informative

会使我的程序混淆为简短或信息丰富。因为我想有一个相当强大的选项控制，这样的情况会很多，以后肯定会出现新的。如果用户传递了错误的选项组合，我想提供一条信息性消息，最好连同 optparse 提供的使用帮助一起提供。目前我用一个 if-else 语句来处理这个问题，我发现它非常丑陋和糟糕。我的梦想是在我的代码中有这样的东西:

parser.set_not_allowed(combination=["--pos", "--end"], 
                       message="--pos and --end can not be used together")

OptionParser 会在解析选项时使用它。

据我所知这并不存在，我问 SO 社区: 你如何处理这个问题？

Answer 1

可能通过扩展 optparse.OptionParser:

class Conflict(object):
    __slots__ = ("combination", "message", "parser")

    def __init__(self, combination, message, parser):
        self.combination = combination
        self.message = str(message)
        self.parser = parser

    def accepts(self, options):
        count = sum(1 for option in self.combination if hasattr(options, option))
        return count <= 1

class ConflictError(Exception):
    def __init__(self, conflict):
        self.conflict = conflict

    def __str__(self):
        return self.conflict.message

class MyOptionParser(optparse.OptionParser):
    def __init__(self, *args, **kwds):
        optparse.OptionParser.__init__(self, *args, **kwds)
        self.conflicts = []

    def set_not_allowed(self, combination, message):
        self.conflicts.append(Conflict(combination, message, self))

    def parse_args(self, *args, **kwds):
        # Force-ignore the default values and parse the arguments first
        kwds2 = dict(kwds)
        kwds2["values"] = optparse.Values()
        options, _ = optparse.OptionParser.parse_args(self, *args, **kwds2)

        # Check for conflicts
        for conflict in self.conflicts:
            if not conflict.accepts(options):
                raise ConflictError(conflict)

        # Parse the arguments once again, now with defaults
        return optparse.OptionParser.parse_args(self, *args, **kwds)

然后您可以在调用 parse_args 的地方处理 ConflictError:

try:
    options, args = parser.parse_args()
except ConflictError as err:
    parser.error(err.message)