tursom/blrec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: sort url info by host

Browse Source
This commit is contained in:
acgnhik 2022-07-17 11:31:38 +08:00
parent fdd93e70a5
commit 7f366463d4
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/blrec/bili/live.py
View File

@ -3,7 +3,7 @@ import json
import logging import logging
import re import re
import time import time
from typing import Dict, List, cast from typing import Any, Dict, List, cast
import aiohttp import aiohttp
from jsonpath import jsonpath from jsonpath import jsonpath
@ -234,7 +234,26 @@ class Live:
if qn not in accept_qn or codec['current_qn'] != qn: if qn not in accept_qn or codec['current_qn'] != qn:
raise NoStreamQualityAvailable(stream_format, stream_codec, qn) raise NoStreamQualityAvailable(stream_format, stream_codec, qn)
urls = [i['host'] + codec['base_url'] + i['extra'] for i in codec['url_info']] def sort_by_host(info: Any) -> int:
host = info['host']
if match := re.search(r'gotcha(\d+)', host):
num = match.group(1)
if num == '04':
return 0
if num == '09':
return 1
if num == '08':
return 2
return int(num)
elif re.search(r'cn-[a-z]+-[a-z]+', host):
return 1000
elif 'mcdn' in host:
return 2000
else:
return 10000
url_info = sorted(codec['url_info'], key=sort_by_host)
urls = [i['host'] + codec['base_url'] + i['extra'] for i in url_info]
if not select_alternative: if not select_alternative:
return urls[0] return urls[0]