Question

情景如下：

我需要实现一个允许两者的命令行参数系统

-p <value>

和

-p

（给出-p没有值）

有没有办法用argparse实现这个，还是我必须手动解析sys.argv？

非常感谢你!!!

Answer 1

如果未指定选项（或未指定参数），则将args.param设置为None，否则设置为指定值

import argparse

UNSPECIFIED = object()

parser = argparse.ArgumentParser()
parser.add_argument('-p', action='store', dest='param', default=UNSPECIFIED, nargs='?')

args = parser.parse_args()

if args.param == UNSPECIFIED:
    print("Param unspecified")
else:
    print(args.param)

Answer 2

你需要的是三种状态，缺席、存在、存在价值。

argparse 模块至少提供了两种简单的方法来做到这一点。

为了最简单、最灵活，您只需添加 action='store' 和 nargs='*'。

#!/usr/bin/python

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('-p', action='store', nargs='*')

args = parser.parse_args()

if args.p is not None:
    print( '-p present' )
else:
    print( '-p not present' )

print( '-p value', args.p )

以下是此方法的输出：

# ./testpar.py
-p not present
-p value None

# ./testpar.py -p
-p present
-p value []

# ./testpar.py -p something
-p present
-p value ['something']

# ./testpar.py -p something anotherthing
-p present
-p value ['something', 'anotherthing']

对于第二种方法，您可以为三种状态中的第二种设置特定值，const=somevalue 和 nargs='?'。此方法仅支持一个值，并且 'const' 的值必须是用户无效输入的值。

#!/usr/bin/python

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('-p', action='store', const='NoValue', nargs='?')

args = parser.parse_args()

if args.p:
    print( '-p present' )
else:
    print( '-p not present' )

print( '-p value', args.p )

这是三种情况的输出，

# ./testpar.py
-p not present
-p value None

# ./testpar.py -p
-p present
-p value NoValue

# ./testpar.py -p something
-p present
-p value something

带或不带值的python参数，指定与否

2 个答案: