Полное руководство по обработке аргументов командной строки в Python

Расширенные примеры обработки аргументов командной строки

Пример 1: argparse с подкомандами и взаимно исключающими опциями

Данный пример демонстрирует создание CLI с двумя подкомандами (start, stop) и взаимоисключающими флагами (-s для start, -x для stop).

import argparse

parser = argparse.ArgumentParser(description='Управление сервисом')
subparsers = parser.add_subparsers(dest='command', help='Доступные команды')

# Подкоманда start
start_parser = subparsers.add_parser('start', help='Запуск сервиса')
start_parser.add_argument('-d', '--daemon', action='store_true', help='Запустить в фоне')

# Подкоманда stop
stop_parser = subparsers.add_parser('stop', help='Остановка сервиса')
stop_parser.add_argument('-f', '--force', action='store_true', help='Принудительная остановка')

# Взаимоисключающие опции для главной команды
group = parser.add_mutually_exclusive_group()
group.add_argument('-s', '--start', action='store_true', help='Запуск')
group.add_argument('-x', '--stop', action='store_true', help='Остановка')

args = parser.parse_args()
if args.command == 'start':
    print('Команда start с флагом daemon =', args.daemon)
elif args.command == 'stop':
    print('Команда stop с флагом force =', args.force)
else:
    print('Общие флаги: start =', args.start, ', stop =', args.stop)

Результат выполнения:

$ python example1.py start -d
Команда start с флагом daemon = True

$ python example1.py -s
Общие флаги: start = True , stop = False

$ python example1.py -s -x
usage: example1.py [-h] [-s | -x] {start,stop} ...
example1.py: error: argument -x: not allowed with argument -s

Пример 2: click с группами команд и опциями с выбором из списка

Используется click версии 8.x. Показывает создание группы команд (config, run) и ограничение значений через click.Choice.

import click

@click.group()
def cli():
    pass

@cli.command()
@click.option('--mode', type=click.Choice(['dev', 'prod'], case_sensitive=False),
              default='dev', help='Режим работы')
def config(mode):
    """Настройка режима работы"""
    click.echo(f'Выбран режим: {mode}')

@cli.command()
@click.argument('script', type=click.Path(exists=True))
@click.option('--iterations', default=10, help='Количество итераций')
def run(script, iterations):
    """Запуск скрипта с заданными итерациями"""
    click.echo(f'Запуск {script} с {iterations} итерациями')

if __name__ == '__main__':
    cli()

$ python example2.py config --mode prod
Выбран режим: prod

$ python example2.py run my_script.py --iterations 5
Запуск my_script.py с 5 итерациями

$ python example2.py config --mode staging
Usage: example2.py config [OPTIONS]
Try 'example2.py config --help' for help.
Error: Invalid value for '--mode': 'staging' is not one of 'dev', 'prod'.

Пример 3: docopt с обработкой позиционных аргументов и флагов

Пример демонстрирует парсинг текстовой справки, где аргументы могут повторяться (как в командах копирования).

"""
Копирование файлов.

Использование:
  copy <source> <dest> [--backup]
  copy --list

Опции:
  -b, --backup  Создать резервную копию
  --list        Показать список файлов
"""
from docopt import docopt

if __name__ == '__main__':
    args = docopt(__doc__)
    if args['--list']:
        print('Вывод списка файлов')
    else:
        print(f'Копирование из {args["<source>"]} в {args["<dest>"]}')
        if args['--backup']:
            print('Создание резервной копии')

$ python example3.py file1.txt file2.txt --backup
Копирование из file1.txt в file2.txt
Создание резервной копии

$ python example3.py --list
Вывод списка файлов

Пример 4: sys.argv и ручной разбор с обработкой ошибок

Полный разбор аргументов через sys.argv без сторонних модулей: поддержка ключа -o и позиционного файла.

import sys

def main():
    argv = sys.argv[1:]
    if not argv:
        print('Ошибка: не указан файл. Использование: script.py  [-o out]')
        sys.exit(1)
    
    output = None
    positional = []
    i = 0
    while i < len(argv):
        if argv[i] == '-o':
            i += 1
            if i >= len(argv):
                print('Ошибка: после -o ожидается значение')
                sys.exit(1)
            output = argv[i]
        else:
            positional.append(argv[i])
        i += 1
    
    if not positional:
        print('Ошибка: не указан источник')
        sys.exit(1)
    
    print('Файл:', positional[0])
    if output:
        print('Выход:', output)

if __name__ == '__main__':
    main()

$ python example4.py data.txt -o result.log
Файл: data.txt
Выход: result.log

$ python example4.py
Ошибка: не указан файл. Использование: script.py  [-o out]

Пример 5: getopt с обработкой нескольких значений для одной опции

Показывает, как с помощью getopt можно принимать несколько аргументов под одну опцию, используя повторение опции.

import sys, getopt

try:
    opts, args = getopt.getopt(sys.argv[1:], 'i:', ['input='])
except getopt.GetoptError as err:
    print(err)
    sys.exit(2)

inputs = []
for opt, arg in opts:
    if opt in ('-i', '--input'):
        inputs.append(arg)

print('Входные файлы:', inputs)
print('Позиционные аргументы:', args)

$ python example5.py -i file1.txt -i file2.txt --input file3.txt other
Входные файлы: ['file1.txt', 'file2.txt', 'file3.txt']
Позиционные аргументы: ['other']