Как рекурсивно обойти словарь в python?

Question

[Андрей]

Помогите с рекурсией, пока изучаю не вник...
Есть список словарей [products], в каждом словаре есть вложенный список [products], который в свою очередь так же имеет (а может и не имеет вложенный список [products]).
Необходимо вывести на печать название каждого элемента текущего [products] и родительского [products]. Если в циклах, это примерно выглядит так:

for prod in products:
    if len(prod['products']) == 0:
        print(main_name['name'])
    else:
        for y in prod['products']:
            if len(y['products']) == 0:
                print(main_name['name'], y['name'])
            else:
                for x in y['products']:
                    if len(x['products']) == 0:
                        print(y['name'], x['name'])
                    else:
                        for w in x['products']:
                            if len(w['products']) == 0:
                                print(y['name'], w['name'])
                             else ...

Уровень вложенности ['products'] не известен, как оформить это в виде рекурсии?
я пробую:

def search_prod(obj):
    if len(obj['products']) == 0:
        return obj['name']
    for y in obj['products']:
        search_prod(y)

for i in products:
    print(search_prod(i))

Но получают None (кроме случаев когда в первом вхождении len(prod['products']) == 0)

Comments

[Сергей Горностаев]

Нужно именно рекурсивное решение? Эффективнее сплющить вложенные списки в один генератором.

Answer 1

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

def recursion(products, parent_name):
	for product in products:
		name = product['name']
		print(f'Текущий - {name}, Родитель - {parent_name}')
		sub_products = product.get('products', [])
		if sub_products:
			recursion(sub_products, name)

recursion(products, 'root')

products - твой список со словарями, name - это как будет называться твой список, в примере я указал 'root', типа корневой элемент.

Поиграйся с этим кодом

Comments

[Андрей]

О, это круто... а если мне понадобится не принтовать а return name и parent_name в переменные, типа: x, y = recursion(products, 'root')?

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

тогда нужны генераторы. Почитай про yield и yield from и тогда поймешь что делает этот код:

def recursion(products, parent_name):
  for product in products:
    name = product['name']
    yield f'Текущий - {name}', f'Родитель - {parent_name}'
    sub_products = product.get('products', [])
    if sub_products:
      yield from recursion(sub_products, name)

for x,y in recursion(products, 'root'):
	print(x,y)

[Андрей]

Спасибо!

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

Андрей, отмечай решением