airtv.py 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. from .common import InfoExtractor
  2. from .youtube import YoutubeIE
  3. from ..utils import (
  4. determine_ext,
  5. int_or_none,
  6. mimetype2ext,
  7. parse_iso8601,
  8. traverse_obj,
  9. )
  10. class AirTVIE(InfoExtractor):
  11. _VALID_URL = r'https?://www\.air\.tv/watch\?v=(?P<id>\w+)'
  12. _TESTS = [{
  13. # without youtube_id
  14. 'url': 'https://www.air.tv/watch?v=W87jcWleSn2hXZN47zJZsQ',
  15. 'info_dict': {
  16. 'id': 'W87jcWleSn2hXZN47zJZsQ',
  17. 'ext': 'mp4',
  18. 'release_date': '20221003',
  19. 'release_timestamp': 1664792603,
  20. 'channel_id': 'vgfManQlRQKgoFQ8i8peFQ',
  21. 'title': 'md5:c12d49ed367c3dadaa67659aff43494c',
  22. 'upload_date': '20221003',
  23. 'duration': 151,
  24. 'view_count': int,
  25. 'thumbnail': 'https://cdn-sp-gcs.air.tv/videos/W/8/W87jcWleSn2hXZN47zJZsQ/b13fc56464f47d9d62a36d110b9b5a72-4096x2160_9.jpg',
  26. 'timestamp': 1664792603,
  27. },
  28. }, {
  29. # with youtube_id
  30. 'url': 'https://www.air.tv/watch?v=sv57EC8tRXG6h8dNXFUU1Q',
  31. 'info_dict': {
  32. 'id': '2ZTqmpee-bQ',
  33. 'ext': 'mp4',
  34. 'comment_count': int,
  35. 'tags': 'count:11',
  36. 'channel_follower_count': int,
  37. 'like_count': int,
  38. 'uploader': 'Newsflare',
  39. 'thumbnail': 'https://i.ytimg.com/vi_webp/2ZTqmpee-bQ/maxresdefault.webp',
  40. 'availability': 'public',
  41. 'title': 'Geese Chase Alligator Across Golf Course',
  42. 'uploader_id': 'NewsflareBreaking',
  43. 'channel_url': 'https://www.youtube.com/channel/UCzSSoloGEz10HALUAbYhngQ',
  44. 'description': 'md5:99b21d9cea59330149efbd9706e208f5',
  45. 'age_limit': 0,
  46. 'channel_id': 'UCzSSoloGEz10HALUAbYhngQ',
  47. 'uploader_url': 'http://www.youtube.com/user/NewsflareBreaking',
  48. 'view_count': int,
  49. 'categories': ['News & Politics'],
  50. 'live_status': 'not_live',
  51. 'playable_in_embed': True,
  52. 'channel': 'Newsflare',
  53. 'duration': 37,
  54. 'upload_date': '20180511',
  55. },
  56. }]
  57. def _get_formats_and_subtitle(self, json_data, video_id):
  58. formats, subtitles = [], {}
  59. for source in traverse_obj(json_data, 'sources', 'sources_desktop', ...):
  60. ext = determine_ext(source.get('src'), mimetype2ext(source.get('type')))
  61. if ext == 'm3u8':
  62. fmts, subs = self._extract_m3u8_formats_and_subtitles(source.get('src'), video_id)
  63. formats.extend(fmts)
  64. self._merge_subtitles(subs, target=subtitles)
  65. else:
  66. formats.append({'url': source.get('src'), 'ext': ext})
  67. return formats, subtitles
  68. def _real_extract(self, url):
  69. display_id = self._match_id(url)
  70. webpage = self._download_webpage(url, display_id)
  71. nextjs_json = self._search_nextjs_data(webpage, display_id)['props']['pageProps']['initialState']['videos'][display_id]
  72. if nextjs_json.get('youtube_id'):
  73. return self.url_result(
  74. f'https://www.youtube.com/watch?v={nextjs_json.get("youtube_id")}', YoutubeIE)
  75. formats, subtitles = self._get_formats_and_subtitle(nextjs_json, display_id)
  76. return {
  77. 'id': display_id,
  78. 'title': nextjs_json.get('title') or self._html_search_meta('og:title', webpage),
  79. 'formats': formats,
  80. 'subtitles': subtitles,
  81. 'description': nextjs_json.get('description') or None,
  82. 'duration': int_or_none(nextjs_json.get('duration')),
  83. 'thumbnails': [
  84. {'url': thumbnail}
  85. for thumbnail in traverse_obj(nextjs_json, ('default_thumbnails', ...))],
  86. 'channel_id': traverse_obj(nextjs_json, 'channel', 'channel_slug'),
  87. 'timestamp': parse_iso8601(nextjs_json.get('created')),
  88. 'release_timestamp': parse_iso8601(nextjs_json.get('published')),
  89. 'view_count': int_or_none(nextjs_json.get('views')),
  90. }