youtube_noapi.py 3.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. # Youtube (Videos)
  2. #
  3. # @website https://www.youtube.com/
  4. # @provide-api yes (https://developers.google.com/apis-explorer/#p/youtube/v3/youtube.search.list)
  5. #
  6. # @using-api no
  7. # @results HTML
  8. # @stable no
  9. # @parse url, title, content, publishedDate, thumbnail, embedded
  10. from functools import reduce
  11. from json import loads
  12. from urllib.parse import quote_plus
  13. # engine dependent config
  14. categories = ['videos', 'music']
  15. paging = True
  16. language_support = False
  17. time_range_support = True
  18. # search-url
  19. base_url = 'https://www.youtube.com/results'
  20. search_url = base_url + '?search_query={query}&page={page}'
  21. time_range_url = '&sp=EgII{time_range}%253D%253D'
  22. time_range_dict = {'day': 'Ag',
  23. 'week': 'Aw',
  24. 'month': 'BA',
  25. 'year': 'BQ'}
  26. embedded_url = '<iframe width="540" height="304" ' +\
  27. 'data-src="https://www.youtube-nocookie.com/embed/{videoid}" ' +\
  28. 'frameborder="0" allowfullscreen></iframe>'
  29. base_youtube_url = 'https://www.youtube.com/watch?v='
  30. # do search-request
  31. def request(query, params):
  32. params['url'] = search_url.format(query=quote_plus(query),
  33. page=params['pageno'])
  34. if params['time_range'] in time_range_dict:
  35. params['url'] += time_range_url.format(time_range=time_range_dict[params['time_range']])
  36. return params
  37. # get response from search-request
  38. def response(resp):
  39. results = []
  40. results_data = resp.text[resp.text.find('ytInitialData'):]
  41. results_data = results_data[results_data.find('{'):results_data.find(';</script>')]
  42. results_json = loads(results_data) if results_data else {}
  43. sections = results_json.get('contents', {})\
  44. .get('twoColumnSearchResultsRenderer', {})\
  45. .get('primaryContents', {})\
  46. .get('sectionListRenderer', {})\
  47. .get('contents', [])
  48. for section in sections:
  49. for video_container in section.get('itemSectionRenderer', {}).get('contents', []):
  50. video = video_container.get('videoRenderer', {})
  51. videoid = video.get('videoId')
  52. if videoid is not None:
  53. url = base_youtube_url + videoid
  54. thumbnail = 'https://i.ytimg.com/vi/' + videoid + '/hqdefault.jpg'
  55. title = get_text_from_json(video.get('title', {}))
  56. content = get_text_from_json(video.get('descriptionSnippet', {}))
  57. embedded = embedded_url.format(videoid=videoid)
  58. author = get_text_from_json(video.get('ownerText', {}))
  59. length = get_text_from_json(video.get('lengthText', {}))
  60. # append result
  61. results.append({'url': url,
  62. 'title': title,
  63. 'content': content,
  64. 'author': author,
  65. 'length': length,
  66. 'template': 'videos.html',
  67. 'embedded': embedded,
  68. 'thumbnail': thumbnail})
  69. # return results
  70. return results
  71. def get_text_from_json(element):
  72. if 'runs' in element:
  73. return reduce(lambda a, b: a + b.get('text', ''), element.get('runs'), '')
  74. else:
  75. return element.get('simpleText', '')