| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | # Youtube (Videos) | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # @website     https://www.youtube.com/ | 
					
						
							|  |  |  | # @provide-api yes (https://developers.google.com/apis-explorer/#p/youtube/v3/youtube.search.list) | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # @using-api   no | 
					
						
							|  |  |  | # @results     HTML | 
					
						
							|  |  |  | # @stable      no | 
					
						
							|  |  |  | # @parse       url, title, content, publishedDate, thumbnail, embedded | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | from urllib import quote_plus | 
					
						
							|  |  |  | from lxml import html | 
					
						
							|  |  |  | from searx.engines.xpath import extract_text | 
					
						
							| 
									
										
										
										
											2015-06-04 18:34:44 +02:00
										 |  |  | from searx.utils import list_get | 
					
						
							| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | # engine dependent config | 
					
						
							|  |  |  | categories = ['videos', 'music'] | 
					
						
							|  |  |  | paging = True | 
					
						
							|  |  |  | language_support = False | 
					
						
							| 
									
										
										
										
											2016-10-30 20:26:38 +01:00
										 |  |  | time_range_support = True | 
					
						
							| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | # search-url | 
					
						
							|  |  |  | base_url = 'https://www.youtube.com/results' | 
					
						
							|  |  |  | search_url = base_url + '?search_query={query}&page={page}' | 
					
						
							| 
									
										
										
										
											2016-10-30 20:26:38 +01:00
										 |  |  | time_range_url = '&sp=EgII{time_range}%253D%253D' | 
					
						
							|  |  |  | time_range_dict = {'day': 'Ag', | 
					
						
							|  |  |  |                    'week': 'Aw', | 
					
						
							| 
									
										
										
										
											2016-12-11 16:41:14 +01:00
										 |  |  |                    'month': 'BA', | 
					
						
							|  |  |  |                    'year': 'BQ'} | 
					
						
							| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | embedded_url = '<iframe width="540" height="304" ' +\ | 
					
						
							|  |  |  |     'data-src="//www.youtube-nocookie.com/embed/{videoid}" ' +\ | 
					
						
							|  |  |  |     'frameborder="0" allowfullscreen></iframe>' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | base_youtube_url = 'https://www.youtube.com/watch?v=' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # specific xpath variables | 
					
						
							|  |  |  | results_xpath = "//ol/li/div[contains(@class, 'yt-lockup yt-lockup-tile yt-lockup-video vve-check')]" | 
					
						
							|  |  |  | url_xpath = './/h3/a/@href' | 
					
						
							|  |  |  | title_xpath = './/div[@class="yt-lockup-content"]/h3/a' | 
					
						
							|  |  |  | content_xpath = './/div[@class="yt-lockup-content"]/div[@class="yt-lockup-description yt-ui-ellipsis yt-ui-ellipsis-2"]' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-06-03 10:12:30 +02:00
										 |  |  | # returns extract_text on the first result selected by the xpath or None | 
					
						
							|  |  |  | def extract_text_from_dom(result, xpath): | 
					
						
							|  |  |  |     r = result.xpath(xpath) | 
					
						
							|  |  |  |     if len(r) > 0: | 
					
						
							|  |  |  |         return extract_text(r[0]) | 
					
						
							|  |  |  |     return None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | # do search-request | 
					
						
							|  |  |  | def request(query, params): | 
					
						
							|  |  |  |     params['url'] = search_url.format(query=quote_plus(query), | 
					
						
							|  |  |  |                                       page=params['pageno']) | 
					
						
							| 
									
										
										
										
											2016-10-30 20:26:38 +01:00
										 |  |  |     if params['time_range'] in time_range_dict: | 
					
						
							|  |  |  |         params['url'] += time_range_url.format(time_range=time_range_dict[params['time_range']]) | 
					
						
							| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     return params | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # get response from search-request | 
					
						
							|  |  |  | def response(resp): | 
					
						
							|  |  |  |     results = [] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     dom = html.fromstring(resp.text) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     # parse results | 
					
						
							|  |  |  |     for result in dom.xpath(results_xpath): | 
					
						
							| 
									
										
										
										
											2015-06-03 10:12:30 +02:00
										 |  |  |         videoid = list_get(result.xpath('@data-context-item-id'), 0) | 
					
						
							|  |  |  |         if videoid is not None: | 
					
						
							|  |  |  |             url = base_youtube_url + videoid | 
					
						
							|  |  |  |             thumbnail = 'https://i.ytimg.com/vi/' + videoid + '/hqdefault.jpg' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             title = extract_text_from_dom(result, title_xpath) or videoid | 
					
						
							|  |  |  |             content = extract_text_from_dom(result, content_xpath) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             embedded = embedded_url.format(videoid=videoid) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             # append result | 
					
						
							|  |  |  |             results.append({'url': url, | 
					
						
							|  |  |  |                             'title': title, | 
					
						
							|  |  |  |                             'content': content, | 
					
						
							|  |  |  |                             'template': 'videos.html', | 
					
						
							|  |  |  |                             'embedded': embedded, | 
					
						
							|  |  |  |                             'thumbnail': thumbnail}) | 
					
						
							| 
									
										
										
										
											2015-05-31 00:25:59 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     # return results | 
					
						
							|  |  |  |     return results |