Python: Может ли optparse иметь атрибут ACTION действовать как STORE и STORE_TRUE?

Question

Я использую optparse, чтобы получить ввод командной строки.

Допустим, что я запускаю скрипт demo.py, и он создает некоторый вывод. Но если я не укажу ввод командной строки, вывод не будет записан в файл.

Я пытаюсь сделать следующее:

python demo.py в командной строке необходимо запустить скрипт, но не писать вывод в любом месте.

python demo.py -o в командной строке должен записать вывод в мое имя по умолчанию output.txt.

python demo.py -o demooutput.txt в командной строке должен записать вывод в файл demooutput.txt.

PS: Я бы не стал переключаться на argparse от optparse.

Answer 1

Для достижения этой цели вы можете использовать optparse-callbacks.

Вот как это будет работать для вашего использования.

parser.add_option("-o", action="callback", dest="output", callback=my_callback) 

def my_callback(option, opt, value, parser): 
    if len(parser.rargs) > 0: 
     next_arg = parser.rargs[0] 
     if not next_arg.startswith("-"): 
      # Next argument is not another option 
      del parser.rargs[0] 
      setattr(parser.values, option.dest, next_arg) 
      return 
    # If not processed, set the default value 
    setattr(parser.values, option.dest, "output.txt") 

Answer 2

Я не думаю, что есть, к сожалению, единственный способ, которым я могу думать, - это взломать проблему, добавив свои собственные логические утверждения. Следующий код должен сделать трюк.

import re, sys 
import optparse from OptionParser  
usage = "usage: %prog [options] arg" 
parser = OptionParser(usage) 
if '-f' in argv: 
    a = argv.index('-f') 
    if (a != len(argv)-1) and re.search('[.]txt', argv[a+1]): 
     parser.add_option("-f", "--foo", dest="foo") 
    else: 
     parser.add_option("-f", dest="foo", action="store_true") 

Answer 3

Это не отвечает на прямой вопрос «как определить действие ...», но он обрабатывает входы простым способом.

Комплект '-o' будет 'store_true'. Если True проверит переменную 'args' для имени файла.

(options, args) = parser.parse_args() 
if options.o: 
    if args: 
     dest = args[0] 
    else: 
     dest = 'output.txt' 
else: 
    dest = '' 

(В argparse эквиваленте был бы определить позиционный спор с nargs='?'.)

Если только эти аргументы, можно также получить с проверкой на имя файла без необходимости `-o'.

Другая возможность - 'store_const', с позиционной 'имя файла' имеет приоритет:

parser = optparse.OptionParser() 
parser.add_option('-o',dest='dest',action='store_const', const='output.txt', default='') 
(options, args) = parser.parse_args() 
if args: 
    options.dest = args[0] 
print options