Question

我写了一个python程序，它接受三个可选的命令行参数，但是我想限制用户输入的args组合：

python script.py -all
python script.py --country_name
python script.py --country_name --city_name

我其他任何组合程序都不应执行。

注意：我正在使用argparse python模块

Answer 1

如果我能看到您的代码，可能会更容易...但是基本上，一旦您运行：

args = parser.parse_args()

然后，您可以放入自己的验证。

if args.all and (args.country_name or args.city_name):
    raise MySuitableError

Answer 2

Argparse有一个mutually exclusive group：

import argparse

def main(parser, args):
    if args.country_city is not None:
        country, city = args.country_city
    print(args)

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Hello")
    group = parser.add_mutually_exclusive_group()
    group.add_argument("-all")
    group.add_argument("--country_city", nargs="+")
    main(parser, parser.parse_args())

这将导致：

python3 test.py -all A --country_city a b
usage: test.py [-h]
                   [-all ALL | --country_city COUNTRY_CITY [COUNTRY_CITY ...]]
test.py: error: argument --country_city: not allowed with argument -all

python3 test.py --country_city a b
Namespace(all=None, country_city=['a', 'b'])

python3 test.py -all A
Namespace(all='A', country_city=None)

Answer 3

我完全不会使用选项，而是可以忽略的位置参数。

p = ArgumentParser()
p.add_argument('country', nargs='?')
p.add_argument('city', nargs='?')

然后

script.py           # equiv to old script.py --all
script.py FI        # equiv to old script.py --countryname FI
script.py FI Turku  # equivalent to old script.py --countryname FI --cityname Turku

限制命令行参数的组合

3 个答案: