Jak mogę wymagać, aby argument mojego skryptu Pythona był zmiennoprzecinkowy między 0.0-1.0 używając argparse?

Question

Chciałbym użyć argparse na Pythonie 2.7, aby jeden z parametrów skryptu był w zakresie od 0,0 do 1,0. Czy obsługuje to argparse.add_argument()? Parametr

Answer 1

Parametr type do musi być obiektem wywoływalnym, który pobiera ciąg znaków i zwraca wartość przekonwertowaną. Możesz napisać opakowanie o numerze float, które sprawdza jego wartość i powoduje zgłoszenie błędu, jeśli jest poza zakresem.

def restricted_float(x): 
    x = float(x) 
    if x < 0.0 or x > 1.0: 
     raise argparse.ArgumentTypeError("%r not in range [0.0, 1.0]"%(x,)) 
    return x 

p = argparse.ArgumentParser() 
p.add_argument("--arg", type=restricted_float) 

Answer 2

Oto metoda, która wykorzystuje parametr add_argumentchoices, z niestandardowej klasy, który jest uważany za „równa się” do każdego pływaka w określonym zakresie:

import argparse 

class Range(object): 
    def __init__(self, start, end): 
     self.start = start 
     self.end = end 
    def __eq__(self, other): 
     return self.start <= other <= self.end 

parser = argparse.ArgumentParser() 
parser.add_argument('--foo', type=float, choices=[Range(0.0, 1.0)])