depau/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2024-11-16 21:07:36 +00:00
Code Issues Releases Wiki Activity

[InstagramUser] Make use of try_get()

Browse source
This commit is contained in:
Andrew Udvare 2018-09-27 04:50:28 -04:00
parent acf3c646db
commit b4b9ab4a08
No known key found for this signature in database
GPG key ID: 1AFD9AFC120C26DD
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
youtube_dl/extractor/instagram.py
View file

@ -365,11 +365,12 @@ class InstagramUserIE(InfoExtractor):
resp = self._download_json('https://www.instagram.com/graphql/query/',
username,
query=query)
media = resp['data']['user']['edge_saved_media']
media = try_get(resp,
lambda x: x['data']['user']['edge_saved_media'])
for edge in media.get('edges', []):
node = edge.get('node')
if not node or not isinstance(node, dict):
node = try_get(edge, lambda x: x['node'], dict)
if not node:
continue
if node.get('__typename') != 'GraphVideo' and node.get('is_video') is not True:
continue
@ -382,16 +383,16 @@ class InstagramUserIE(InfoExtractor):
yield info
page_info = media.get('page_info')
if not page_info or not isinstance(page_info, dict):
page_info = try_get(media, lambda x: x['page_info'], dict)
if not page_info:
break
has_next_page = page_info.get('has_next_page')
if not has_next_page:
break
cursor = page_info.get('end_cursor')
if not cursor or not isinstance(cursor, compat_str):
cursor = try_get(page_info, lambda x: x['end_cursor'], compat_str)
if not cursor:
break
def _real_extract(self, url):