You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

112 lines
4.4 KiB

7 years ago
7 years ago
  1. # coding: utf-8
  2. from __future__ import unicode_literals
  3. import re
  4. from .turner import TurnerBaseIE
  5. from ..utils import (
  6. float_or_none,
  7. int_or_none,
  8. strip_or_none,
  9. )
  10. class TBSIE(TurnerBaseIE):
  11. _VALID_URL = r'https?://(?:www\.)?(?P<site>tbs|tntdrama)\.com/(?:movies|shows/[^/]+/(?:clips|season-\d+/episode-\d+))/(?P<id>[^/?#]+)'
  12. _TESTS = [{
  13. 'url': 'http://www.tntdrama.com/shows/the-alienist/clips/monster',
  14. 'info_dict': {
  15. 'id': '8d384cde33b89f3a43ce5329de42903ed5099887',
  16. 'ext': 'mp4',
  17. 'title': 'Monster',
  18. 'description': 'Get a first look at the theatrical trailer for TNT’s highly anticipated new psychological thriller The Alienist, which premieres January 22 on TNT.',
  19. 'timestamp': 1508175329,
  20. 'upload_date': '20171016',
  21. },
  22. 'params': {
  23. # m3u8 download
  24. 'skip_download': True,
  25. }
  26. }, {
  27. 'url': 'http://www.tbs.com/shows/search-party/season-1/episode-1/explicit-the-mysterious-disappearance-of-the-girl-no-one-knew',
  28. 'only_matching': True,
  29. }, {
  30. 'url': 'http://www.tntdrama.com/movies/star-wars-a-new-hope',
  31. 'only_matching': True,
  32. }]
  33. def _real_extract(self, url):
  34. site, display_id = re.match(self._VALID_URL, url).groups()
  35. webpage = self._download_webpage(url, display_id)
  36. video_data = self._parse_json(self._search_regex(
  37. r'<script[^>]+?data-drupal-selector="drupal-settings-json"[^>]*?>({.+?})</script>',
  38. webpage, 'drupal setting'), display_id)['turner_playlist'][0]
  39. media_id = video_data['mediaID']
  40. title = video_data['title']
  41. streams_data = self._download_json(
  42. 'http://medium.ngtv.io/media/%s/tv' % media_id,
  43. media_id)['media']['tv']
  44. duration = None
  45. chapters = []
  46. formats = []
  47. for supported_type in ('unprotected', 'bulkaes'):
  48. stream_data = streams_data.get(supported_type, {})
  49. m3u8_url = stream_data.get('secureUrl') or stream_data.get('url')
  50. if not m3u8_url:
  51. continue
  52. if stream_data.get('playlistProtection') == 'spe':
  53. m3u8_url = self._add_akamai_spe_token(
  54. 'http://token.vgtf.net/token/token_spe',
  55. m3u8_url, media_id, {
  56. 'url': url,
  57. 'site_name': site[:3].upper(),
  58. 'auth_required': video_data.get('authRequired') == '1',
  59. })
  60. formats.extend(self._extract_m3u8_formats(
  61. m3u8_url, media_id, 'mp4', m3u8_id='hls', fatal=False))
  62. duration = float_or_none(stream_data.get('totalRuntime') or video_data.get('duration'))
  63. if not chapters:
  64. for chapter in stream_data.get('contentSegments', []):
  65. start_time = float_or_none(chapter.get('start'))
  66. duration = float_or_none(chapter.get('duration'))
  67. if start_time is None or duration is None:
  68. continue
  69. chapters.append({
  70. 'start_time': start_time,
  71. 'end_time': start_time + duration,
  72. })
  73. self._sort_formats(formats)
  74. thumbnails = []
  75. for image_id, image in video_data.get('images', {}).items():
  76. image_url = image.get('url')
  77. if not image_url or image.get('type') != 'video':
  78. continue
  79. i = {
  80. 'id': image_id,
  81. 'url': image_url,
  82. }
  83. mobj = re.search(r'(\d+)x(\d+)', image_url)
  84. if mobj:
  85. i.update({
  86. 'width': int(mobj.group(1)),
  87. 'height': int(mobj.group(2)),
  88. })
  89. thumbnails.append(i)
  90. return {
  91. 'id': media_id,
  92. 'title': title,
  93. 'description': strip_or_none(video_data.get('descriptionNoTags') or video_data.get('shortDescriptionNoTags')),
  94. 'duration': duration,
  95. 'timestamp': int_or_none(video_data.get('created')),
  96. 'season_number': int_or_none(video_data.get('season')),
  97. 'episode_number': int_or_none(video_data.get('episode')),
  98. 'cahpters': chapters,
  99. 'thumbnails': thumbnails,
  100. 'formats': formats,
  101. }