Когда дело доходит до создания интерфейсов командной строки (CLI) в Python, модуль argparse является мощным инструментом, упрощающим анализ аргументов. Одним из распространенных требований является предоставление значений по умолчанию для аргументов. В этой статье блога мы рассмотрим различные методы установки значений по умолчанию в виде массивов в argparse, дополненные разговорными объяснениями и примерами кода.

Метод 1: использование параметра «default».
Самый простой способ назначить массив в качестве значения по умолчанию в argparse — использовать параметр «default». Давайте рассмотрим сценарий, в котором мы хотим принять список имен в качестве аргумента со значением по умолчанию [‘Джон’, ‘Джейн’]:

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--names', nargs='*', default=['John', 'Jane'])
args = parser.parse_args()
print(args.names)

В этом примере, если мы запустим скрипт без указания имен в качестве аргументов, он будет использовать значение по умолчанию [‘Джон’, ‘Джейн’].

Метод 2. Использование настраиваемого действия.
Другой подход заключается в определении класса настраиваемого действия, который обрабатывает анализ аргументов и устанавливает значение по умолчанию в виде массива. Вот пример:

import argparse
class ArrayAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        setattr(namespace, self.dest, values.split(','))
parser = argparse.ArgumentParser()
parser.add_argument('--names', action=ArrayAction, default=['John', 'Jane'])
args = parser.parse_args()
print(args.names)

С помощью этого метода мы определяем класс настраиваемого действия (ArrayAction), который разбивает входную строку запятыми и присваивает полученный список нужному атрибуту.

Метод 3: использование лямбда-функции
Вы также можете использовать лямбда-функцию, чтобы установить массив в качестве значения по умолчанию в argparse. Вот пример:

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--names', nargs='*', default=lambda: ['John', 'Jane'])
args = parser.parse_args()
print(args.names)

Используя лямбда-функцию, мы можем отложить вычисление значения по умолчанию до тех пор, пока оно не понадобится, что позволяет нам динамически назначать массив.

Метод 4: использование вспомогательной функции
Если вы предпочитаете более модульный подход, вы можете определить вспомогательную функцию, которая возвращает значение массива по умолчанию. Вот пример:

import argparse
def get_default_names():
    return ['John', 'Jane']
parser = argparse.ArgumentParser()
parser.add_argument('--names', nargs='*', default=get_default_names())
args = parser.parse_args()
print(args.names)

В этом методе функция get_default_names()инкапсулирует логику получения значения по умолчанию, что делает код более читабельным и удобным в обслуживании.

В этой статье мы рассмотрели несколько методов установки массива в качестве значения по умолчанию в argparse. Используя параметр «default», пользовательские действия, лямбда-функции и вспомогательные функции, вы можете адаптировать argparse для обработки аргументов массива по умолчанию в ваших CLI Python. Вооружившись этими методами, вы теперь можете с легкостью создавать более гибкие и надежные интерфейсы командной строки.

Не забудьте выбрать метод, который лучше всего подходит для вашего случая использования, и наслаждайтесь гибкостью и мощью argparse в своих проектах Python!