-#!/usr/bin/env python
-# vim:ts=4:sts=4:sw=4:noet
-
-from common import grab_xml, grab_json, download_rtmp, Node
-import itertools
-
-BASE_URL = "http://www.abc.net.au/iview/"
-CONFIG_URL = BASE_URL + "xml/config.xml"
-HASH_URL = BASE_URL + "images/iview.jpg"
-NS = {
- "auth": "http://www.abc.net.au/iView/Services/iViewHandshaker",
-}
-
-class IviewNode(Node):
- def __init__(self, title, parent, params, vpath):
- Node.__init__(self, title, parent)
- self.params = params
- self.vpath = vpath
- self.filename = self.title + "." + vpath.rsplit(".", 1)[1]
- self.can_download = True
-
- def download(self):
- auth_doc = grab_xml(self.params["auth"], 0)
- server = self.params["server_streaming"]
- token = auth_doc.xpath("//auth:token/text()", namespaces=NS)[0]
- playpath = auth_doc.xpath("//auth:path/text()", namespaces=NS)[0]
- if playpath == "playback/_definst_/":
- playpath = "flash/" + playpath
- vbase = server + "?auth=" + token
- vpath, ext = self.vpath.rsplit(".", 1)
- vpath = ext + ":" + playpath + vpath
- return download_rtmp(self.filename, vbase, vpath, HASH_URL)
-
-class IviewSeriesNode(Node):
- def __init__(self, title, parent, params, series_ids):
- Node.__init__(self, title, parent)
- self.params = params
- self.series_ids = series_ids
-
- def fill_children(self):
- for series_id in self.series_ids:
- self.fill_children_for_id(series_id)
-
- def fill_children_for_id(self, series_id):
- series_doc = grab_json(self.params["api"] + "series=" + series_id, 3600)
- for episode_list in series_doc:
- if episode_list["a"] == series_id:
- episode_list = episode_list["f"]
- break
- else:
- return
-
- for episode in episode_list:
- vpath = episode["n"]
- episode_title = episode["b"].strip()
- if not episode_title.startswith(self.title):
- episode_title = self.title + " " + episode_title
- if episode_title.lower().endswith(" (final)"):
- episode_title = episode_title[:-8]
- IviewNode(episode_title, self, self.params, vpath)
-
-class SeriesInfo(object):
- def __init__(self, title):
- self.title = title
- self.series_ids = set()
- self.categories = set()
-
- def add_series_id(self, series_id):
- self.series_ids.add(series_id)
-
- def add_categories(self, categories):
- self.categories.update(categories)
+from common import grab_json, grab_xml, Node, download_hls
+import requests_cache
+import urllib.parse
+
+API_URL = "http://iview.abc.net.au/api"
+AUTH_URL = "http://iview.abc.net.au/auth"
+
+def format_episode_title(series, ep):
+ if ep:
+ return series + " " + ep
+ else:
+ return series
+
+class IviewEpisodeNode(Node):
+ def __init__(self, title, parent, video_key):
+ Node.__init__(self, title, parent)
+ self.video_key = video_key
+ self.filename = title + ".ts"
+ self.can_download = True
+
+ def find_hls_url(self, playlist):
+ for video in playlist:
+ if video["type"] == "program":
+ return video["hls-high"].replace("http:", "https:")
+ raise Exception("Missing hls-high program stream for " + self.video_key)
+
+ def get_auth_details(self):
+ with requests_cache.disabled():
+ auth_doc = grab_xml(AUTH_URL)
+ NS = {
+ "auth": "http://www.abc.net.au/iView/Services/iViewHandshaker",
+ }
+ token = auth_doc.xpath("//auth:tokenhd/text()", namespaces=NS)[0]
+ token_url = auth_doc.xpath("//auth:server/text()", namespaces=NS)[0]
+ token_hostname = urllib.parse.urlparse(token_url).netloc
+ return token, token_hostname
+
+ def add_auth_token_to_url(self, video_url, token, token_hostname):
+ parsed_url = urllib.parse.urlparse(video_url)
+ hacked_url = parsed_url._replace(netloc=token_hostname, query="hdnea=" + token)
+ video_url = urllib.parse.urlunparse(hacked_url)
+ return video_url
+
+ def download(self):
+ info = grab_json(API_URL + "/programs/" + self.video_key)
+ video_url = self.find_hls_url(info["playlist"])
+ token, token_hostname= self.get_auth_details()
+ video_url = self.add_auth_token_to_url(video_url, token, token_hostname)
+ return download_hls(self.filename, video_url)
+
+
+class IviewIndexNode(Node):
+ def __init__(self, title, parent, url):
+ Node.__init__(self, title, parent)
+ self.url = url
+ self.series_map = {}
+
+ def add_episode(self, ep_info):
+ video_key = ep_info["episodeHouseNumber"]
+ series_title = ep_info["seriesTitle"]
+ title = ep_info.get("title", None)
+ episode_title = format_episode_title(series_title, title)
+
+ series_node = self.series_map.get(series_title, None)
+ if not series_node:
+ series_node = Node(series_title, self)
+ self.series_map[series_title] = series_node
+
+ IviewEpisodeNode(episode_title, series_node, video_key)
+
+ def fill_children(self):
+ info = grab_json(self.url)
+ for index_list in info["index"]:
+ for ep_info in index_list["episodes"]:
+ self.add_episode(ep_info)
+
+class IviewFlatNode(Node):
+ def __init__(self, title, parent, url):
+ Node.__init__(self, title, parent)
+ self.url = url
+
+ def add_episode(self, ep_info):
+ video_key = ep_info["episodeHouseNumber"]
+ series_title = ep_info["seriesTitle"]
+ title = ep_info.get("title", None)
+ episode_title = format_episode_title(series_title, title)
+
+ IviewEpisodeNode(episode_title, self, video_key)
+
+ def fill_children(self):
+ info = grab_json(self.url)
+ for ep_info in info:
+ self.add_episode(ep_info)
+