mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-24 12:15:40 +01:00
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
import copy
|
|
|
|
from svtplay_dl.utils.http import HTTP
|
|
from svtplay_dl.utils.output import ETA
|
|
from svtplay_dl.utils.output import output
|
|
from svtplay_dl.utils.output import progressbar
|
|
|
|
|
|
class VideoRetriever:
|
|
def __init__(self, config, url, bitrate, output, **kwargs):
|
|
self.config = config
|
|
self.url = url
|
|
self.bitrate = int(bitrate) if bitrate else 0
|
|
self.kwargs = kwargs
|
|
self.http = HTTP(config)
|
|
self.finished = False
|
|
self.audio = kwargs.pop("audio", None)
|
|
self.files = kwargs.pop("files", None)
|
|
self.keycookie = kwargs.pop("keycookie", None)
|
|
self.authorization = kwargs.pop("authorization", None)
|
|
self.output = output
|
|
self.segments = kwargs.pop("segments", None)
|
|
self.output_extention = None
|
|
channels = kwargs.pop("channels", None)
|
|
codec = kwargs.pop("codec", "h264")
|
|
self.format = f"{codec}-{channels}" if channels else codec
|
|
|
|
def __repr__(self):
|
|
return f"<Video(fetcher={self.__class__.__name__}, bitrate={self.bitrate} format={self.format}>"
|
|
|
|
@property
|
|
def name(self):
|
|
pass
|
|
|
|
def _download_url(self, url, audio=False, total_size=None):
|
|
cookies = self.kwargs["cookies"]
|
|
data = self.http.request("get", url, cookies=cookies, headers={"Range": "bytes=0-8192"})
|
|
if not total_size:
|
|
try:
|
|
total_size = data.headers["Content-Range"]
|
|
total_size = total_size[total_size.find("/") + 1 :]
|
|
total_size = int(total_size)
|
|
except KeyError:
|
|
raise KeyError("Can't get the total size.")
|
|
|
|
bytes_so_far = 8192
|
|
if audio:
|
|
file_d = output(copy.copy(self.output), self.config, "m4a")
|
|
else:
|
|
file_d = output(self.output, self.config, "mp4")
|
|
|
|
if file_d is None:
|
|
return
|
|
file_d.write(data.content)
|
|
eta = ETA(total_size)
|
|
while bytes_so_far < total_size:
|
|
|
|
if not self.config.get("silent"):
|
|
eta.update(bytes_so_far)
|
|
progressbar(total_size, bytes_so_far, "".join(["ETA: ", str(eta)]))
|
|
|
|
old = bytes_so_far + 1
|
|
bytes_so_far = total_size
|
|
|
|
bytes_range = f"bytes={old}-{bytes_so_far}"
|
|
|
|
data = self.http.request("get", url, cookies=cookies, headers={"Range": bytes_range})
|
|
file_d.write(data.content)
|
|
|
|
file_d.close()
|
|
progressbar(bytes_so_far, total_size, "ETA: complete")
|
|
# progress_stream.write('\n')
|
|
self.finished = True
|