mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-24 20:25:41 +01:00
153 lines
4.6 KiB
Python
153 lines
4.6 KiB
Python
# ex:ts=4:sw=4:sts=4:et
|
|
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
|
|
from __future__ import absolute_import
|
|
import sys
|
|
import os
|
|
import re
|
|
|
|
from svtplay_dl.utils import get_http_data, is_py3
|
|
from svtplay_dl.output import progressbar, progress_stream, ETA
|
|
from svtplay_dl.log import log
|
|
from svtplay_dl.utils.urllib import urlparse
|
|
from svtplay_dl.error import UIException
|
|
from svtplay_dl.fetcher import VideoRetriever
|
|
|
|
|
|
class HLSException(UIException):
|
|
def __init__(self, url, message):
|
|
self.url = url
|
|
super(HLSException, self).__init__(message)
|
|
|
|
|
|
class LiveHLSException(HLSException):
|
|
def __init__(self, url):
|
|
super(LiveHLSException, self).__init__(
|
|
url, "This is a live HLS stream, and they are not supported.")
|
|
|
|
|
|
def _get_full_url(url, srcurl):
|
|
if url[:4] == 'http':
|
|
return url
|
|
|
|
urlp = urlparse(srcurl)
|
|
|
|
# remove everything after last / in the path of the URL
|
|
baseurl = re.sub(r'^([^\?]+)/[^/]*(\?.*)?$', r'\1', srcurl)
|
|
returl = "%s/%s" % (baseurl, url)
|
|
|
|
# Append optional query parameters
|
|
if urlp.query:
|
|
returl += "?%s" % urlp.query
|
|
|
|
return returl
|
|
|
|
def hlsparse(url):
|
|
data = get_http_data(url)
|
|
files = (parsem3u(data))[1]
|
|
streams = {}
|
|
|
|
for i in files:
|
|
bitrate = float(i[1]["BANDWIDTH"])/1000
|
|
streams[int(bitrate)] = _get_full_url(i[0], url)
|
|
return streams
|
|
|
|
class HLS(VideoRetriever):
|
|
def name(self):
|
|
return "hls"
|
|
|
|
def download(self):
|
|
if self.options.live and not self.options.force:
|
|
raise LiveHLSException(self.url)
|
|
|
|
m3u8 = get_http_data(self.url)
|
|
globaldata, files = parsem3u(m3u8)
|
|
encrypted = False
|
|
key = None
|
|
try:
|
|
keydata = globaldata["KEY"]
|
|
encrypted = True
|
|
except KeyError:
|
|
pass
|
|
|
|
if encrypted:
|
|
try:
|
|
from Crypto.Cipher import AES
|
|
except ImportError:
|
|
log.error("You need to install pycrypto to download encrypted HLS streams")
|
|
sys.exit(2)
|
|
|
|
match = re.search(r'URI="(https?://.*?)"', keydata)
|
|
key = get_http_data(match.group(1))
|
|
rand = os.urandom(16)
|
|
decryptor = AES.new(key, AES.MODE_CBC, rand)
|
|
if self.options.output != "-":
|
|
extension = re.search(r"(\.[a-z0-9]+)$", self.options.output)
|
|
if not extension:
|
|
self.options.output = "%s.ts" % self.options.output
|
|
log.info("Outfile: %s", self.options.output)
|
|
if os.path.isfile(self.options.output) and not self.options.force:
|
|
log.info("File already exists. use --force to overwrite")
|
|
return
|
|
file_d = open(self.options.output, "wb")
|
|
else:
|
|
if is_py3:
|
|
file_d = sys.stdout.buffer
|
|
else:
|
|
file_d = sys.stdout
|
|
|
|
n = 0
|
|
eta = ETA(len(files))
|
|
for i in files:
|
|
item = _get_full_url(i[0], self.url)
|
|
|
|
if self.options.output != "-":
|
|
eta.increment()
|
|
progressbar(len(files), n, ''.join(['ETA: ', str(eta)]))
|
|
n += 1
|
|
|
|
data = get_http_data(item)
|
|
if encrypted:
|
|
data = decryptor.decrypt(data)
|
|
file_d.write(data)
|
|
|
|
if self.options.output != "-":
|
|
file_d.close()
|
|
progress_stream.write('\n')
|
|
|
|
def parsem3u(data):
|
|
if not data.startswith("#EXTM3U"):
|
|
raise ValueError("Does not apprear to be a ext m3u file")
|
|
|
|
files = []
|
|
streaminfo = {}
|
|
globdata = {}
|
|
|
|
data = data.replace("\r", "\n")
|
|
for l in data.split("\n")[1:]:
|
|
if not l:
|
|
continue
|
|
if l.startswith("#EXT-X-STREAM-INF:"):
|
|
# not a proper parser
|
|
info = [x.strip().split("=", 1) for x in l[18:].split(",")]
|
|
for i in range(0, len(info)):
|
|
if info[i][0] == "BANDWIDTH":
|
|
streaminfo.update({info[i][0]: info[i][1]})
|
|
if info[i][0] == "RESOLUTION":
|
|
streaminfo.update({info[i][0]: info[i][1]})
|
|
elif l.startswith("#EXT-X-ENDLIST"):
|
|
break
|
|
elif l.startswith("#EXT-X-"):
|
|
globdata.update(dict([l[7:].strip().split(":", 1)]))
|
|
elif l.startswith("#EXTINF:"):
|
|
dur, title = l[8:].strip().split(",", 1)
|
|
streaminfo['duration'] = dur
|
|
streaminfo['title'] = title
|
|
elif l[0] == '#':
|
|
pass
|
|
else:
|
|
files.append((l, streaminfo))
|
|
streaminfo = {}
|
|
|
|
return globdata, files
|
|
|