Filter out items other than videos in search (#249)

This commit is contained in:
Vishnunarayan K I
2018-03-09 13:17:46 +05:30
committed by Ritiek Malhotra
parent f943080edb
commit 4ad77de97f

View File

@@ -82,6 +82,8 @@ def generate_youtube_url(raw_song, meta_tags, tries_remaining=5):
log.debug('query: {0}'.format(query))
data = pafy.call_gdata('search', query)
data['items'] = list(filter(lambda x: x['id'].get('videoId') is not None,
data['items']))
query_results = {'part': 'contentDetails,snippet,statistics',
'maxResults': 50,
'id': ','.join(i['id']['videoId'] for i in data['items'])}