github/DouyinLiveRecorder
1
0
Fork 0
mirror of https://github.com/ihmily/DouyinLiveRecorder.git synced 2025-12-26 05:48:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add support for 60fps recording of douyu live streams

Browse Source
This commit is contained in:
ihmily 2024-06-17 18:06:54 +08:00
parent c81a1f0196
commit d296540ae9
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@ -488,7 +488,7 @@ def get_huya_stream_url(json_data: dict, video_quality: str) -> Dict[str, Any]:
result['flv_url'] = flv_url result['flv_url'] = flv_url
result['m3u8_url'] = m3u8_url result['m3u8_url'] = m3u8_url
result['is_live'] = True result['is_live'] = True
result['record_url'] = flv_url # m3u8经常会出现断流 result['record_url'] = flv_url
return result return result
@ -509,10 +509,12 @@ def get_douyu_stream_url(json_data: dict, cookies: str, video_quality: str, prox
rid = str(json_data["room_id"]) rid = str(json_data["room_id"])
json_data.pop("room_id", None) json_data.pop("room_id", None)
rate = video_quality_options.get(video_quality, '0') # 默认为原画，只有登录后才能获取最高画质，需要配置cookie rate = video_quality_options.get(video_quality, '0')
flv_data = get_douyu_stream_data(rid, rate, cookies=cookies, proxy_addr=proxy_address) flv_data = get_douyu_stream_data(rid, rate, cookies=cookies, proxy_addr=proxy_address)
flv_url = flv_data['data'].get('url', None) rtmp_url = flv_data['data'].get('rtmp_url', None)
if flv_url: rtmp_live = flv_data['data'].get('rtmp_live', None)
if rtmp_live:
flv_url = f'{rtmp_url}/{rtmp_live}'
json_data['flv_url'] = flv_url json_data['flv_url'] = flv_url
json_data['record_url'] = flv_url json_data['record_url'] = flv_url
return json_data return json_data
@ -528,9 +530,8 @@ def get_yy_stream_url(json_data: dict) -> Dict[str, Any]:
} }
if 'avp_info_res' in json_data: if 'avp_info_res' in json_data:
stream_line_addr = json_data['avp_info_res']['stream_line_addr'] stream_line_addr = json_data['avp_info_res']['stream_line_addr']
# 获取最后一个键的值
cdn_info = list(stream_line_addr.values())[0] cdn_info = list(stream_line_addr.values())[0]
flv_url = cdn_info['cdn_info']['url'] # 清晰度暂时默认高清 flv_url = cdn_info['cdn_info']['url']
result['flv_url'] = flv_url result['flv_url'] = flv_url
result['is_live'] = True result['is_live'] = True
result['record_url'] = flv_url result['record_url'] = flv_url

3
spider.py
View File

@ -440,7 +440,8 @@ def get_douyu_stream_data(rid: str, rate: str = '-1', proxy_addr: Union[str, Non
'rate': rate, # 0蓝光、3超清、2高清、-1默认 'rate': rate, # 0蓝光、3超清、2高清、-1默认
} }
app_api = 'https://m.douyu.com/hgapi/livenc/room/getStreamUrl' # app_api = 'https://m.douyu.com/hgapi/livenc/room/getStreamUrl'
app_api = f'https://www.douyu.com/lapi/live/getH5Play/{rid}'
json_str = get_req(url=app_api, proxy_addr=proxy_addr, headers=headers, data=data) json_str = get_req(url=app_api, proxy_addr=proxy_addr, headers=headers, data=data)
json_data = json.loads(json_str) json_data = json.loads(json_str)
return json_data return json_data