mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-24 12:15:40 +01:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
from __future__ import absolute_import
|
|
import copy
|
|
|
|
from svtplay_dl.utils.output import output, ETA, progressbar
|
|
from svtplay_dl.utils.http import HTTP
|
|
|
|
|
|
class VideoRetriever(object):
|
|
def __init__(self, config, url, bitrate=0, **kwargs):
|
|
self.config = config
|
|
self.url = url
|
|
self.bitrate = int(bitrate)
|
|
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 = kwargs.pop("output", None)
|
|
self.segments = kwargs.pop("segments", None)
|
|
self.output_extention = None
|
|
|
|
def __repr__(self):
|
|
return "<Video(fetcher={}, bitrate={}>".format(self.__class__.__name__, self.bitrate)
|
|
|
|
@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 = "bytes={0}-{1}".format(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
|