]>
jfr.im git - yt-dlp.git/blob - yt_dlp/extractor/huya.py
2 from __future__
import unicode_literals
7 from ..compat
import compat_urlparse
, compat_b64decode
19 from .common
import InfoExtractor
22 class HuyaLiveIE(InfoExtractor
):
23 _VALID_URL
= r
'https?://(?:www\.|m\.)?huya\.com/(?P<id>[^/#?&]+)(?:\D|$)'
27 'url': 'https://www.huya.com/572329',
36 'skip_download': True,
39 'url': 'https://www.huya.com/xiaoyugame',
58 def _real_extract(self
, url
):
59 video_id
= self
._match
_id
(url
)
60 webpage
= self
._download
_webpage
(url
, video_id
=video_id
)
61 json_stream
= self
._search
_regex
(r
'"stream":\s+"([a-zA-Z0-9+=/]+)"', webpage
, 'stream', default
=None)
63 raise ExtractorError('Video is offline', expected
=True)
64 stream_data
= self
._parse
_json
(compat_b64decode(json_stream
).decode(), video_id
=video_id
,
65 transform_source
=js_to_json
)
66 room_info
= try_get(stream_data
, lambda x
: x
['data'][0]['gameLiveInfo'])
68 raise ExtractorError('Can not extract the room info', expected
=True)
69 title
= room_info
.get('roomName') or room_info
.get('introduction') or self
._html
_search
_regex
(
70 r
'<title>([^<]+)</title>', webpage
, 'title')
71 screen_type
= room_info
.get('screenType')
72 live_source_type
= room_info
.get('liveSourceType')
73 stream_info_list
= stream_data
['data'][0]['gameStreamInfoList']
75 for stream_info
in stream_info_list
:
76 stream_url
= stream_info
.get('sFlvUrl')
79 stream_name
= stream_info
.get('sStreamName')
80 re_secret
= not screen_type
and live_source_type
in (0, 8, 13)
81 params
= dict(compat_urlparse
.parse_qsl(unescapeHTML(stream_info
['sFlvAntiCode'])))
84 fm
, ss
= self
.encrypt(params
, stream_info
, stream_name
)
85 for si
in stream_data
.get('vMultiStreamInfo'):
86 rate
= si
.get('iBitRate')
88 params
['ratio'] = rate
90 params
.pop('ratio', None)
92 params
['wsSecret'] = hashlib
.md5(
93 '_'.join([fm
, params
['u'], stream_name
, ss
, params
['wsTime']]))
95 'ext': stream_info
.get('sFlvUrlSuffix'),
96 'format_id': str_or_none(stream_info
.get('iLineIndex')),
98 'url': update_url_query(f
'{stream_url}/{stream_name}.{stream_info.get("sFlvUrlSuffix")}',
100 **self
._RESOLUTION
.get(si
.get('sDisplayName'), {}),
103 self
._sort
_formats
(formats
)
109 'view_count': room_info
.get('totalCount'),
110 'thumbnail': room_info
.get('screenshot'),
111 'description': room_info
.get('contentIntro'),
113 'Origin': 'https://www.huya.com',
114 'Referer': 'https://www.huya.com/',
118 def encrypt(self
, params
, stream_info
, stream_name
):
119 ct
= int_or_none(params
.get('wsTime'), 16) + random
.random()
120 presenter_uid
= stream_info
['lPresenterUid']
121 if not stream_name
.startswith(str(presenter_uid
)):
124 uid
= int_or_none(ct
% 1e7
* 1e6
% 0xffffffff)
125 u1
= uid
& 0xffffffff00000000
126 u2
= uid
& 0xffffffff
128 u
= u1 | u2
>> 24 | u3
<< 8
131 'seqid': str_or_none(int_or_none(ct
* 1000) + uid
),
133 'uuid': int_or_none(ct
% 1e7
* 1e6
% 0xffffffff),
136 fm
= compat_b64decode(params
['fm']).decode().split('_', 1)[0]
137 ss
= hashlib
.md5('|'.join([params
['seqid'], params
['ctype'], params
['t']]))