Question

我在我的程序中使用aprgeparse创建一个参数，允许用户从亚马逊s3存储桶中删除文件。我以这种方式创建它：

parser.add_argument("-r", "--remove",type = str, help= "Delete a file (enter full path to file)" , default = '' )

然后检查文件是否可用，如果是，我将其删除：

if args.remove:
    b = Bucket(conn,default_bucket)
    k = Key(b)
    if b.get_key(args.remove):
        b.delete_key(args.remove)
        print ( ' {0} Has been removed'.format(args.remove.split("/")[-1]) )
    else:
        print ( " No files named '{0}' found ".format(args.remove.split("/")[-1] ) )
        print ( ' Please check the bucket path or file name. Use -l command to list files')

我想知道是否有方法可以提示用户他是否真的打算删除文件，这通常是删除文件的情况（每个程序都这样做）。像这样的东西

>python myprog.py -r foo.txt
>Are your sure [Y/N] : Y
>File deleted

Answer 1

从阅读docs开始，当我看到那个参数时，我会使用argparse.Action的子类来执行自定义行为。我将在文档中包含稍微修改过的代码版本供您使用。

class CheckDeleteAction(argparse.Action):
    def __init__(self, option_strings, dest, nargs=None, **kwargs):
        if nargs is not None:
            raise ValueError("nargs not allowed")
        super(CheckDeleteAction, self).__init__(option_strings, dest, **kwargs)

    def __call__(self, parser, namespace, values, option_string):
        # put your prompting and file deletion logic here
        pass

添加此parser.add_argument("-r", "--remove", type=str, help="msg", default='', action=CheckDeleteAction)以使用该课程。

Answer 2

我有同样的问题（在不同的CLI中）。我建议你使用here给出的答案。看看詹姆斯给出的答案。它相当容易理解。像这样使用它：

if args.remove:
    b = Bucket(conn,default_bucket)
    k = Key(b)
    if b.get_key(args.remove):
        if yes_no('Are you sure: ')
            b.delete_key(args.remove)
            print ( ' {0} Has been removed'.format(args.remove.split("/")[-1]) )
    else:
        print ( " No files named '{0}' found ".format(args.remove.split("/")[-1] ) )
        print ( ' Please check the bucket path or file name. Use -l command to list

Argparse确认

2 个答案: