Python argparse:强制列表项是唯一的 [英] Python argparse: Force a list item to be unique

问题描述

能够使用下面的 choices=servers 验证列表项很好.

Being able to validate the list items using choices=servers below is nice.

servers = [ "ApaServer", "BananServer", "GulServer", "SolServer", "RymdServer", "SkeppServer", "HavsServer", "SovServer" ]
parser = argparse.ArgumentParser()
parser.add_argument('-o', '--only', nargs='*', choices=servers, help='Space separated list of case sensitive server names to process')

是否可以强制列表中的项目唯一，以便不允许重复?

Is it possible to force an item in the list to be unique, so that no duplicates are allowed?

推荐答案

使用 argparse 正确丢弃重复项的方法是创建您自己的 argparse.Action 类按照其他答案的建议使用 set :

The way to properly discard duplicates using argparse would be to create your own argparse.Action class that takes care of using set as suggestted by other answers:

import argparse

class UniqueAppendAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        unique_values = set(values)
        setattr(namespace, self.dest, unique_values)

servers = ["ApaServer", "BananServer", "GulServer", "SolServer",
           "RymdServer", "SkeppServer", "HavsServer", "SovServer" ]
parser = argparse.ArgumentParser()
parser.add_argument('-o', '--only', nargs='*', choices=servers, action=UniqueAppendAction,
                    help='Space separated list of case sensitive server names to process')
print parser.parse_args()

示例输出:

$ python test.py -o ApaServer ApaServer
Namespace(only=set(['ApaServer']))

这篇关于Python argparse:强制列表项是唯一的的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！