опция argparse для передачи списка в качестве опции

Я предпочитаю передавать строку с разделителями, которую я разбираю позже в скрипте. Причины этого есть; список может быть любого типа int или str, а иногда и с помощью nargs Я столкнуться с проблемами, если есть несколько необязательных аргументов и аргументы.

parser = ArgumentParser()
parser.add_argument('-l', '--list', help='delimited list input', type=str)
args = parser.parse_args()
my_list = [int(item) for item in args.list.split(',')]

затем,

python test.py -l "265340,268738,270774,270817" [other arguments]

или

python test.py -l 265340,268738,270774,270817 [other arguments]

будет работать нормально. Разделителем может быть пробел, который хотя бы применять кавычки вокруг значения аргумента, как в Примере вопрос.

дополнительно nargs, вы можете использовать choices если вы знаете список заранее:

>>> parser = argparse.ArgumentParser(prog='game.py')
>>> parser.add_argument('move', choices=['rock', 'paper', 'scissors'])
>>> parser.parse_args(['rock'])
Namespace(move='rock')
>>> parser.parse_args(['fire'])
usage: game.py [-h] {rock,paper,scissors}
game.py: error: argument move: invalid choice: 'fire' (choose from 'rock',
'paper', 'scissors')

если вы собираетесь сделать один переключатель принимать несколько параметров, то вы используете nargs='+'. Если ваш пример '- l ' На самом деле принимает целые числа:

a = argparse.ArgumentParser()
a.add_argument(
    '-l', '--list',  # either of this switches
    nargs='+',       # one or more parameters to this switch
    type=int,        # /parameters/ are ints
    dest='list',     # store in 'list'.
    default=[],      # since we're not specifying required.
)

print a.parse_args("-l 123 234 345 456".split(' '))
print a.parse_args("-l 123 -l=234 -l345 --list 456".split(' '))

производит

Namespace(list=[123, 234, 345, 456])
Namespace(list=[456])  # Attention!

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

альтернативой является использование append действие:

a = argparse.ArgumentParser()
a.add_argument(
    '-l', '--list',  # either of this switches
    type=int,        # /parameters/ are ints
    dest='list',     # store in 'list'.
    default=[],      # since we're not specifying required.
    action='append', # add to the list instead of replacing it
)

print a.parse_args("-l 123 -l=234 -l345 --list 456".split(' '))

которая производит

Namespace(list=[123, 234, 345, 456])

или вы можете написать пользовательский обработчик / действие чтобы проанализировать значения, разделенные запятыми, чтобы вы могли сделать

-l 123,234,345 -l 456

на add_argument(),type - это просто вызываемый объект, который получает строку и возвращает значение параметра.

import ast

def arg_as_list(s):                                                            
    v = ast.literal_eval(s)                                                    
    if type(v) is not list:                                                    
        raise argparse.ArgumentTypeError("Argument \"%s\" is not a list" % (s))
    return v                                                                   


def foo():
    parser.add_argument("--list", type=arg_as_list, default=[],
                        help="List of values")

Это позволит:

$ ./tool --list "[1,2,3,4]"

используя параметр nargs в методе add_argument от argparse

Я использую nargs='' в качестве параметра add_argument. Я специально использовал nargs=' ' к опции выбора значений по умолчанию, если я не передаю никаких явных аргументов

включая фрагмент кода в качестве примера:

пример: temp_args1.py

Обратите Внимание: приведенный ниже пример кода написан на python3. Путем изменения инструкции печати формат, может работать в python2

    #!/usr/local/bin/python3.6

    from argparse import ArgumentParser

    description = 'testing for passing multiple arguments and to get list of args'
    parser = ArgumentParser(description=description)
    parser.add_argument('-i', '--item', action='store', dest='alist',
                        type=str, nargs='*', default=['item1', 'item2', 'item3'],
                        help="Examples: -i item1 item2, -i item3")
    opts = parser.parse_args()

    print("List of items: {}".format(opts.alist))

Примечание: я собираю несколько строковых аргументов, которые хранятся в списке - opts.алист Если вам нужен список целых чисел, измените параметр type на parser.add_argument в int

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

    python3.6 temp_agrs1.py -i item5 item6 item7
    List of items: ['item5', 'item6', 'item7']

    python3.6 temp_agrs1.py -i item10
    List of items: ['item10']

    python3.6 temp_agrs1.py
    List of items: ['item1', 'item2', 'item3']

может быть самый простой ответ

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-l", "--tolist", help="input to list", action="store_true")
parser.add_argument("newlist", type=str, help="generate a list")
args = parser.parse_args()
if args.tolist:
    print(args.newlist.split(" "))