tv2hu.py 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. from .common import InfoExtractor
  2. from ..utils import (
  3. UnsupportedError,
  4. traverse_obj,
  5. )
  6. class TV2HuIE(InfoExtractor):
  7. IE_NAME = 'tv2play.hu'
  8. _VALID_URL = r'https?://(?:www\.)?tv2play\.hu/(?!szalag/)(?P<id>[^#&?]+)'
  9. _TESTS = [{
  10. 'url': 'https://tv2play.hu/mintaapak/mintaapak_213_epizod_resz',
  11. 'info_dict': {
  12. 'id': '249240',
  13. 'ext': 'mp4',
  14. 'title': 'Mintaapák - 213. epizód',
  15. 'series': 'Mintaapák',
  16. 'duration': 2164,
  17. 'description': 'md5:7350147e75485a59598e806c47967b07',
  18. 'thumbnail': r're:^https?://.*\.jpg$',
  19. 'release_date': '20210825',
  20. 'episode_number': 213,
  21. },
  22. 'params': {
  23. 'skip_download': True,
  24. },
  25. }, {
  26. 'url': 'https://tv2play.hu/taxi_2',
  27. 'md5': '585e58e2e090f34603804bb2c48e98d8',
  28. 'info_dict': {
  29. 'id': '199363',
  30. 'ext': 'mp4',
  31. 'title': 'Taxi 2',
  32. 'series': 'Taxi 2',
  33. 'duration': 5087,
  34. 'description': 'md5:47762155dc9a50241797ded101b1b08c',
  35. 'thumbnail': r're:^https?://.*\.jpg$',
  36. 'release_date': '20210118',
  37. },
  38. 'params': {
  39. 'skip_download': True,
  40. },
  41. }]
  42. def _real_extract(self, url):
  43. video_id = self._match_id(url)
  44. json_data = self._download_json(f'https://tv2play.hu/api/search/{video_id}', video_id)
  45. if json_data['contentType'] == 'showpage':
  46. ribbon_ids = traverse_obj(json_data, ('pages', ..., 'tabs', ..., 'ribbonIds'), get_all=False, expected_type=list)
  47. entries = [self.url_result(f'https://tv2play.hu/szalag/{ribbon_id}',
  48. ie=TV2HuSeriesIE.ie_key(), video_id=ribbon_id) for ribbon_id in ribbon_ids]
  49. return self.playlist_result(entries, playlist_id=video_id)
  50. elif json_data['contentType'] != 'video':
  51. raise UnsupportedError(url)
  52. video_id = str(json_data['id'])
  53. player_id = json_data.get('playerId')
  54. series_json = json_data.get('seriesInfo', {})
  55. video_json_url = self._download_json(f'https://tv2play.hu/api/streaming-url?playerId={player_id}', video_id)['url']
  56. video_json = self._download_json(video_json_url, video_id)
  57. m3u8_url = self._proto_relative_url(traverse_obj(video_json, ('bitrates', 'hls')))
  58. formats, subtitles = self._extract_m3u8_formats_and_subtitles(m3u8_url, video_id)
  59. return {
  60. 'id': video_id,
  61. 'title': json_data['title'],
  62. 'series': json_data.get('seriesTitle'),
  63. 'duration': json_data.get('length'),
  64. 'description': json_data.get('description'),
  65. 'thumbnail': 'https://tv2play.hu' + json_data.get('thumbnailUrl'),
  66. 'release_date': json_data.get('uploadedAt').replace('.', ''),
  67. 'season_number': series_json.get('seasonNr'),
  68. 'episode_number': series_json.get('episodeNr'),
  69. 'formats': formats,
  70. 'subtitles': subtitles,
  71. }
  72. class TV2HuSeriesIE(InfoExtractor):
  73. IE_NAME = 'tv2playseries.hu'
  74. _VALID_URL = r'https?://(?:www\.)?tv2play\.hu/szalag/(?P<id>[^#&?]+)'
  75. _TESTS = [{
  76. 'url': 'https://tv2play.hu/szalag/59?rendezes=nepszeruseg',
  77. 'playlist_mincount': 284,
  78. 'info_dict': {
  79. 'id': '59',
  80. },
  81. }]
  82. def _real_extract(self, url):
  83. playlist_id = self._match_id(url)
  84. json_data = self._download_json(f'https://tv2play.hu/api/ribbons/{playlist_id}/0?size=100000', playlist_id)
  85. entries = []
  86. for card in json_data.get('cards', []):
  87. video_id = card.get('slug')
  88. if video_id:
  89. entries.append(self.url_result(
  90. f'https://tv2play.hu/{video_id}', TV2HuIE, video_id))
  91. return self.playlist_result(entries, playlist_id=playlist_id)