Python argparse check Auswahl vor Typ

Question

Ich versuche, einen Benutzer zu ermöglichen, einen Funktionsnamen zu übergeben. Aus irgendeinem Grund scheint argparse die Typüberprüfung/-konvertierung durchzuführen, BEVOR sie die Auswahlmöglichkeiten prüft. Ist das ein Fehler? Beste Sache zu tun?

import argparse 

def foo(): 
    return 'foo' 

def bar(): 
    return 'bar' 

parser = argparse.ArgumentParser() 
functions = {f.__name__:f for f in [foo, bar]} 
parser.add_argument("function", type=lambda f: functions.get(f), help="which function", choices=functions) 
args = parser.parse_args() 
print(args.function()) 

Dies wirft:

$ python blah.py foo 
usage: blah.py [-h] {foo,bar} 
blah.py: error: argument function: invalid choice: <function foo at 0x7f65746dd848> (choose from 'foo', 'bar') 

Answer 1

Ja, während der Analyse der type dann choices Reihenfolge ist klar und absichtlich (und nicht nur zufällig). Bei der Vorbereitung arg_strings zum namespace zuweisen nennt es _get_values, das tut:

def _get_values(self, action, arg_strings) 
     .... (various nargs tests) 
     value = self._get_value(action, arg_string) 
     self._check_value(action, value) 
     return value 

wo _get_value die action.type Funktion gilt, und _check_value Tests

value not in action.choices 

Für choices hat Parsen nur auf die in zu reagieren (__contains__) Ausdruck.

So müssen choices Werte nach der Konvertierung widerspiegeln. Wenn typeint ist, dann ist choices=[1,2,3] korrekt, ['1','2','3'] ist nicht.

Es gibt einige (weitgehend unaufgelöste) Bugprobleme bei der Anzeige der Auswahlmöglichkeiten. Lange Listen, z.B. range(100) arbeiten in Parsing, aber nicht gut angezeigt. Und die Anzeige erfordert auch, dass choices iterierbar ist (z. B. eine Liste, ein Tupel, ein Wörterbuch). Dieses Anzeigeproblem betrifft die Verwendung, die Hilfe und die Fehlermeldungen (jedes Format choices etwas anders).

metavar ist Ihr leistungsstärkstes Werkzeug zum Ersetzen einer unerwünschten choices Liste. Ich müsste einen Testfall ausführen, um zu sehen, ob er die Dinge für alle 3 Situationen löst.

Answer 2

Offenbar so, aber Sie können dies umgehen, indem einfach functions.keys() als Ihre Wahlen, z.B.

import argparse 

def foo(): 
    return 'foo' 

def bar(): 
    return 'bar' 

parser = argparse.ArgumentParser() 
functions = {f.__name__:f for f in [foo, bar]} 
parser.add_argument("function", type=lambda f: functions.get(f), help="which function", choices=functions.values()) 
args = parser.parse_args() 
print(args.function()) 

Wenn Sie jedoch diese Art der Schnittstelle zur Verfügung stellen möchten (Abbildungsfunktionen Zeilenargumente gebieten) Sie vielleicht einen Blick auf click nehmen wollen.