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

refactor: update quality map

Browse Source
This commit is contained in:
ihmily 2025-02-05 20:52:36 +08:00
parent e189640d3a
commit ef97e01dba
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
streamget/stream.py
View File

@ -159,7 +159,11 @@ def get_kuaishou_stream_url(json_data: dict, video_quality: str) -> dict:
flv_url_list = json_data['flv_url_list']
flv_url_list = sorted(flv_url_list, key=lambda x: x['bitrate'], reverse=True)
# uses quality_mapping_bitrate to get the index of the quality
quality_index_bitrate_value = quality_mapping_bitrate[video_quality]
quality_str = str(video_quality).upper()
if quality_str.isdigit():
quality_index_bitrate_value = list(quality_mapping_bitrate.values())[int(quality_str)]
else:
quality_index_bitrate_value = quality_mapping_bitrate.get(quality_str, 99999)
# find the value below `quality_index_bitrate_value`, or else use the previous one.
quality_index = next(
(i for i, x in enumerate(flv_url_list) if x['bitrate'] <= quality_index_bitrate_value), None)