Argeparse確認

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

我有同樣的問題（在不同的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 

Answer 2

東西從閱讀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)來利用這個類。