My<cod/>.net
Как лучше всего анализировать аргументы командной строки?
Какой самый простой , лаконичный и гибкий метод или библиотека для разбора аргументов командной строки Python?
Другие ответы действительно упоминают, что argparse -это путь к новому Python, но не приводят примеров использования. Для полноты картины, вот краткое описание того, как использовать argparse:
1) инициализировать
import argparse
# Instantiate the parser
parser = argparse.ArgumentParser(description='Optional app description')
2) Добавить Аргументы
# Required positional argument
parser.add_argument('pos_arg', type=int,
help='A required integer positional argument')
# Optional positional argument
parser.add_argument('opt_pos_arg', type=int, nargs='?',
help='An optional integer positional argument')
# Optional argument
parser.add_argument('--opt_arg', type=int,
help='An optional integer argument')
# Switch
parser.add_argument('--switch', action='store_true',
help='A boolean switch')
3) разбор
args = parser.parse_args()
4) доступ
print("Argument values:")
print(args.pos_arg)
print(args.opt_pos_arg)
print(args.opt_arg)
print(args.switch)
5) Проверьте Значения
if args.pos_arg > 10:
parser.error("pos_arg cannot be larger than 10")
Использование
Правильное использование:
$ ./app 1 2 --opt_arg 3 --switch
Argument values:
1
2
3
True
Неверные аргументы:
$ ./app foo 2 --opt_arg 3 --switch
usage: convert [-h] [--opt_arg OPT_ARG] [--switch] pos_arg [opt_pos_arg]
app: error: argument pos_arg: invalid int value: 'foo'
$ ./app 11 2 --opt_arg 3
Argument values:
11
2
3
False
usage: app [-h] [--opt_arg OPT_ARG] [--switch] pos_arg [opt_pos_arg]
convert: error: pos_arg cannot be larger than 10
Полная помощь:
$ ./app -h
usage: app [-h] [--opt_arg OPT_ARG] [--switch] pos_arg [opt_pos_arg]
Optional app description
positional arguments:
pos_arg A required integer positional argument
opt_pos_arg An optional integer positional argument
optional arguments:
-h, --help show this help message and exit
--opt_arg OPT_ARG An optional integer argument
--switch A boolean switch
Этот ответ предполагает optparse , который подходит для более старых версий Python. Для Python 2.7 и выше argparse заменяет optparse . Смотрите этот ответ для получения дополнительной информации.
Как отмечали другие люди, вам лучше пойти с optparse, чем с getopt. getopt-это в значительной степени отображение one-to-one стандартных функций библиотеки getopt(3) C, и не очень простое в использовании.
optparse, будучи немного более подробным, гораздо лучше структурирован и проще расширяется в дальнейшем.
Вот типичная строка для добавления опции в ваш парсер:
parser.add_option('-q', '--query',
action="store", dest="query",
help="query string", default="spam")
Он в значительной степени говорит сам за себя; во время обработки он примет-q или --query в качестве опций, сохранит аргумент в атрибуте query и будет иметь значение по умолчанию, если вы его не укажете. Он также является самодокументирующим в том, что вы объявляете аргумент help (который будет использоваться при запуске с параметром -h/--help) прямо там с опцией.
Обычно вы разбираете свои аргументы с помощью:
options, args = parser.parse_args()
Это будет, по умолчанию, разбор стандартных аргументов, переданных скрипту (sys.argv[1:])
options.query затем будет установлено значение, которое вы передали скрипту.
Вы создаете парсер просто сделав это
parser = optparse.OptionParser()
Это все основы, которые вам нужны. Вот полный сценарий Python, который показывает это:
import optparse
parser = optparse.OptionParser()
parser.add_option('-q', '--query',
action="store", dest="query",
help="query string", default="spam")
options, args = parser.parse_args()
print 'Query string:', options.query
5 строк python, которые показывают вам основы.
Сохраните его в sample.py и запустите один раз с помощью
python sample.py
и однажды с
python sample.py --query myquery
Кроме того, вы обнаружите, что optparse очень легко расширить. В одном из моих проектов я создал командный класс, который позволяет легко вложить подкоманды в дерево команд. Он использует optparse сильно для того, чтобы связывать команды вместе. Это не то, что я могу легко объяснить в нескольких строках, но не стесняйтесь искать в моем репозитории основной класс, а также класс, который использует его и парсер опций
Использование docopt
С 2012 года Python имеет очень простой, мощный и действительно классный модуль для разбора аргументов под названием docopt . Он работает с Python 2.6 до 3.5 и не нуждается в установке (просто скопируйте его). Вот пример, взятый из его документации:
"""Naval Fate.
Usage:
naval_fate.py ship new <name>...
naval_fate.py ship <name> move <x> <y> [--speed=<kn>]
naval_fate.py ship shoot <x> <y>
naval_fate.py mine (set|remove) <x> <y> [--moored | --drifting]
naval_fate.py (-h | --help)
naval_fate.py --version
Options:
-h --help Show this screen.
--version Show version.
--speed=<kn> Speed in knots [default: 10].
--moored Moored (anchored) mine.
--drifting Drifting mine.
"""
from docopt import docopt
if __name__ == '__main__':
arguments = docopt(__doc__, version='Naval Fate 2.0')
print(arguments)
Итак, вот ОНО: 2 строки кода плюс ваша строка doc, которая является существенной, и вы получаете ваши аргументы, проанализированные и доступные в вашем объекте arguments. Я же говорил тебе, что это круто, не так ли ;-)
Использование python-fire
С 2017 года python-fire имеет еще один классный модуль, который может дать интерфейс CLI вашему коду, когда вы делаете разбор нулевого аргумента. Вот простой пример из документации (эта небольшая программа предоставляет функцию double в командной строке):
import fire
class Calculator(object):
def double(self, number):
return 2 * number
if __name__ == '__main__':
fire.Fire(Calculator)
Из командной строки можно выполнить команду:
> calculator.py double 10
20
> calculator.py double --number=15
30
Потрясающе, правда?
Я предпочитаю щелчок . Он абстрагирует управляющие параметры и позволяет "(...) создание красивых интерфейсов командной строки в композиционном виде с минимальным количеством кода".
Вот пример использования:
import click
@click.command()
@click.option('--count', default=1, help='Number of greetings.')
@click.option('--name', prompt='Your name',
help='The person to greet.')
def hello(count, name):
"""Simple program that greets NAME for a total of COUNT times."""
for x in range(count):
click.echo('Hello %s!' % name)
if __name__ == '__main__':
hello()
Он также автоматически создает красиво отформатированные страницы справки:
$ python hello.py --help
Usage: hello.py [OPTIONS]
Simple program that greets NAME for a total of COUNT times.
Options:
--count INTEGER Number of greetings.
--name TEXT The person to greet.
--help Show this message and exit.
Почти все используют getopt
Вот пример кода для doc :
import getopt, sys
def main():
try:
opts, args = getopt.getopt(sys.argv[1:], "ho:v", ["help", "output="])
except getopt.GetoptError:
# print help information and exit:
usage()
sys.exit(2)
output = None
verbose = False
for o, a in opts:
if o == "-v":
verbose = True
if o in ("-h", "--help"):
usage()
sys.exit()
if o in ("-o", "--output"):
output = a
Итак, одним словом, вот как это работает.
У вас есть два типа вариантов. Те, кто получает Аргументы, и те, кто их получает прямо как выключатели.
sys.argv -это в значительной степени ваш char** argv в C. Как и в C, вы пропускаете первый элемент, который является именем вашей программы, и анализируете только аргументы: sys.argv[1:]
Getopt.getopt разберет его в соответствии с правилом, которое вы даете в аргументации.
"ho:v" здесь описаны короткие аргументы : -ONELETTER . : означает, что -o принимает один аргумент.
Наконец, ["help", "output="] описывает длинные аргументы ( --MORETHANONELETTER ).
= после вывода еще раз означает, что вывод принимает один аргумент.
Результатом является список пар (опция, аргумент)
Если опция не принимает никаких аргументов (например, здесь --help ), то часть arg является пустой строкой.
Затем обычно требуется выполнить цикл по этому списку и проверить имя параметра, как в Примере.
Надеюсь, это Вам помогло.
Используйте optparse , который поставляется вместе со стандартной библиотекой. Например:
#!/usr/bin/env python
import optparse
def main():
p = optparse.OptionParser()
p.add_option('--person', '-p', default="world")
options, arguments = p.parse_args()
print 'Hello %s' % options.person
if __name__ == '__main__':
main()
Источник: использование Python для создания инструментов командной строки UNIX
Однако по состоянию на Python 2.7 optparse является устаревшим, см.: Почему используется argparse, а не optparse?
Просто в случае, если вам может понадобиться, это может помочь, если вам нужно захватить аргументы unicode на Win32 (2K, XP и т. д):
from ctypes import *
def wmain(argc, argv):
print argc
for i in argv:
print i
return 0
def startup():
size = c_int()
ptr = windll.shell32.CommandLineToArgvW(windll.kernel32.GetCommandLineW(), byref(size))
ref = c_wchar_p * size.value
raw = ref.from_address(ptr)
args = [arg for arg in raw]
windll.kernel32.LocalFree(ptr)
exit(wmain(len(args), args))
startup()
Легкий аргумент командной строки по умолчанию
Хотя argparse отлично подходит и является правильным ответом для полностью документированных переключателей командной строки и расширенных функций, вы можете использовать параметры функции по умолчанию для обработки простых позиционных аргументов очень просто.
import sys
def get_args(name='default', first='a', second=2):
return first, int(second)
first, second = get_args(*sys.argv)
print first, second
Аргумент 'name' захватывает имя скрипта и не используется. Тестовый вывод выглядит следующим образом:
> ./test.py
a 2
> ./test.py A
A 2
> ./test.py A 20
A 20
Для простых скриптов, где мне просто нужны некоторые значения по умолчанию, я нахожу это вполне достаточным. Вы также можете включить некоторые типы принуждения в возвращаемые значения или значения командной строки будут строками.
Я думаю, что лучший способ для больших проектов-это optparse, но если вы ищете простой способ, возможно, http://werkzeug.pocoo.org/documentation/script -это что-то для вас.
from werkzeug import script
# actions go here
def action_foo(name=""):
"""action foo does foo"""
pass
def action_bar(id=0, title="default title"):
"""action bar does bar"""
pass
if __name__ == '__main__':
script.run()
Таким образом, в основном каждая функция action_* подвергается воздействию командной строки и приятной справка генерируется бесплатно.
python foo.py
usage: foo.py <action> [<options>]
foo.py --help
actions:
bar:
action bar does bar
--id integer 0
--title string default title
foo:
action foo does foo
--name string
Я предпочитаю optparse getopt. Это очень декларативно: вы говорите ему названия опций и эффекты, которые они должны иметь (например, установка логического поля), и он возвращает вам словарь, заполненный в соответствии с вашими спецификациями.
http://docs.python.org/lib/module-optparse.html
consoleargs заслуживает того, чтобы быть упомянутым здесь. Он очень прост в использовании. Проверить его:
from consoleargs import command
@command
def main(url, name=None):
"""
:param url: Remote URL
:param name: File name
"""
print """Downloading url '%r' into file '%r'""" % (url, name)
if __name__ == '__main__':
main()
Теперь в консоли:
% python demo.py --help
Usage: demo.py URL [OPTIONS]
URL: Remote URL
Options:
--name -n File name
% python demo.py http://www.google.com/
Downloading url ''http://www.google.com/'' into file 'None'
% python demo.py http://www.google.com/ --name=index.html
Downloading url ''http://www.google.com/'' into file ''index.html''
Код Argparse может быть длиннее, чем фактический код реализации!
Это проблема, которую я нахожу с самыми популярными вариантами синтаксического анализа аргументов, заключается в том, что если ваши параметры только скромны, код для их документирования становится непропорционально большим для пользы, которую они предоставляют.
Относительный новичок в сцене разбора аргументов (я думаю) - это plac .
Он делает некоторые признанные компромиссы с argparse, но использует встроенную документацию и просто обертывает функцию типа main() function:
def main(excel_file_path: "Path to input training file.",
excel_sheet_name:"Name of the excel sheet containing training data including columns 'Label' and 'Description'.",
existing_model_path: "Path to an existing model to refine."=None,
batch_size_start: "The smallest size of any minibatch."=10.,
batch_size_stop: "The largest size of any minibatch."=250.,
batch_size_step: "The step for increase in minibatch size."=1.002,
batch_test_steps: "Flag. If True, show minibatch steps."=False):
"Train a Spacy (http://spacy.io/) text classification model with gold document and label data until the model nears convergence (LOSS < 0.5)."
pass # Implementation code goes here!
if __name__ == '__main__':
import plac; plac.call(main)
Вот метод, а не библиотека, которая, кажется, работает для меня.
Цели здесь должны быть лаконичными, каждый аргумент разбирается одной строкой, args выстраиваются для удобства чтения, код прост и не зависит от каких-либо специальных модулей (только os + sys), предупреждает о пропущенных или неизвестных аргументах изящно, использует простой цикл for/range() и работает через python 2.x и 3.x
Показаны два переключения флагов (-д-В), и двух значений контролируемых параметров (-я ХХХ и-o ХХХ).
import os,sys
def HelpAndExit():
print("<<your help output goes here>>")
sys.exit(1)
def Fatal(msg):
sys.stderr.write("%s: %s\n" % (os.path.basename(sys.argv[0]), msg))
sys.exit(1)
def NextArg(i):
'''Return the next command line argument (if there is one)'''
if ((i+1) >= len(sys.argv)):
Fatal("'%s' expected an argument" % sys.argv[i])
return(1, sys.argv[i+1])
### MAIN
if __name__=='__main__':
verbose = 0
debug = 0
infile = "infile"
outfile = "outfile"
# Parse command line
skip = 0
for i in range(1, len(sys.argv)):
if not skip:
if sys.argv[i][:2] == "-d": debug ^= 1
elif sys.argv[i][:2] == "-v": verbose ^= 1
elif sys.argv[i][:2] == "-i": (skip,infile) = NextArg(i)
elif sys.argv[i][:2] == "-o": (skip,outfile) = NextArg(i)
elif sys.argv[i][:2] == "-h": HelpAndExit()
elif sys.argv[i][:1] == "-": Fatal("'%s' unknown argument" % sys.argv[i])
else: Fatal("'%s' unexpected" % sys.argv[i])
else: skip = 0
print("%d,%d,%s,%s" % (debug,verbose,infile,outfile))
Цель NextArg()-вернуть следующий аргумент при проверке отсутствующих данных,а 'skip' пропускает цикл при использовании NextArg(), сохраняя разбор флага до одной строки.
Я расширил подход Erco, чтобы учесть необходимые позиционные Аргументы и необязательные аргументы. Они должны предшествовать-d, - v и т. д. аргументы.
Позиционные и необязательные аргументы могут быть получены с помощью PosArg(i) и OptArg(i, default) соответственно. При обнаружении необязательного аргумента начальная позиция поиска опций (например,- i) перемещается на 1 вперед, чтобы избежать возникновения фатального 'unexpected'.
import os,sys
def HelpAndExit():
print("<<your help output goes here>>")
sys.exit(1)
def Fatal(msg):
sys.stderr.write("%s: %s\n" % (os.path.basename(sys.argv[0]), msg))
sys.exit(1)
def NextArg(i):
'''Return the next command line argument (if there is one)'''
if ((i+1) >= len(sys.argv)):
Fatal("'%s' expected an argument" % sys.argv[i])
return(1, sys.argv[i+1])
def PosArg(i):
'''Return positional argument'''
if i >= len(sys.argv):
Fatal("'%s' expected an argument" % sys.argv[i])
return sys.argv[i]
def OptArg(i, default):
'''Return optional argument (if there is one)'''
if i >= len(sys.argv):
Fatal("'%s' expected an argument" % sys.argv[i])
if sys.argv[i][:1] != '-':
return True, sys.argv[i]
else:
return False, default
### MAIN
if __name__=='__main__':
verbose = 0
debug = 0
infile = "infile"
outfile = "outfile"
options_start = 3
# --- Parse two positional parameters ---
n1 = int(PosArg(1))
n2 = int(PosArg(2))
# --- Parse an optional parameters ---
present, a3 = OptArg(3,50)
n3 = int(a3)
options_start += int(present)
# --- Parse rest of command line ---
skip = 0
for i in range(options_start, len(sys.argv)):
if not skip:
if sys.argv[i][:2] == "-d": debug ^= 1
elif sys.argv[i][:2] == "-v": verbose ^= 1
elif sys.argv[i][:2] == "-i": (skip,infile) = NextArg(i)
elif sys.argv[i][:2] == "-o": (skip,outfile) = NextArg(i)
elif sys.argv[i][:2] == "-h": HelpAndExit()
elif sys.argv[i][:1] == "-": Fatal("'%s' unknown argument" % sys.argv[i])
else: Fatal("'%s' unexpected" % sys.argv[i])
else: skip = 0
print("Number 1 = %d" % n1)
print("Number 2 = %d" % n2)
print("Number 3 = %d" % n3)
print("Debug = %d" % debug)
print("verbose = %d" % verbose)
print("infile = %s" % infile)
print("outfile = %s" % outfile)