как заставить argparse читать аргументы из файла с опцией, а не с префиксом

Вы можете решить эту проблему с помощью пользовательского argparse.Action, который открывает file, анализирует содержимое файла, а затем добавляет аргументы.

Например, это будет очень простое действие:

class LoadFromFile (argparse.Action):
    def __call__ (self, parser, namespace, values, option_string = None):
        with values as f:
            # parse arguments in the file and store them in the target namespace
            parser.parse_args(f.read().split(), namespace)

Что вы можете использовать следующим образом:

parser = argparse.ArgumentParser()
# other arguments
parser.add_argument('--file', type=open, action=LoadFromFile)
args = parser.parse_args()

Результирующее пространство имен в args будет также содержать любую конфигурацию, которая также была загружена из файла, где файл содержал аргументы с тем же синтаксисом, что и в командной строке (например, --foo 1 --bar 2).

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

def __call__ (self, parser, namespace, values, option_string=None):
    with values as f:
        contents = f.read()

    # parse arguments in the file and store them in a blank namespace
    data = parser.parse_args(contents.split(), namespace=None)
    for k, v in vars(data).items():
        # set arguments in the target namespace if they haven’t been set yet
        if getattr(namespace, k, None) is not None:
            setattr(namespace, k, v)

Конечно, вы также можете немного усложнить чтение файла, например сначала прочитать из JSON.

Вы прокомментировали это

Мне нужно иметь возможность написать свою собственную функцию, чтобы читать этот файл и возвращать аргументы (это не в формате один аргумент на строку) -

В существующем обработчике файла префикса есть возможность изменить способ чтения файла. Файл читается «приватным» методом parser._read_args_from_files, но он вызывает простой общедоступный метод, который преобразует строку в строку, действие по умолчанию с одним аргументом на строку:

def convert_arg_line_to_args(self, arg_line):
    return [arg_line]

Это было написано таким образом, чтобы вы могли легко настроить его. https://docs.python.org/3/library/argparse.html#argparse.ArgumentParser.convert_arg_line_to_args

Полезным переопределением этого метода является то, что каждое слово, разделенное пробелом, рассматривается как аргумент:

def convert_arg_line_to_args(self, arg_line):
    for arg in arg_line.split():
        if not arg.strip():
            continue
        yield arg

В файле модульного тестирования test_argparse.py есть тестовый пример для этой альтернативы.

Но если вы по-прежнему хотите инициировать это чтение с параметром аргумента, а не с символом префикса, то подход с пользовательским действием является хорошим.

Однако вы могли бы написать свою собственную функцию, которая обрабатывает argv до того, как она будет передана в parser. Его можно было бы смоделировать по образцу parser._read_args_from_files.

Таким образом, вы можете написать функцию, например:

def read_my_file(argv):
    # if there is a '-A' string in argv, replace it, and the following filename
    # with the contents of the file (as strings)
    # you can adapt code from _read_args_from_files
    new_argv = []
    for a in argv:
        ....
        # details left to user
    return new_argv

Затем вызовите свой парсер с помощью:

parser.parse_args(read_my_file(sys.argv[1:]))

И да, это можно обернуть в подкласс ArgumentParser.

Action при вызове получает parser и namespace среди своих аргументов.

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

class ArgfileAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        extra_args = <parse_the_file>(values)
        #`namespace' is updated in-place when specified
        parser.parse_args(extra_args,namespace)