Как обойти словарь в Python?

Question

[Андрей]

Есть многоуровневый словарь, из которого нужно забрать определенные значения (если есть). Как можно укоротить такую конструкцию?

for doc in article['key_1']['key_2']:
            if doc.get('attachment'):
                if doc['attachment'].get('file'):
                    if doc['attachment']['file'] == 'val_1':
                        name = doc['attachment']['key_4']

P.S. вариант рабочий, но думаю должно быть более элегантное решение...

Comments

[Viktor T2]

get же можно последовательно брать

if doc.get('attachment').get('file') == 'val_1':

.

[Николай Карлов]

Viktor T2, от None нельзя брать get()

d = {'a': {'b': None}}
d.get('a').get('b').get('c')
Traceback (most recent call last):
  File "<pyshell#3>", line 1, in <module>
    d.get('a').get('b').get('c')
AttributeError: 'NoneType' object has no attribute 'get'

[Николай Карлов]

Вы бы привели пример структуры данных. Что-то типа такого, только как это выглядит в вашей задаче:

article = {
    'key_1': {
        'key_2': {
            'content': {},
            'attachment': {
                'file': 'val_1',
                'key_4': 'some_value',
            },
        },
    },
}

А то мозги с утра не хотят додумывать...) Вы пишите про многоуровневый словарь, но не закрался ли туда список словарей в 'key_2'?

[Алан Гибизов]

Приведите в вопросе пример словаря, по которому идёт поиск.

[Viktor T2]

Николай Карлов,
Тогда так, через try

try:
    if doc.get('attachment').get('file') == 'val_1':
        name = doc['attachment']['key_4']
    print('Success', name)
except:
    name = None
    print('Bad', name)

или так:

try:
    if doc['attachment']['file'] == 'val_1':
        name = doc['attachment']['key_4']
    print('Success', name)
except:
    name = None
    print('Bad', name)

[Андрей]

Николай Карлов, нет, тогда бы я начал пример как for doc in article['key_1']

[Игорь Маркин]

Николай Карлов, .get() вторым параметром принимает значение по-умолчанию. Можно отдавать пустой словарь вместо None, тогда не будет ошибки.

Answer 1

[Игорь Маркин]

Вариант

article = article.get('key_1', {}).get('key_2')
if article:           
    for doc in article:
        attachment = doc.get('attachment', {})
        if attachment.get('file') == 'val_1':
            name = attachment.get('key_4')

Answer 2

[Михаил Кростелев]

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

def get_value_from_dict(dct, path):
    if not path:
        return
    
    key = path[0]
    new_path = path[1:]
    if curr_value := dct.get(key, None):
        if new_path:
            if type(curr_value) == dict:
                if new_path:
                    return get_value_from_dict(curr_value, new_path)
            else:
                return
        else:
            return curr_value

   

d={
    "lvl1_1": {
        "lvl2_1": {
            "lvl3_1": "value",
            "lvl3_2": {
                "lvl4_1": "value"
            },
        "lvl2_2": "value"
        }
    }
} 

print(get_value_from_dict(d, ["lvl1_1", "lvl2_1", "lvl3_2", "lvl4_1"]))

Возможно там есть некоторые ошибки, наверняка можно было более красиво написать, но поиграйтесь с этим кодом.