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.

192 lines
7.3 KiB

  1. # coding: utf-8
  2. from __future__ import unicode_literals
  3. import re
  4. from .common import InfoExtractor
  5. from ..compat import compat_HTTPError
  6. from ..utils import (
  7. determine_ext,
  8. ExtractorError,
  9. int_or_none,
  10. float_or_none,
  11. js_to_json,
  12. parse_iso8601,
  13. remove_end,
  14. strip_or_none,
  15. try_get,
  16. )
  17. class TV2IE(InfoExtractor):
  18. _VALID_URL = r'https?://(?:www\.)?tv2\.no/v/(?P<id>\d+)'
  19. _TEST = {
  20. 'url': 'http://www.tv2.no/v/916509/',
  21. 'info_dict': {
  22. 'id': '916509',
  23. 'ext': 'flv',
  24. 'title': 'Se Frode Gryttens hyllest av Steven Gerrard',
  25. 'description': 'TV 2 Sportens huspoet tar avskjed med Liverpools kaptein Steven Gerrard.',
  26. 'timestamp': 1431715610,
  27. 'upload_date': '20150515',
  28. 'duration': 156.967,
  29. 'view_count': int,
  30. 'categories': list,
  31. },
  32. }
  33. _API_DOMAIN = 'sumo.tv2.no'
  34. _PROTOCOLS = ('HDS', 'HLS', 'DASH')
  35. _GEO_COUNTRIES = ['NO']
  36. def _real_extract(self, url):
  37. video_id = self._match_id(url)
  38. api_base = 'http://%s/api/web/asset/%s' % (self._API_DOMAIN, video_id)
  39. formats = []
  40. format_urls = []
  41. for protocol in self._PROTOCOLS:
  42. try:
  43. data = self._download_json(
  44. api_base + '/play.json?protocol=%s&videoFormat=SMIL+ISMUSP' % protocol,
  45. video_id, 'Downloading play JSON')['playback']
  46. except ExtractorError as e:
  47. if isinstance(e.cause, compat_HTTPError) and e.cause.code == 401:
  48. error = self._parse_json(e.cause.read().decode(), video_id)['error']
  49. error_code = error.get('code')
  50. if error_code == 'ASSET_PLAYBACK_INVALID_GEO_LOCATION':
  51. self.raise_geo_restricted(countries=self._GEO_COUNTRIES)
  52. elif error_code == 'SESSION_NOT_AUTHENTICATED':
  53. self.raise_login_required()
  54. raise ExtractorError(error['description'])
  55. raise
  56. items = try_get(data, lambda x: x['items']['item'])
  57. if not items:
  58. continue
  59. if not isinstance(items, list):
  60. items = [items]
  61. for item in items:
  62. if not isinstance(item, dict):
  63. continue
  64. video_url = item.get('url')
  65. if not video_url or video_url in format_urls:
  66. continue
  67. format_id = '%s-%s' % (protocol.lower(), item.get('mediaFormat'))
  68. if not self._is_valid_url(video_url, video_id, format_id):
  69. continue
  70. format_urls.append(video_url)
  71. ext = determine_ext(video_url)
  72. if ext == 'f4m':
  73. formats.extend(self._extract_f4m_formats(
  74. video_url, video_id, f4m_id=format_id, fatal=False))
  75. elif ext == 'm3u8':
  76. if not data.get('drmProtected'):
  77. formats.extend(self._extract_m3u8_formats(
  78. video_url, video_id, 'mp4', entry_protocol='m3u8_native',
  79. m3u8_id=format_id, fatal=False))
  80. elif ext == 'mpd':
  81. formats.extend(self._extract_mpd_formats(
  82. video_url, video_id, format_id, fatal=False))
  83. elif ext == 'ism' or video_url.endswith('.ism/Manifest'):
  84. pass
  85. else:
  86. formats.append({
  87. 'url': video_url,
  88. 'format_id': format_id,
  89. 'tbr': int_or_none(item.get('bitrate')),
  90. 'filesize': int_or_none(item.get('fileSize')),
  91. })
  92. if not formats and data.get('drmProtected'):
  93. raise ExtractorError('This video is DRM protected.', expected=True)
  94. self._sort_formats(formats)
  95. asset = self._download_json(
  96. api_base + '.json', video_id,
  97. 'Downloading metadata JSON')['asset']
  98. title = asset['title']
  99. thumbnails = [{
  100. 'id': thumbnail.get('@type'),
  101. 'url': thumbnail.get('url'),
  102. } for _, thumbnail in (asset.get('imageVersions') or {}).items()]
  103. return {
  104. 'id': video_id,
  105. 'url': video_url,
  106. 'title': title,
  107. 'description': strip_or_none(asset.get('description')),
  108. 'thumbnails': thumbnails,
  109. 'timestamp': parse_iso8601(asset.get('createTime')),
  110. 'duration': float_or_none(asset.get('accurateDuration') or asset.get('duration')),
  111. 'view_count': int_or_none(asset.get('views')),
  112. 'categories': asset.get('keywords', '').split(','),
  113. 'formats': formats,
  114. }
  115. class TV2ArticleIE(InfoExtractor):
  116. _VALID_URL = r'https?://(?:www\.)?tv2\.no/(?:a|\d{4}/\d{2}/\d{2}(/[^/]+)+)/(?P<id>\d+)'
  117. _TESTS = [{
  118. 'url': 'http://www.tv2.no/2015/05/16/nyheter/alesund/krim/pingvin/6930542',
  119. 'info_dict': {
  120. 'id': '6930542',
  121. 'title': 'Russen hetses etter pingvintyveri - innrømmer å ha åpnet luken på buret',
  122. 'description': 'De fire siktede nekter fortsatt for å ha stjålet pingvinbabyene, men innrømmer å ha åpnet luken til de små kyllingene.',
  123. },
  124. 'playlist_count': 2,
  125. }, {
  126. 'url': 'http://www.tv2.no/a/6930542',
  127. 'only_matching': True,
  128. }]
  129. def _real_extract(self, url):
  130. playlist_id = self._match_id(url)
  131. webpage = self._download_webpage(url, playlist_id)
  132. # Old embed pattern (looks unused nowadays)
  133. assets = re.findall(r'data-assetid=["\'](\d+)', webpage)
  134. if not assets:
  135. # New embed pattern
  136. for v in re.findall(r'(?s)TV2ContentboxVideo\(({.+?})\)', webpage):
  137. video = self._parse_json(
  138. v, playlist_id, transform_source=js_to_json, fatal=False)
  139. if not video:
  140. continue
  141. asset = video.get('assetId')
  142. if asset:
  143. assets.append(asset)
  144. entries = [
  145. self.url_result('http://www.tv2.no/v/%s' % asset_id, 'TV2')
  146. for asset_id in assets]
  147. title = remove_end(self._og_search_title(webpage), ' - TV2.no')
  148. description = remove_end(self._og_search_description(webpage), ' - TV2.no')
  149. return self.playlist_result(entries, playlist_id, title, description)
  150. class KatsomoIE(TV2IE):
  151. _VALID_URL = r'https?://(?:www\.)?(?:katsomo|mtv)\.fi/(?:#!/)?(?:[^/]+/[0-9a-z-]+-\d+/[0-9a-z-]+-|[^/]+/\d+/[^/]+/)(?P<id>\d+)'
  152. _TEST = {
  153. 'url': 'https://www.mtv.fi/sarja/mtv-uutiset-live-33001002003/lahden-pelicans-teki-kovan-ratkaisun-ville-nieminen-pihalle-1181321',
  154. 'info_dict': {
  155. 'id': '1181321',
  156. 'ext': 'mp4',
  157. 'title': 'MTV Uutiset Live',
  158. 'description': 'Päätöksen teki Pelicansin hallitus.',
  159. 'timestamp': 1575116484,
  160. 'upload_date': '20191130',
  161. 'duration': 37.12,
  162. 'view_count': int,
  163. 'categories': list,
  164. },
  165. 'params': {
  166. # m3u8 download
  167. 'skip_download': True,
  168. },
  169. }
  170. _API_DOMAIN = 'api.katsomo.fi'
  171. _PROTOCOLS = ('HLS', 'MPD')
  172. _GEO_COUNTRIES = ['FI']