]> code.delx.au - webdl/commitdiff
sbs: ability to download mpd/dash files (only works with streamlink)
authorJames Bunton <jamesbunton@delx.net.au>
Sun, 10 Jun 2018 01:36:31 +0000 (11:36 +1000)
committerJames Bunton <jamesbunton@delx.net.au>
Sun, 10 Jun 2018 01:36:31 +0000 (11:36 +1000)
common.py
sbs.py

index 74e8a9d3e72b9fff4c3054eaaba35b367da43f47..da2051fcf9b7197e34161d6873ec510dcae10200 100644 (file)
--- a/common.py
+++ b/common.py
@@ -282,6 +282,25 @@ def download_hls(filename, video_url):
     else:
         return False
 
+def download_mpd(filename, video_url):
+    streamlink = find_streamlink()
+
+    filename = sanify_filename(filename)
+    video_url = "dash://" + video_url
+    logging.info("Downloading: %s", filename)
+
+    cmd = [
+        streamlink,
+        "-f",
+        "-o", filename,
+        video_url,
+        "best",
+    ]
+    if exec_subprocess(cmd):
+        return convert_to_mp4(filename)
+    else:
+        return False
+
 def download_http(filename, video_url):
     filename = sanify_filename(filename)
     logging.info("Downloading: %s", filename)
diff --git a/sbs.py b/sbs.py
index a26316858846a7c2c3073665639aa301c07b22dc..92b27cdb307282ab8b49bbdcbddb16d5d8cfc03a 100644 (file)
--- a/sbs.py
+++ b/sbs.py
@@ -1,5 +1,5 @@
 import requests_cache
-from common import grab_html, grab_json, grab_xml, download_hls, Node, append_to_qs
+from common import grab_html, grab_json, grab_xml, download_hls, download_mpd, Node, append_to_qs
 
 import json
 
@@ -24,11 +24,13 @@ class SbsVideoNode(Node):
         player_params = self.get_player_params(doc)
         release_url = player_params["releaseUrls"]["html"]
 
-        video_url = self.get_video_url(release_url)
-        if not video_url:
-            raise Exception("Unsupported video %s: %s" % (self.video_id, self.title))
         filename = self.title + ".ts"
-        return download_hls(filename, video_url)
+
+        hls_url = self.get_hls_url(release_url)
+        if hls_url:
+            return download_hls(filename, hls_url)
+        else:
+            return download_mpd(filename, release_url)
 
     def get_player_params(self, doc):
         for script in doc.xpath("//script"):
@@ -43,7 +45,7 @@ class SbsVideoNode(Node):
                         return json.loads(line[p1:p2])
         raise Exception("Unable to find player params for %s: %s" % (self.video_id, self.title))
 
-    def get_video_url(self, release_url):
+    def get_hls_url(self, release_url):
         with requests_cache.disabled():
             doc = grab_xml("http:" + release_url.replace("http:", "").replace("https:", ""))
             video = doc.xpath("//smil:video", namespaces=NS)