Add ability to choose downloader via config

This commit is contained in:
rlaphoenix
2023-05-12 06:42:33 +01:00
parent b92708ef45
commit cb82febb7c
7 changed files with 33 additions and 13 deletions

View File

@@ -1,6 +1,5 @@
from __future__ import annotations
import asyncio
import logging
import re
import sys
@@ -24,7 +23,7 @@ from requests import Session
from rich import filesize
from devine.core.constants import AnyTrack
from devine.core.downloaders import aria2c
from devine.core.downloaders import downloader
from devine.core.drm import DRM_T, ClearKey, Widevine
from devine.core.tracks import Audio, Subtitle, Tracks, Video
from devine.core.utilities import is_close_match
@@ -308,14 +307,14 @@ class HLS:
segment_save_path.parent.mkdir(parents=True, exist_ok=True)
segment_save_path.write_bytes(res.content)
else:
asyncio.run(aria2c(
downloader(
uri=segment.uri,
out=segment_save_path,
headers=session.headers,
proxy=proxy,
silent=attempts != 5,
segmented=True
))
)
break
except Exception as ee:
if stop_event.is_set() or attempts == 5: