def get_filesize(self):
if not self._fsize:
try:
cl = 'content-length'
opener = compat_opener()
opener.addheaders = [('User-Agent', HEADERS.get('User-Agent'))]
self._fsize = int(opener.open(self.url).headers[cl])
except (compat_urlerr, compat_httperr):
self._fsize = 0
return self._fsize
def download(self,
filepath="",
unsafe=False,
quiet=False,
callback=lambda *x: None):
savedir = filename = ""
retVal = {}
if filepath and os.path.isdir(filepath):
savedir, filename = filepath, self.filename if not unsafe else self.unsafe_filename
elif filepath:
savedir, filename = os.path.split(filepath)
else:
filename = self.filename if not unsafe else self.unsafe_filename
filepath = os.path.join(savedir, filename)
if 'vtt' in filepath and filepath.endswith('.vtt'):
vttfilePath = filepath.replace('.vtt', '.srt')
if os.path.isfile(vttfilePath):
retVal = {"status": "True", "msg": "already downloaded"}
return retVal
if os.path.isfile(filepath):
retVal = {"status": "True", "msg": "already downloaded"}
return retVal
temp_filepath = filepath + ".part"
status_string = (' {:,} Bytes [{:.2%}] received. Rate: [{:4.0f} '
'KB/s]. ETA: [{:.0f} secs]')
if early_py_version:
status_string = (' {0:} Bytes [{1:.2%}] received. Rate:'
' [{2:4.0f} KB/s]. ETA: [{3:.0f} secs]')
try:
req = compat_request(
self.url, headers={'User-Agent': HEADERS.get('User-Agent')})
response = compat_urlopen(req)
except compat_urlerr as e:
retVal = {
"status":
"False",
"msg":
"URLError : either your internet connection is not working or server aborted the request"
}
return retVal
except compat_httperr as e:
if e.code == 401:
retVal = {
"status": "False",
"msg": "Udemy Says (HTTP Error 401 : Unauthorized)"
}
else:
retVal = {
"status":
"False",
"msg":
"HTTPError-{} : direct download link is expired run the udemy-dl with '--skip-sub' option ..."
.format(e.code)
}
return retVal
else:
total = int(response.info()['Content-Length'].strip())
chunksize, bytesdone, t0 = 16384, 0, time.time()
fmode, offset = "wb", 0
if os.path.exists(temp_filepath):
if os.stat(temp_filepath).st_size < total:
offset = os.stat(temp_filepath).st_size
fmode = "ab"
try:
outfh = open(temp_filepath, fmode)
except Exception as e:
if os.name == 'nt':
file_length = len(temp_filepath)
if file_length > 256:
retVal = {
"status":
"False",
"msg":
"file length is too long to create. try downloading to other drive (e.g :- -o 'E:\\')"
}
return retVal
retVal = {"status": "False", "msg": "Reason : {}".format(e)}
return retVal
if offset:
resume_opener = compat_opener()
resume_opener.addheaders = [('User-Agent',
HEADERS.get('User-Agent')),
("Range", "bytes=%s-" % offset)]
try:
response = resume_opener.open(self.url)
except compat_urlerr as e:
retVal = {
"status":
"False",
"msg":
"URLError : either your internet connection is not working or server aborted the request"
}
return retVal
except compat_httperr as e:
if e.code == 401:
retVal = {
"status": "False",
"msg": "Udemy Says (HTTP Error 401 : Unauthorized)"
}
else:
retVal = {
"status":
"False",
"msg":
"HTTPError-{} : direct download link is expired run the udemy-dl with '--skip-sub' option ..."
.format(e.code)
}
return retVal
else:
bytesdone = offset
self._active = True
while self._active:
chunk = response.read(chunksize)
outfh.write(chunk)
elapsed = time.time() - t0
bytesdone += len(chunk)
if elapsed:
try:
rate = ((float(bytesdone) - float(offset)) /
1024.0) / elapsed
eta = (total - bytesdone) / (rate * 1024.0)
except ZeroDivisionError as e:
outfh.close()
try:
os.unlink(temp_filepath)
except Exception as e:
pass
retVal = {
"status":
"False",
"msg":
"ZeroDivisionError : it seems, lecture has malfunction or is zero byte(s) .."
}
return retVal
else:
rate = 0
eta = 0
progress_stats = (bytesdone, bytesdone * 1.0 / total, rate,
eta)
if not chunk:
outfh.close()
break
if not quiet:
status = status_string.format(*progress_stats)
sys.stdout.write("\r" + status + ' ' * 4 + "\r")
sys.stdout.flush()
if callback:
callback(total, *progress_stats)
if self._active:
os.rename(temp_filepath, filepath)
retVal = {"status": "True", "msg": "download"}
else:
outfh.close()
retVal = {"status": "True", "msg": "download"}
return retVal