From 4ad77de97f0570d21a8a8107f86c9356332ff945 Mon Sep 17 00:00:00 2001 From: Vishnunarayan K I <31964688+vn-ki@users.noreply.github.com> Date: Fri, 9 Mar 2018 13:17:46 +0530 Subject: [PATCH] Filter out items other than videos in search (#249) --- core/youtube_tools.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/core/youtube_tools.py b/core/youtube_tools.py index 0e736e9..91c9e4f 100644 --- a/core/youtube_tools.py +++ b/core/youtube_tools.py @@ -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'])}