Question

我想知道是否可以根据是否给出值来改变参数的行为。

通常，我想提供一种输入密码的交互方式。

示例用法：

$ script.py =＆gt;使用参数'p'的默认参数值

$ script.py -p mypass =＆gt;使用'mypass'作为'p'的值

$ script.py -p =＆gt;以交互方式获取密码

Answer 1

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("-p", nargs='?', const="mypass", default="default")

args = parser.parse_args()

if args.p:
    print args.p

然后，您可以检查p的值并以交互方式获取密码

Answer 2

一个简单的课程：

class AskUser:
    def __init__(self, type, func, *args, **kwargs):
        self.type = type
        self.func = func
        self.args = args
        self.kwargs = kwargs
    def __call__(self, value):
        if value == '':
            value = self.func(*self.args, **self.kwargs)
        return self.type(value)

你可以直接插入argparse：

parser = argparse.ArgumentParser()
parser.add_argument('user', nargs='?', type=AskUser(str, input, 'Enter user name: '), default='')
parser.add_argument('file', nargs='?', type=AskUser(Path, filedialog.asksaveasfilename,
    title='Select a file', filetypes=(('CSV files', '*.csv'),), defaultextension='.csv'), default='')
args = parser.parse_args()

如果没有给出值，[Python argparse]将参数的行为设置为交互式

2 个答案: