]> code.delx.au - webdl/blobdiff - sbs.py
Added license
[webdl] / sbs.py
diff --git a/sbs.py b/sbs.py
index 164482066ca80100ce7185d1d14dff93fc6aca09..bf599417e48bd565199fcd0586a9160c282ded72 100644 (file)
--- a/sbs.py
+++ b/sbs.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
-# vim:ts=4:sts=4:sw=4:noet
 
-from common import grab_text, grab_html, grab_json, grab_xml, download_rtmp, download_urllib, Node, append_to_qs
+from common import grab_text, grab_html, grab_json, grab_xml, download_hds, Node, append_to_qs
 
 import collections
 import urlparse
@@ -10,99 +9,104 @@ BASE = "http://www.sbs.com.au"
 VIDEO_MENU = BASE + "/ondemand/js/video-menu"
 VIDEO_URL = BASE + "/ondemand/video/single/%s"
 VIDEO_MAGIC = {
-       "v": "2.5.14",
-       "fp": "MAC 11,1,102,55",
-       "r": "FLQDD",
-       "g": "YNANAXRIYFYO",
+    "v": "2.5.14",
+    "fp": "MAC 11,1,102,55",
+    "r": "FLQDD",
+    "g": "YNANAXRIYFYO",
 }
 SWF_URL = "http://resources.sbs.com.au/vod/theplatform/core/current/swf/flvPlayer.swf"
 
 NS = {
-       "smil": "http://www.w3.org/2005/SMIL21/Language",
+    "smil": "http://www.w3.org/2005/SMIL21/Language",
 }
 
 
 class SbsNode(Node):
-       def __init__(self, title, parent, video_id):
-               Node.__init__(self, title, parent)
-               self.title = title
-               self.video_id = video_id.split("/")[-1]
-               self.can_download = True
-
-       def download(self):
-               doc = grab_html(VIDEO_URL % self.video_id, 0)
-               meta_video = doc.xpath("//meta[@property='og:video']")[0]
-               swf_url = meta_video.attrib["content"]
-               swf_url_qs = urlparse.parse_qs(urlparse.urlparse(swf_url).query)
-               desc_url = swf_url_qs["v"][0]
-
-               doc = grab_text(desc_url, 0)
-               doc_qs = urlparse.parse_qs(doc)
-               desc_url = doc_qs["releaseUrl"][0]
-
-               doc = grab_xml(desc_url, 0)
-               video = doc.xpath("//smil:video", namespaces=NS)[0]
-               video_url = video.attrib["src"]
-               ext = urlparse.urlsplit(video_url).path.rsplit(".", 1)[1]
-               filename = self.title + "." + ext
-               video_url = append_to_qs(video_url, VIDEO_MAGIC)
-               return download_urllib(filename, video_url, referrer=SWF_URL)
+    def __init__(self, title, parent, video_id):
+        Node.__init__(self, title, parent)
+        self.title = title
+        self.video_id = video_id.split("/")[-1]
+        self.can_download = True
+
+    def download(self):
+        doc = grab_html(VIDEO_URL % self.video_id, 0)
+        meta_video = doc.xpath("//meta[@property='og:video']")[0]
+        swf_url = meta_video.attrib["content"]
+        swf_url_qs = urlparse.parse_qs(urlparse.urlparse(swf_url).query)
+        desc_url = swf_url_qs["v"][0]
+
+        doc = grab_text(desc_url, 0)
+        doc_qs = urlparse.parse_qs(doc)
+        desc_url = doc_qs["releaseUrl"][0]
+
+        doc = grab_xml(desc_url, 0)
+        error = doc.xpath("//smil:param[@name='exception']/@value", namespaces=NS)
+        if error:
+            raise Exception("Error downloading, SBS said: " + error[0])
+
+        video = doc.xpath("//smil:video", namespaces=NS)[0]
+        video_url = video.attrib["src"]
+        if not video_url:
+            raise Exception("Unsupported video '%s': %s" % (self.title, desc_url))
+        filename = self.title + ".flv"
+        video_url = append_to_qs(video_url, VIDEO_MAGIC)
+        return download_hds(filename, video_url, pvswf=SWF_URL)
 
 class SbsNavNode(Node):
-       def __init__(self, title, parent, url):
-               Node.__init__(self, title, parent)
-               self.url = url
-
-       def fill_children(self):
-               try:
-                       doc = grab_json(BASE + self.url, 3600)
-               except ValueError:
-                       # SBS sends XML as an error message :\
-                       return
-               if len(doc.get("entries", [])) == 0:
-                       return
-               for entry in doc["entries"]:
-                       self.fill_entry(entry)
-
-       def fill_entry(self, entry):
-               title = entry["title"]
-               video_id = entry["id"]
-               SbsNode(title, self, video_id)
+    def __init__(self, title, parent, url):
+        Node.__init__(self, title, parent)
+        self.url = url
+
+    def fill_children(self):
+        try:
+            doc = grab_json(BASE + self.url, 3600)
+        except ValueError:
+            # SBS sends XML as an error message :\
+            return
+        if len(doc.get("entries", [])) == 0:
+            return
+        for entry in doc["entries"]:
+            self.fill_entry(entry)
+
+    def fill_entry(self, entry):
+        title = entry["title"]
+        video_id = entry["id"]
+        SbsNode(title, self, video_id)
 
 class SbsRootNode(Node):
-       def __init__(self, parent):
-               Node.__init__(self, "SBS", parent)
-
-       def fill_children(self):
-               menu = grab_json(VIDEO_MENU, 3600, skip_assignment=True)
-               for name in menu.keys():
-                       self.fill_category(self, menu[name])
-
-       def create_nav_node(self, name, parent, cat_data, url_key):
-               try:
-                       url = cat_data[url_key]
-               except KeyError:
-                       return
-               if url.strip():
-                       SbsNavNode(name, parent, url)
-
-       def fill_category(self, parent, cat_data):
-               if not cat_data.has_key("children"):
-                       name = cat_data["name"]
-                       self.create_nav_node(name, parent, cat_data, "url")
-                       return
-
-               node = Node(cat_data["name"], parent)
-               self.create_nav_node("-Featured", node, cat_data, "furl")
-               self.create_nav_node("-Latest", node, cat_data, "url")
-               self.create_nav_node("-Most Popular", node, cat_data, "purl")
-
-               children = cat_data.get("children", [])
-               if isinstance(children, dict):
-                       children = [children[k] for k in sorted(children.keys())]
-               for child_cat in children:
-                       self.fill_category(node, child_cat)
+    def __init__(self, parent):
+        Node.__init__(self, "SBS", parent)
+
+    def fill_children(self):
+        menu = grab_json(VIDEO_MENU, 3600, skip_assignment=True)
+        for name in menu.keys():
+            self.fill_category(self, menu[name])
+
+    def create_nav_node(self, name, parent, cat_data, url_key):
+        try:
+            url = cat_data[url_key]
+        except KeyError:
+            return
+        if url.strip():
+            SbsNavNode(name, parent, url)
+
+    def fill_category(self, parent, cat_data):
+        if not cat_data.has_key("children"):
+            name = cat_data["name"]
+            self.create_nav_node(name, parent, cat_data, "url")
+            return
+
+        node = Node(cat_data["name"], parent)
+        self.create_nav_node("-Featured", node, cat_data, "furl")
+        self.create_nav_node("-Latest", node, cat_data, "url")
+        self.create_nav_node("-Most Popular", node, cat_data, "purl")
+
+        children = cat_data.get("children", [])
+        if isinstance(children, dict):
+            children = [children[k] for k in sorted(children.keys())]
+        for child_cat in children:
+            self.fill_category(node, child_cat)
 
 def fill_nodes(root_node):
-       SbsRootNode(root_node)
+    SbsRootNode(root_node)