heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

extract_items: allow extracting items that are normally dug into for more

Browse Source 
By checking first if it's in item_types rather than checking if it can be dug into first.
For example: this allows extracting things like sectionListRenderer
This commit is contained in:
James Taylor 2019-12-26 19:39:48 -08:00
parent 8e8a1b70b6
commit 0c6a37e9aa
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
youtube/yt_data_extract/common.py
View File

@ -376,8 +376,12 @@ def extract_items_from_renderer(renderer, item_types=_item_types):
# mode 2: dig into the current renderer # mode 2: dig into the current renderer
key, value = list(renderer.items())[0] key, value = list(renderer.items())[0]
# the renderer is an item
if key in item_types:
items.append(renderer)
# has a list in it, add it to the iter stack # has a list in it, add it to the iter stack
if key in nested_renderer_list_dispatch: elif key in nested_renderer_list_dispatch:
renderer_list, continuation = nested_renderer_list_dispatch[key](value) renderer_list, continuation = nested_renderer_list_dispatch[key](value)
if renderer_list: if renderer_list:
iter_stack.append(current_iter) iter_stack.append(current_iter)
@ -390,10 +394,6 @@ def extract_items_from_renderer(renderer, item_types=_item_types):
renderer = nested_renderer_dispatch[key](value) renderer = nested_renderer_dispatch[key](value)
continue # don't reset renderer to None continue # don't reset renderer to None
# the renderer is an item
elif key in item_types:
items.append(renderer)
renderer = None renderer = None
def extract_items(response, item_types=_item_types): def extract_items(response, item_types=_item_types):