AlexYU/AlexLive
1
0
Fork 0
Code Pull Requests Activity

fix:get speed of m3u8 playlist timeout()

Browse Source
This commit is contained in:
guorong.zheng 2024-12-13 11:15:26 +08:00
parent 2fb0f55e0c
commit 5dd60de1e7
1 changed files with 18 additions and 13 deletions

31
utils/speed.py

@ -45,19 +45,24 @@ async def get_speed_m3u8(url: str, timeout: int = config.sort_timeout) -> dict[s
"""
info = {'speed': None, 'delay': None}
try:
url = quote(url, safe=':/?$&=@')
m3u8_obj = m3u8.load(url, timeout=2)
speed_list = []
start_time = time()
for segment in m3u8_obj.segments:
if time() - start_time > timeout:
break
ts_url = segment.absolute_uri
download_info = await get_speed_with_download(ts_url, timeout)
speed_list.append(download_info['speed'])
if info['delay'] is None and download_info['delay'] is not None:
info['delay'] = download_info['delay']
info['speed'] = sum(speed_list) / len(speed_list) if speed_list else 0
url = quote(url, safe=':/?$&=@').partition('$')[0]
async with ClientSession(connector=TCPConnector(ssl=False), trust_env=True) as session:
async with session.head(url, timeout=2) as response:
if response.headers.get('Content-Length'):
m3u8_obj = m3u8.load(url, timeout=2)
speed_list = []
start_time = time()
for segment in m3u8_obj.segments:
if time() - start_time > timeout:
break
ts_url = segment.absolute_uri
download_info = await get_speed_with_download(ts_url, timeout)
speed_list.append(download_info['speed'])
if info['delay'] is None and download_info['delay'] is not None:
info['delay'] = download_info['delay']
info['speed'] = sum(speed_list) / len(speed_list) if speed_list else 0
else:
return info
except:
pass
finally: