Реализовываете интерфейсы командной строки в стиле "[команда] [действие] [параметр]"?

Какой самый "чистый" способ реализовать интерфейс командной строки, похожий на git, например:

git push origin/master
git remote add origin git://example.com master

В идеале также позволяет более гибкий анализ, например,

jump_to_folder app theappname v2
jump_to_folder app theappname source
jump_to_folder app theappname source v2
jump_to_folder app theappname build v1
jump_to_folder app theappname build 1
jump_to_folder app theappname v2 build

jump_to_folder это имя сценария, app это команда, theappname является параметром "fixed-location", "build" и "v2" и т. д. являются аргументами (например, возможные аргументы могут быть любым числом / любым числом с префиксом a v или build/source/tmp/config)

Я мог бы просто вручную проанализировать аргументы с серией if/else/elifs, но должен быть более элегантный способ сделать это?

В качестве теоретического примера я мог бы описать схему пользовательского интерфейса.

app:
    fixed: application_name

    optional params:
        arg subsection:
            "build"
            "source"
            "tmp"
            "config"

        arg version:
            integer
            "v" + integer

Затем проанализируйте предоставленные аргументы через схему выше и получите словарь:

>>> print schema.parse(["app", "theappname", "v1", "source"])
{
    "application_name": "theappname",
    "params":{
        "subsection": "source",
        "version":"v1"
    }
}

Существует ли такая система? Если нет, то как бы я реализовал что-то подобное?

6 ответов

Решение

Для этого идеально подходит argparse, в частности " подкоманды " и позиционные аргументы

import argparse


def main():
    arger = argparse.ArgumentParser()

    # Arguments for top-level, e.g "subcmds.py -v"
    arger.add_argument("-v", "--verbose", action="count", default=0)

    subparsers = arger.add_subparsers(dest="command")

    # Make parser for "subcmds.py info ..."
    info_parser = subparsers.add_parser("info")
    info_parser.add_argument("-m", "--moo", dest="moo")

    # Make parser for "subcmds.py create ..."
    create_parser = subparsers.add_parser("create")
    create_parser.add_argument("name")
    create_parser.add_argument("additional", nargs="*")

    # Parse
    opts = arger.parse_args()

    # Print option object for debug
    print opts

    if opts.command == "info":
        print "Info command"
        print "--moo was %s" % opts.moo

    elif opts.command == "create":
        print "Creating %s" % opts.name
        print "Additional: %s" % opts.additional

    else:
        # argparse will error on unexpected commands, but
        # in case we mistype one of the elif statements...
        raise ValueError("Unhandled command %s" % opts.command)


if __name__ == '__main__':
    main()

Это можно использовать так:

$ python subcmds.py create myapp v1 blah
Namespace(additional=['v1', 'blah'], command='create', name='myapp', verbose=0)
Creating myapp
Additional: ['v1', 'blah']
$ python subcmds.py info --moo
usage: subcmds.py info [-h] [-m MOO]
subcmds.py info: error: argument -m/--moo: expected one argument
$ python subcmds.py info --moo 1
Namespace(command='info', moo='1', verbose=0)
Info command
--moo was 1

cmd модуль, вероятно, будет хорошо работать для этого.

Пример:

import cmd

class Calc(cmd.Cmd):
    def do_add(self, arg):
        print sum(map(int, arg.split()))

if __name__ == '__main__':
    Calc().cmdloop()

Запустить его:

$python calc.py
(Cmd) add 4 5
9
(Cmd) help

Undocumented commands:
======================
add  help

(Cmd)

См. Документацию по Python или сайт PyMOTW для получения дополнительной информации.

Прямо из одного из моих сценариев:

import sys

def prog1_func1_act1(): print "pfa1"
def prog2_func2_act2(): print "pfa2"

commands = {
    "prog1 func1 act1": prog1_func1_act1,
    "prog2 func2 act2": prog2_func2_act2
}

try:
    commands[" ".join(sys.argv[1:])]()
except KeyError:
    print "Usage: ", commands.keys()

Это довольно быстрое и грязное решение, но отлично подходит для моего использования. Если бы я немного его очистил, я бы добавил argparse в микс для разбора позиционных и ключевых аргументов.

В Python есть модуль для разбора параметров командной строки, optparse.

Возможно, вы захотите взглянуть на скалы - Framework Formulation Framework командной строки

Вот мое предложение.

  1. Измените свою грамматику немного.

  2. Используйте optparse.

В идеале также позволяет более гибкий анализ, например,

jump_to_folder -n theappname -v2 cmd 
jump_to_folder -n theappname cmd source 
jump_to_folder -n theappname -v2 cmd source 
jump_to_folder -n theappname -v1 cmd build 
jump_to_folder -n theappname -1 cmd build 
jump_to_folder -n theappname -v2 cmd build

Тогда у вас есть 1 или 2 аргумента: команда всегда является первым аргументом. Это необязательный аргумент - всегда второй аргумент.

Все остальное - варианты, в произвольном порядке.

Другие вопросы по тегам