Как правильно обработать None для аргументов в argparse?

Question

[DarkWood]

Здравствуйте.

Хочу сделать скрипт с возможностью управления аргументами, передаваемыми из командной строки. Использую для этого модуль argparse. На простом тестовом примере столкнулся с тем, что не понимаю как обработать None для тех аргументов, которые не были переданы в конкретный вызов.

У скрипта есть поведение по умолчанию, если аргументы не были переданы совсем. Потом уже идет их вызов. Но если хотя бы один дан не был, то я получаю логичную ошибку о том, что None в таком-то и таком-то случае использовать нельзя.

Собственно, вопрос: как организовать проверку сего момента? Или мною был выбран вовсе неправильный подход?

import argparse
import json
import sys


def add_item(string):
    k, v = string.split(",")
    with open("test.json", "r", encoding="utf8") as in_file:
        db = json.load(in_file)
        if k in db:
            db[k] = v
        else:
            db[k] = v
    with open("test.json", "w", encoding="utf8") as out_file:
        json.dump(db, out_file, ensure_ascii=False)


def check_item(string):
    with open("test.json", "r", encoding="utf8") as in_file:
        db = json.load(in_file)
    if string in db:
        print(db[string])
    else:
        print("Don't found key: " + string)


parser = argparse.ArgumentParser()

parser.add_argument('-a', '--add', action='store', dest='add_value',
                    help='Add value to dictionary')

parser.add_argument('-v', '--value', action='store', dest='check_value',
                    help='Check value in dictionary')

if __name__ == "__main__":
    namespace = parser.parse_args(sys.argv[1:])
    if not len(sys.argv) > 1:
        with open("test.json", "r", encoding="utf8") as in_file:
            database = json.load(in_file)
        for key, value in database.items():
            print(key + " - " + value)
    else:
        add_item(namespace.add_value)
        check_item(namespace.check_value)

Comments

[longclaps]

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

if k in db:
    db[k] = v
else:
    db[k] = v

Рука-лицо. Да ладно уж, вываливай что попало.

[DarkWood]

longclaps,
Во-первых, какое отношение ваш комментарий имеет к сути?
Во-вторых, что вас не устраивает в процитированном куске кода?

[longclaps]

DarkWood,
Во-первых, наш комментарий имеет прямое отношение к сути (кстати, о какой сути идёт речь?).
Во-вторых, в процитированном куске кода нас не устраивает идиотизм. Читать такое неловко, а вываливать это сюда - стыдно.

[DarkWood]

Я специализируюсь на Adobe Illustrator и мне часто тоже приходится читать кучу всякого странного в вопросах пользователей. Включая идиотизм. Для ответа я вижу три варианта: корректно объяснить в чем ошибка помимо основного вопроса, если таковая была; пропустить и ответить по существу; написать подобное вашему. Можно еще в Google отправить, если вещь совсем ясельная.

Я тоже иногда отвечаю как вы сейчас. Но количество идиотизма это не уменьшает. Потому чаще всё-таки пропускаю мимо. И прошу вас понять, что по теме Python он тоже не уменьшиться, что бы вы ни писали. Учитывая, что язык доступен из каждого утюга - количество идиотизма в обозримом будущем будет только расти.

Ну и вернемся к моему вопросу. Раз вы спрашиваете в чем суть, что, возможно, не потрудились прочитать его описание. Вопрос заключается не в работе со словарями, а в правильной организации взаимодействия с несколькими аргументами командной строки, когда был вызван только один из них, а другие выдают невозможность обработки значения None. Я честно написал, что не понимаю как это сделать.

[longclaps]

DarkWood, давайте так: я посмотрю ваш код и попытаюсь понять, что у вас там неладно. А вы, ради бога, вбейте себе в голову:

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

Так вот, вы не умеете писать сложносочинённые предложения. В этой бредятине черт ногу сломит. Пишите короткими ясными фразами. Как я.

[DarkWood]

Код выше, описание выше, вопрос выше. Пояснения в этим комментариях и к ответу ниже.

Не ищите здесь скрытых камней. Я хочу понять ровно то, что написал: как оперировать несколькими аргументами в argparse.

[longclaps]

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

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--add', type=str, nargs='+')
parser.add_argument('--check', type=str, nargs='+')
args = parser.parse_args()
########
# вот тут хоть обчитайся своих джейсонов, а у меня просто словарик
db = {'bla': 'бла'}
########

for k in args.check:
    print(k, ':', db.get(k))

print('-----------------')

for s in args.add:
    k, v = s.split('=')
    db[k] = v

print(db)

командная строка:
> prog.py --add qwe=rty abc=42 --check bla what
выхлоп:

bla : бла
what : None
-----------------
{'bla': 'бла', 'qwe': 'rty', 'abc': '42'}

[DarkWood]

longclaps, да, читать - не ваш конек. Вы опять зачем-то пишите мне про словари. И зачем-то переписываете функции, которые несут в моем вопросе минимум смысловой нагрузки. Тем временем, не вызванный аргумент по-прежнему будет отдавать ошибку обработки None. Например,

prog.py --add qwe=rty
Traceback (most recent call last):
  File "E:\Useful\python_work\my_projects\video_watcher\argparse_test.py", line 12, in <module>
    for k in args.check:
TypeError: 'NoneType' object is not iterable

Так что я бы поспорил с тем кому "трудно втыкать в несколько вещей".

[longclaps]

DarkWood, мой друг, попробуй быть чуть менее тупым, чем обычно.
- Что делает json.load(in_file)?
- Он считывает in_file в словарь.
- Предоставил ты мне этот файл?
- Нет.
- А что ты мне дал?
- Ты дал мне какой-то код, который падает, пытаясь итерироваться по значениям для несущнствующег ключа.

Я не "переписал" этот бред (да, бред), а просто набросал решение, силясь понять твой замысел. Ну что, двинемся дальше? Выложишь свой test.json? Выложишь свой список аргументов командной строки? Сообщение об ошибке уже есть, спасибо - и да, если его подавить - поползут другие ошибки этого странного кода.
И да - аргументы (и функций, и командной строки) не вызывают - привыкай уже.

[DarkWood]

longclaps, скажите, пожалуйста, вы умеете читать? А то, может, и правда нет?

Вы понимаете значение слова "тестовый"?

Зачем вы мне рассказываете что делает json.load? Зачем вам файлы, которые не имеют никакого значения для вопроса? Как и представленные функции, повторюсь уже в который раз.

Код до использования аргументов может быть совершенно любым. Сути вопроса это не изменит.

import argparse
import sys

parser = argparse.ArgumentParser()

parser.add_argument('-u', '--up')
parser.add_argument('-l', '--low')

if __name__ == "__main__":
    namespace = parser.parse_args(sys.argv[1:])
    print(namespace.up.upper())

При передаче одного аргумента всё отработает прекрасно.

prog.py -u test
TEST

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

import argparse
import sys

parser = argparse.ArgumentParser()

parser.add_argument('-u', '--up')
parser.add_argument('-l', '--low')

if __name__ == "__main__":
    namespace = parser.parse_args(sys.argv[1:])
    print(namespace.up.upper())
    print(namespace.low.lower())

prog.py -u test
Traceback (most recent call last):
  File "prog.py", line 12, in <module>
    print(namespace.low.lower())
AttributeError: 'NoneType' object has no attribute 'lower'

Я вам еще раз повторюсь, что вопрос не про функции и словари, а про None для тех аргументов командной строки, которые не были переданы в конкретный вызов. Я хочу, чтобы переданный отработал спокойно, а не переданные не вызывали ошибок.

[longclaps]

Я хочу, чтобы переданный отработал спокойно, а не переданные не вызывали ошибок.

DarkWood, специально для тебя:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-u', '--up')
parser.add_argument('-l', '--low', default='MUDAK')

if __name__ == "__main__":
    namespace = parser.parse_args()
    print(namespace.up.upper())
    print(namespace.low.lower())

[DarkWood]

longclaps, только в этом случае аргумент, которому не передавалось значение из командной строки, будет отрабатывать со значением по умолчанию. Мне нужно четко: было передано значение аргументу - он работает, не было передано - не работает ни в каком виде. И я знаю, что так можно.

Возьмем, например, ffmpeg. Там куча аргументов. И каждый из них работает только если ему было передано значение.

[longclaps]

DarkWood,

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-u', '--up')
parser.add_argument('-l', '--low')

if __name__ == "__main__":
    namespace = parser.parse_args()
    if namespace.up is not None:
        print(namespace.up.upper())
    if namespace.low is not None:
        print(namespace.low.lower())

[DarkWood]

longclaps, спасибо. Правда, я ожидал чего-то более элегантного.

Answer 1

[o5a]

else:
        add_item(namespace.add_value)
        check_item(namespace.check_value)

Обе запускаются безусловно, независимо от заполнения параметров в namespace. Так что или добавить условие заполненности соответствующего параметра add_value/check_value, и потом запускать соответствующую процедуру, или же, если идея была всегда запускать отображение значения, то вместо разделения на add/check_value, сразу распарсить ввод на param, value и уже их передавать в свою функцию, например так

else:
        param, value = namespace.add_value.split(",") if namespace.add_value else [namespace.check_value, None]
        add_item(param, value)
        check_item(param)

И соответственно изменить add_item/check_item, чтобы работали уже с параметром как таковым.

Comments

[DarkWood]

Спасибо за ответ. Но приведенный пример именно тестовый, на котором я пытался понять как работают несколько аргументов в argparse. В реальном проекте аргументов будет около десятка. Соответственно, как я понимаю, мне нужно будет проверять какие были переданы? Вы написали "заполнение параметров в namespace" - это довольно точно описывает то, что я хочу понять. Если аргумент не вызывался и имеет None в namespace, то я не получаю никаких ошибок, а вызванные спокойно отрабатывают.