Z modułem optparse Pythona, jak utworzyć opcję, która pobiera zmienną liczbę argumentów?

Question

Z Perl Getopt::Long można łatwo określić opcje wiersza polecenia, które mają zmienną liczbę argumentów:

foo.pl --files a.txt    --verbose 
foo.pl --files a.txt b.txt c.txt --verbose 

Czy istnieje sposób, aby to zrobić bezpośrednio z modułem Pythona optparse? O ile mogę powiedzieć, atrybut opcji nargs może być użyty do określenia stałej liczby argumentów opcji i nie widziałem innych alternatyw w dokumentacji.

Answer 1

Wierzę, że optparse nie obsługuje tego, czego potrzebujesz (nie bezpośrednio - jak zauważyłeś, możesz to zrobić, jeśli chcesz wykonać dodatkową pracę wywołania zwrotnego!). Możesz także zrobić to najprościej z rozszerzeniem trzeciej strony argparse, które obsługuje zmienne liczby argumentów (a także dodaje kilka innych poręcznych bitów funkcji).

This URL dokumenty argparse „s add_argument - przechodzące nargs='*' opcja pozwala wziąć zero lub więcej argumentów, '+' pozwala trwać jeden lub więcej argumentów, itp

Answer 2

Mój błąd: właśnie znalazłem to Callback Example 6.

Answer 3

Czy nie lepiej z tym?

foo.pl --files a.txt,b.txt,c.txt --verbose 

Answer 4

Zajęło mi to trochę czasu, aby się dowiedzieć, ale można użyć akcji wywołania zwrotnego do opcji, aby to zrobić. Sprawdź, w jaki sposób pobieram dowolną liczbę argumentów do flagi "--file" w tym przykładzie.

from optparse import OptionParser, 

def cb(option, opt_str, value, parser): 
     args=[] 
     for arg in parser.rargs: 
       if arg[0] != "-": 
         args.append(arg) 
       else: 
         del parser.rargs[:len(args)] 
         break 
     if getattr(parser.values, option.dest): 
       args.extend(getattr(parser.values, option.dest)) 
     setattr(parser.values, option.dest, args) 

parser=OptionParser() 
parser.add_option("-q", "--quiet", 
     action="store_false", dest="verbose", 
     help="be vewwy quiet (I'm hunting wabbits)") 
parser.add_option("-f", "--filename", 
     action="callback", callback=cb, dest="file") 

(options, args) = parser.parse_args() 

print options.file 
print args 

Jedynym skutkiem ubocznym jest uzyskanie argumentów na liście zamiast krotki. Ale można to łatwo naprawić, dla mojego konkretnego przypadku lista jest pożądana.

Answer 5

Oto jeden sposób: Weź generujący ciąg fileLst w postaci ciągu znaków, a następnie użyć http://docs.python.org/2/library/glob.html zrobić ekspansję ugh to nie może działać bez ucieczki *

Właściwie lepszy sposób: Pythona myprog.py -V -l 1000 /home/dominic/radar/*.json < - Jeśli jest to Twoja linia poleceń

parser, zdecyduje = parse_args()

inFileLst = parser.largs

Answer 6

Niedawno miałem ten problem osobiście: byłem na Pythonie 2.6 i potrzebowałem opcji, aby pobrać zmienną liczbę argumentów. Próbowałem użyć rozwiązanie Dave'a, ale okazało się, że nie będzie działać bez również wyraźnie określających nargs 0.

def arg_list(option, opt_str, value, parser): 
    args = set() 
    for arg in parser.rargs: 
     if arg[0] == '-': 
      break 
     args.add(arg) 
     parser.rargs.pop(0) 
    setattr(parser.values, option.dest, args) 

parser=OptionParser() 
parser.disable_interspersed_args() 
parser.add_option("-f", "--filename", action="callback", callback=arg_list, 
        dest="file", nargs=0) 

(options, args) = parser.parse_args() 

Problemem było to, że domyślnie zakłada się nowa opcja dodana przez add_options mieć nargs = 1 i gdy nargs> 0 OptionParser wyskoczy przedmioty z rargs i przypisze je do wartości przed wywołaniem wywołań zwrotnych. Tak więc, dla opcji, które nie określają nargs, rargs będzie zawsze wyłączony o jeden do czasu wywołania callback.

To wywołanie zwrotne może być używane dla dowolnej liczby opcji, po prostu wywołaj funkcję callback jako funkcję, która ma zostać wywołana zamiast setattr.