如何拒絕負數作爲參數Argparse模塊

Question

我有一個時間參數，它可以是任何數字，除了底片和零

parser.add_argument("-t", "--time", 
        default=2, type=int, 
        help="Settings up the resolution time") 

我如何正確使用選擇的選擇嗎？

Answer 1

您可以通過type=參數add_argument的任何轉換函數。使用您自己的會話功能，其中包括額外的檢查。

def non_negative_int(x): 
    i = int(x) 
    if i < 0: 
     raise ValueError('Negative values are not allowed') 
    return i 

parser.add_argument("-t", "--time", 
        default=2, type=non_negative_int, 
        help="Settings up the resolution time")