R-yan/l1ving_youtube-dl
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[PeerTube] Refactor code to avoid repetition

Browse Source
This commit is contained in:
3risian 2020-01-11 16:01:12 +11:00 committed by GitHub
parent 7057804c94
commit 74026525e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
youtube_dl/extractor/peertube.py
View File

@ -543,13 +543,23 @@ class PeerTubeIE(InfoExtractor):
subtitles = self.extract_subtitles(host, video_id) subtitles = self.extract_subtitles(host, video_id)
def try_get_second_level_data(section, field):
return try_get(video, lambda x: x[section][field], compat_str)
def account_data(field): def account_data(field):
return try_get(video, lambda x: x['account'][field], compat_str) return try_get_second_level_data('account', field)
def channel_data(field): def channel_data(field):
return try_get(video, lambda x: x['channel'][field], compat_str) return try_get_second_level_data('channel', field)
category = try_get(video, lambda x: x['category']['label'], compat_str) def make_id_string(name_field, host_field):
name = str_or_none(name_field)
host = str_or_none(host_field)
if name and host:
return '%s@%s' % (name, host)
return None
category = try_get_second_level_data('category', 'label')
categories = [category] if category else None categories = [category] if category else None
nsfw = video.get('nsfw') nsfw = video.get('nsfw')
@ -565,15 +575,13 @@ class PeerTubeIE(InfoExtractor):
'thumbnail': urljoin(url, video.get('thumbnailPath')), 'thumbnail': urljoin(url, video.get('thumbnailPath')),
'timestamp': unified_timestamp(video.get('publishedAt')), 'timestamp': unified_timestamp(video.get('publishedAt')),
'uploader': account_data('displayName'), 'uploader': account_data('displayName'),
'uploader_id': '%s@%s' % (account_data('name'), account_data('host')), 'uploader_id': make_id_string(account_data('name'), account_data('host')),
'uploader_url': account_data('url'), 'uploader_url': account_data('url'),
'channel': channel_data('displayName'), 'channel': channel_data('displayName'),
'channel_id': '%s@%s' % (channel_data('name'), channel_data('host')), 'channel_id': make_id_string(channel_data('name'), channel_data('host')),
'channel_url': channel_data('url'), 'channel_url': channel_data('url'),
'language': try_get( 'language': try_get_second_level_data('language', 'id'),
video, lambda x: x['language']['id'], compat_str), 'license': try_get_second_level_data('licence', 'label'),
'license': try_get(
video, lambda x: x['licence']['label'], compat_str),
'duration': int_or_none(video.get('duration')), 'duration': int_or_none(video.get('duration')),
'view_count': int_or_none(video.get('views')), 'view_count': int_or_none(video.get('views')),
'like_count': int_or_none(video.get('likes')), 'like_count': int_or_none(video.get('likes')),