Работа с argparse: простой пример

В Python есть модуль под названием argparse, который позволяет разработчикам принимать аргументы командной строки в своих программах. Он обеспечивает простой и удобный способ обработки аргументов командной строки, которые, как правило, используются для настройки поведения программы.

Рассмотрим простой пример использования argparse. Представьте, что вы создаете небольшую утилиту командной строки, которая должна вести себя по-разному в зависимости от полученного аргумента. Например, если аргумент командной строки равен A, программа должна выполнить определенное действие, если аргумент равен B, она должна выполнить другое действие, а если аргумент отсутствует, программа должна выводить справку и завершать работу.

Вот как это можно реализовать с помощью argparse:

import argparse

# Создаем парсер аргументов
parser = argparse.ArgumentParser()

# Добавляем аргумент
parser.add_argument('arg', choices=['A', 'B'], help='Аргумент должен быть либо A, либо B')

# Парсим аргументы
args = parser.parse_args()

# В зависимости от полученного аргумента выполняем нужное действие
if args.arg == 'A':
    print('Выполняем действие A')
elif args.arg == 'B':
    print('Выполняем действие B')

В этом коде мы создаем парсер аргументов, добавляем в него один аргумент, который может принимать значения A или B, а затем парсим аргументы командной строки. Если аргумент равен A, выполняется одно действие, если B — другое. Если аргумент отсутствует или его значение не является ни A, ни B, argparse автоматически выводит справку и завершает программу.

Таким образом, argparse является мощным инструментом для обработки аргументов командной строки в программах на Python.