Модуль Python OptionParser - несколько записей одной переменной? - PullRequest
Новая
       18

Модуль Python OptionParser - несколько записей одной переменной?

6 голосов
/ 03 мая 2010

Я пишу небольшой скрипт на python для получения статистики с нескольких серверов или одного сервера, и я использую OptionParser для разбора ввода командной строки. 

#!/usr/bin/python

import sys
from optparse import OptionParser
...
parser.add_option("-s", "--server", dest="server", metavar="SERVER", type="string", 
                  help="server(s) to gather stats [default: localhost]")
...

Моя цель - сделать что-то вроде 

#test.py -s server1 -s server2

и это каким-то образом добавит оба этих значения в объекте options.server, чтобы я мог их перебирать, независимо от того, имеют ли они 1 значение или 10.

Любые мысли / помощь приветствуется. Спасибо.

Ответы [ 5 ]

13 голосов
/ 03 мая 2010 
import optparse

parser = optparse.OptionParser()
parser.add_option('-t', '--test', action='append')

options, args = parser.parse_args()
for i, opt in enumerate(options.test):
    print 'option %s: %s' % (i, opt)
3 голосов
/ 02 мая 2012

Да, это можно сделать с помощью optparse.

Это пример: 

./test.py --categories=aaa --categories=bbb --categories ccc arg1 arg2 arg3

который печатает: 

arguments: ['arg1', 'arg2', 'arg3']
options: {'categories': ['aaa', 'bbb', 'ccc']}

Полный рабочий пример ниже: 

#!/usr/bin/env python

import os, sys
from optparse import OptionParser
from optparse import Option, OptionValueError

VERSION = '0.9.4'

class MultipleOption(Option):
    ACTIONS = Option.ACTIONS + ("extend",)
    STORE_ACTIONS = Option.STORE_ACTIONS + ("extend",)
    TYPED_ACTIONS = Option.TYPED_ACTIONS + ("extend",)
    ALWAYS_TYPED_ACTIONS = Option.ALWAYS_TYPED_ACTIONS + ("extend",)

    def take_action(self, action, dest, opt, value, values, parser):
        if action == "extend":
            values.ensure_value(dest, []).append(value)
        else:
            Option.take_action(self, action, dest, opt, value, values, parser)


def main():
    PROG = os.path.basename(os.path.splitext(__file__)[0])
    long_commands = ('categories')
    short_commands = {'cat':'categories'}
    description = """Just a test"""
    parser = OptionParser(option_class=MultipleOption,
                          usage='usage: %prog [OPTIONS] COMMAND [BLOG_FILE]',
                          version='%s %s' % (PROG, VERSION),
                          description=description)
    parser.add_option('-c', '--categories', 
                      action="extend", type="string",
                      dest='categories', 
                      metavar='CATEGORIES', 
                      help='comma separated list of post categories')

    if len(sys.argv) == 1:
        parser.parse_args(['--help'])

    OPTIONS, args = parser.parse_args()
    print "arguments:", args
    print "options:", OPTIONS

if __name__ == '__main__':
    main()

Больше информации на http://docs.python.org/library/optparse.html#adding-new-actions

2 голосов
/ 03 мая 2010

Может быть проще принять список серверов через запятую: 

# test.py -s server1,server2

и разделите значение в вашем скрипте.

2 голосов
/ 03 мая 2010

Вы можете попробовать проверить argparse . Он предоставляет параметр "nargs", означающий, что вы сможете что-то делать вместе с 

#test.py -s server1 server2

Я знаю, что это не совсем то, что требовалось, но это может быть достойный компромисс без особых хлопот. :)

1 голос
/ 07 апреля 2012

Вот известный рабочий пример (от http://docs.python.org/library/email-examples.html):

parser.add_option('-r', '--recipient',
                  type='string', action='append', metavar='RECIPIENT',
                  default=[], dest='recipients',
                  help='A To: header value (at least one required)')

Я думаю, что использование что-то вроде: 

python script.py -r some@where.com -r no@where.com -r any@where.com
...