[SnapchatStory] Make use of try_get()

This commit is contained in:
Andrew Udvare 2018-09-27 04:55:36 -04:00
parent b81c1d9eef
commit 3c945d26a6
No known key found for this signature in database
GPG Key ID: 1AFD9AFC120C26DD

View File

@ -2,8 +2,8 @@
from __future__ import unicode_literals from __future__ import unicode_literals
from .common import InfoExtractor from .common import InfoExtractor
from ..compat import compat_urllib_parse_urlencode from ..compat import compat_str, compat_urllib_parse_urlencode
from ..utils import ExtractorError, int_or_none from ..utils import ExtractorError, int_or_none, try_get
class SnapchatStoryIE(InfoExtractor): class SnapchatStoryIE(InfoExtractor):
@ -66,8 +66,12 @@ class SnapchatStoryIE(InfoExtractor):
query=dict(request_origin='ORIGIN_WEB_PLAYER')) query=dict(request_origin='ORIGIN_WEB_PLAYER'))
story = data['story'] story = data['story']
title = story['metadata']['title'] title = try_get(story, lambda x: x['metadata']['title'], compat_str)
alt_title = story['metadata'].get('subTitles') if not title:
title = 'Untitled'
alt_title = try_get(story,
lambda x: x['metadata']['subTitles'],
compat_str)
return self.playlist_result(self._entries(story, title, alt_title), return self.playlist_result(self._entries(story, title, alt_title),
playlist_title=title, playlist_title=title,