Есть ли способ контролировать порядок, в котором аргументы оцениваются в Python argparse - PullRequest
Новая
       28

Есть ли способ контролировать порядок, в котором аргументы оцениваются в Python argparse

3 голосов
/ 29 марта 2019

Давайте рассмотрим пример ниже.Парсер содержит два аргумента - входной файл и многословность.Функция Set_verbosity_level () используется для управления значением переменной уровня модуля / глобальной (в моей реальной жизни переменной уровня пакета) до 0-4.Функция CheckFile () выполняет тесты внутри входного файла (в реальной жизни в зависимости от типа).Я хотел бы печатать сообщения в CheckFile () в зависимости от многословия.Проблема в том, что argparse вызывает CheckFile () перед Set_verbosity_level (), поэтому уровень детализации всегда равен 0 / по умолчанию в CheckFile ...

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

    import argparse

    VERBOSITY = 0


    def Set_verbosity_level():
        """Set the verbosity level.
        """

        def type_func(value):
            a_value = int(value)
            globals()['VERBOSITY'] = value
            print("Verbosity inside Set_verbosity_level(): " + str(globals()['VERBOSITY']))
            return value

        return type_func


    class CheckFile(argparse.FileType):
        """
        Check whatever in the file
        """

        def __init__(self, mode='r', **kwargs):
            super(CheckFile, self).__init__(mode, **kwargs)

        def __call__(self, string):
            # Do whatever processing/checking/transformation
            # e.g print some message according to verbosity
            print("Verbosity inside  CheckFile(): " + str(globals()['VERBOSITY']))
            return super(CheckFile, self).__call__(string)


    def make_parser():
        """The main argument parser."""
        parser = argparse.ArgumentParser(add_help=True)

        parser.add_argument("-V",
                            "--verbosity",
                            default=0,
                            type=Set_verbosity_level(),
                            help="Increase output verbosity.",
                            required=False)

        parser.add_argument('-i', '--inputfile',
                            help="Input file",
                            type=CheckFile(mode='r'),
                            required=True)

        return parser


    if __name__ == '__main__':
        myparser = make_parser()
        args = myparser.parse_args()
        print("Verbosity in Main: " + str(VERBOSITY))

Вызов этого скрипта дает: 

$python test.py -i test.bed -V 2
Verbosity inside  CheckFile(): 0
Verbosity inside Set_verbosity_level(): 2
Verbosity in Main: 2

Ответы [ 2 ]

1 голос
/ 29 марта 2019

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

python test.py -V 2 -i test.bed

Это выводит: 

Verbosity inside Set_verbosity_level(): 2
Verbosity inside  CheckFile(): 2

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

1 голос
/ 29 марта 2019

Я не знаю, что вы можете принудительно прочитать переменную argparse, но вы можете использовать встроенные в синтаксический анализатор командной строки питоны в своей основной функции: 

import sys

# Your classes here #

if __name__ == '__main__':
    verbosity = 0
    for i, sysarg in enumerate(sys.argv):
        if str(sysarg).strip().lower().replace('-','') in ['v', 'verbose']:
            try:
                verbosity = sys.argv[i + 1]
            except IndexError:
                print("No verbosity level specified")
    # more code

Это не очень элегантно и не очень красиво, но это один из способов убедиться, что вы сначала получите многословие.

Вы также можете обновить класс CheckFile, добавив в него функцию проверки многословия: 

class CheckFile(argparse.FileType):
    """
    Check whatever in the file
    """

    def __init__(self, mode='r', **kwargs):
        super(CheckFile, self).__init__(mode, **kwargs)

    def _check_verbosity(self):
        verbosity = 0
        for i, sysarg in enumerate(sys.argv):
            if str(sysarg).strip().lower().replace('-','') in ['v', 'verbose']:
                try:
                verbosity = sys.argv[i + 1]
            except IndexError:
                print("No verbosity level specified")

        return verbosity

    def __call__(self, string):
        # Do whatever processing/checking/transformation
        # e.g print some message according to verbosity
        print("Verbosity inside  CheckFile(): {}".format(self._check_verbosity()))
        return super(CheckFile, self).__call__(string)

Опять же, я знаю, что это не совсем ответ на ваш argparse вопрос, но это решение вашей проблемы

...