限制命令行参数的组合

Question

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

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有一个互斥组：

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