Установить и потребовать сценарий Python по умолчанию OptionParser - PullRequest
Новая
       43

Установить и потребовать сценарий Python по умолчанию OptionParser

1 голос
/ 19 марта 2011

Следующие операторы "parser.add_option" работают, но если скрипт выполняется без опции / arg, он не будет жаловаться. Если параметр / аргумент не указан, я бы хотел, чтобы по умолчанию отображалась справка (-h / --help). 

usage = "usage: %prog [options] arg"
parser = OptionParser(usage)
parser.add_option('-d', '--directory',
        action='store', dest='directory',
        default=None, help='specify directory')
parser.add_option('-f', '--file',
        action='store', dest='filename',
        default=None, help='specify file')
parser.add_option('-v', '--version',
                  action="store_true", dest="show_version",
                  default=False, help='displays the version number')
(options, args) = parser.parse_args()
#if len(args) < 1:
#    parser.error("incorrect number of arguments")

Во-вторых, если я включаю следующий фрагмент кода, то получаю «ошибка: неверное количество аргументов» даже при указании параметра / аргумента. 

if len(args) < 1:
parser.error("incorrect number of arguments")

Спасибо.

Обновлен код с ошибкой трассировки ниже 

def main():
    usage = "usage: %prog [options] arg"
    parser = OptionParser(usage)
    parser.add_option('-d', '--directory',
            action='store', dest='directory',
            default=None, help='specify directory')
    parser.add_option('-f', '--file',
            action='store', dest='filename',
            default=None, help='specify file')
    parser.add_option('-v', '--version',
                      action="store_true", dest="show_version",
                      default=False, help='displays the version number')
    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit()
    (options, args) = parser.parse_args()

#if options.show_version:
#    prog = os.path.basename(sys.argv[0])
#    version_str = "1.0"
#    print "version is: %s %s" % (prog, version_str)
#    sys.exit(0)

filenames_or_wildcards = []

# remove next line if you do not want allow to run the script without the -f -d
# option, but with arguments
filenames_or_wildcards = args # take all filenames passed in the command line

# if -f was specified add them (in current working directory)
if options.filename is not None:
    filenames_or_wildcards.append(options.filename)

Traceback 

$ python boto-backup.py Traceback (most recent call last):   File "boto-backup.py", line 41, in <module>
    filenames_or_wildcards = args # take all filenames passed in the command line NameError: name 'args' is not defined

1 Ответ

1 голос
/ 19 марта 2011

Я бы сделал что-то вроде этого: 

from optparse import OptionParser
import sys

def main():
    usage = "usage: %prog [options] arg"
    parser = OptionParser(usage)
    parser.add_option('-d', '--directory',
            action='store', dest='directory',
            default=None, help='specify directory')
    parser.add_option('-f', '--file',
            action='store', dest='filename',
            default=None, help='specify file')
    parser.add_option('-v', '--version',
                      action="store_true", dest="show_version",
                      default=False, help='displays the version number')
    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit()
    (options, args) = parser.parse_args()
    # rest of program...

if __name__ == '__main__':
    main()

Итак, мы настроили парсер и опции, а затем проверили, был ли какой-либо ввод из командной строки.
Если нет, мы распечатаем справкусообщение и выход.В противном случае мы продолжаем выполнение программы.

При запуске без аргументов командной строки вывод: 

Usage: your_script_name_here.py [options] arg

Options:
  -h, --help            show this help message and exit
  -d DIRECTORY, --directory=DIRECTORY
                        specify directory
  -f FILENAME, --file=FILENAME
                        specify file
  -v, --version         displays the version number

Edit (в ответ на обновленный код):
Пробел/ отступ важен в Python.
Убедитесь, что остальная часть вашего кода имеет отступ, так что он принадлежит функции main().
Начиная с filenames_or_wildcards = [], ваш код выходит за рамки main() функция и, следовательно, не имеет переменной с именем args.

...