projects / yt-dlp.git / blame
| Commit | Line | Data |
|---|---|---|
| 1da19488 YCH |
1 | # coding: utf-8 |
| 2 | from __future__ import unicode_literals | |
| 3 | ||
| dc27fd8b RA |
4 | import base64 |
| 5 | import time | |
| 6 | import uuid | |
| 7 | ||
| 1da19488 | 8 | from .common import InfoExtractor |
| dc27fd8b RA |
9 | from ..compat import ( |
| 10 | compat_HTTPError, | |
| 11 | compat_str, | |
| 12 | ) | |
| 13 | from ..utils import ( | |
| 14 | ExtractorError, | |
| 15 | int_or_none, | |
| 16 | ) | |
| 1da19488 YCH |
17 | |
| 18 | ||
| 19 | class MGTVIE(InfoExtractor): | |
| 30eaa3a7 | 20 | _VALID_URL = r'https?://(?:www\.)?mgtv\.com/(v|b)/(?:[^/]+/)*(?P<id>\d+)\.html' |
| a292cba2 | 21 | IE_DESC = '芒果TV' |
| dc27fd8b | 22 | _GEO_COUNTRIES = ['CN'] |
| 1da19488 | 23 | |
| dc35ba0e | 24 | _TESTS = [{ |
| 1da19488 | 25 | 'url': 'http://www.mgtv.com/v/1/290525/f/3116640.html', |
| 1da19488 YCH |
26 | 'info_dict': { |
| 27 | 'id': '3116640', | |
| 28 | 'ext': 'mp4', | |
| dc27fd8b | 29 | 'title': '我是歌手 第四季', |
| 1da19488 YCH |
30 | 'description': '我是歌手第四季双年巅峰会', |
| 31 | 'duration': 7461, | |
| ec85ded8 | 32 | 'thumbnail': r're:^https?://.*\.jpg$', |
| 1da19488 | 33 | }, |
| dc35ba0e | 34 | }, { |
| 30eaa3a7 | 35 | 'url': 'http://www.mgtv.com/b/301817/3826653.html', |
| dc35ba0e S |
36 | 'only_matching': True, |
| 37 | }] | |
| 1da19488 YCH |
38 | |
| 39 | def _real_extract(self, url): | |
| 40 | video_id = self._match_id(url) | |
| dc27fd8b RA |
41 | try: |
| 42 | api_data = self._download_json( | |
| 43 | 'https://pcweb.api.mgtv.com/player/video', video_id, query={ | |
| 44 | 'tk2': base64.urlsafe_b64encode(b'did=%s|pno=1030|ver=0.3.0301|clit=%d' % (compat_str(uuid.uuid4()).encode(), time.time()))[::-1], | |
| 45 | 'video_id': video_id, | |
| 46 | }, headers=self.geo_verification_headers())['data'] | |
| 47 | except ExtractorError as e: | |
| 48 | if isinstance(e.cause, compat_HTTPError) and e.cause.code == 401: | |
| 49 | error = self._parse_json(e.cause.read().decode(), None) | |
| 50 | if error.get('code') == 40005: | |
| 51 | self.raise_geo_restricted(countries=self._GEO_COUNTRIES) | |
| 52 | raise ExtractorError(error['msg'], expected=True) | |
| 53 | raise | |
| 1da19488 | 54 | info = api_data['info'] |
| 30eaa3a7 | 55 | title = info['title'].strip() |
| dc27fd8b RA |
56 | stream_data = self._download_json( |
| 57 | 'https://pcweb.api.mgtv.com/player/getSource', video_id, query={ | |
| 58 | 'pm2': api_data['atc']['pm2'], | |
| 59 | 'video_id': video_id, | |
| 60 | }, headers=self.geo_verification_headers())['data'] | |
| 61 | stream_domain = stream_data['stream_domain'][0] | |
| 1da19488 YCH |
62 | |
| 63 | formats = [] | |
| dc27fd8b | 64 | for idx, stream in enumerate(stream_data['stream']): |
| 30eaa3a7 RA |
65 | stream_path = stream.get('url') |
| 66 | if not stream_path: | |
| 67 | continue | |
| 68 | format_data = self._download_json( | |
| 69 | stream_domain + stream_path, video_id, | |
| 70 | note='Download video info for format #%d' % idx) | |
| 71 | format_url = format_data.get('info') | |
| 72 | if not format_url: | |
| b9e7bc55 | 73 | continue |
| dc27fd8b | 74 | tbr = int_or_none(stream.get('filebitrate') or self._search_regex( |
| 30eaa3a7 RA |
75 | r'_(\d+)_mp4/', format_url, 'tbr', default=None)) |
| 76 | formats.append({ | |
| 77 | 'format_id': compat_str(tbr or idx), | |
| 78 | 'url': format_url, | |
| 79 | 'ext': 'mp4', | |
| 80 | 'tbr': tbr, | |
| 81 | 'protocol': 'm3u8_native', | |
| 4dcd4b7b S |
82 | 'http_headers': { |
| 83 | 'Referer': url, | |
| 84 | }, | |
| 9a37ff82 | 85 | 'format_note': stream.get('name'), |
| 30eaa3a7 | 86 | }) |
| 1da19488 YCH |
87 | self._sort_formats(formats) |
| 88 | ||
| 89 | return { | |
| 90 | 'id': video_id, | |
| 30eaa3a7 | 91 | 'title': title, |
| 1da19488 YCH |
92 | 'formats': formats, |
| 93 | 'description': info.get('desc'), | |
| 94 | 'duration': int_or_none(info.get('duration')), | |
| 95 | 'thumbnail': info.get('thumb'), | |
| 96 | } |