code.delx.au - webdl/blob - brightcove.py

   1 import re
   2 import sys
   3
   4 from common import grab_json, download_hls, Node, append_to_qs
   5
   6 CH9_TOKEN = "ogxhPgSphIVa2hhxbi9oqtYwtg032io4B4-ImwddYliFWHqS0UfMEw.."
   7 CH10_TOKEN = "lWCaZyhokufjqe7H4TLpXwHSTnNXtqHxyMvoNOsmYA_GRaZ4zcwysw.."
   8
   9 BRIGHTCOVE_API = "http://api.brightcove.com/services/library?"
  10
  11
  12 class BrightcoveVideoNode(Node):
  13     def __init__(self, title, parent, token, video_id):
  14         Node.__init__(self, title, parent)
  15         self.can_download = True
  16         self.token = token
  17         self.video_id = video_id
  18
  19     def download(self):
  20         desc_url = append_to_qs(BRIGHTCOVE_API, {
  21             "token": self.token,
  22             "command": "find_video_by_id",
  23             "video_fields": "HLSURL",
  24             "video_id": self.video_id,
  25         })
  26         doc = grab_json(desc_url, 3600)
  27         video_url = doc["HLSURL"]
  28         if not video_url:
  29             print("No HLS stream available for: " + self.title)
  30             return False
  31
  32         filename = self.title + ".ts"
  33         return download_hls(filename, video_url)
  34
  35 class BrightcoveRootNode(Node):
  36     def __init__(self, title, parent, token):
  37         Node.__init__(self, title, parent)
  38         self.token = token
  39         self.series_nodes = {}
  40
  41     def get_series_node(self, series_name):
  42         series_name = series_name.split("-")[0].strip()
  43         key = series_name.lower()
  44         node = self.series_nodes.get(key, None)
  45         if node is None:
  46             node = Node(series_name, self)
  47             self.series_nodes[key] = node
  48         return node
  49
  50     def fill_children(self):
  51         page_number = 0
  52         while page_number < 100:
  53             sys.stdout.write(".")
  54             sys.stdout.flush()
  55             url = self.get_all_videos_url(page_number)
  56             page_number += 1
  57
  58             page = grab_json(url, 3600)
  59             items = page["items"]
  60             if len(items) == 0:
  61                 break
  62
  63             for video_desc in items:
  64                 self.process_video(video_desc)
  65         print
  66
  67     def process_video(self, video_desc):
  68         if not video_desc["customFields"]:
  69             return
  70
  71         video_id = video_desc["id"]
  72         title = self.get_video_title(video_desc)
  73         series_name = self.get_series_name(video_desc)
  74
  75         parent_node = self.get_series_node(series_name)
  76         BrightcoveVideoNode(title, parent_node, self.token, video_id)
  77
  78     def get_video_title(self, video_desc):
  79         raise NotImplementedError()
  80
  81     def get_series_name(self, video_desc):
  82         raise NotImplementedError()
  83
  84     def get_all_videos_url(self, page_number):
  85         raise NotImplementedError()
  86
  87
  88 class Ch9RootNode(BrightcoveRootNode):
  89     def __init__(self, root_node):
  90         BrightcoveRootNode.__init__(self, "Nine", root_node, CH9_TOKEN)
  91
  92     def get_video_title(self, video_desc):
  93         title = video_desc["name"]
  94         season = video_desc["customFields"].get("season", "")
  95         episode = video_desc["customFields"].get("episode", "1").rjust(2, "0")
  96         series = self.get_series_name(video_desc)
  97
  98         if re.match(R"ep(isode)?\s*[0-9]+\s*:", title.lower()):
  99             title = title.split(":", 1)[1].strip()
 100
 101         title = "%s - %sx%s - %s" % (
 102             series,
 103             season,
 104             episode,
 105             title
 106         )
 107         return title
 108
 109     def get_series_name(self, video_desc):
 110         series = video_desc["customFields"].get("series", None)
 111         if not series:
 112             series = video_desc["name"]
 113         return series
 114
 115     def get_all_videos_url(self, page_number):
 116         return append_to_qs(BRIGHTCOVE_API, {
 117             "token": self.token,
 118             "command": "search_videos",
 119             "video_fields": "id,name,customFields",
 120             "custom_fields": "series,season,episode",
 121             "sort_by": "PUBLISH_DATE",
 122             "page_number": str(page_number),
 123         })
 124
 125
 126 class Ch10RootNode(BrightcoveRootNode):
 127     def __init__(self, root_node):
 128         BrightcoveRootNode.__init__(self, "Ten", root_node, CH10_TOKEN)
 129
 130     def get_video_title(self, video_desc):
 131         return video_desc["name"]
 132
 133     def get_series_name(self, video_desc):
 134         return video_desc["customFields"]["tv_show"]
 135
 136     def get_all_videos_url(self, page_number):
 137         return append_to_qs(BRIGHTCOVE_API, {
 138             "token": self.token,
 139             "command": "search_videos",
 140             "video_fields": "id,name,customFields",
 141             "custom_fields": "tv_show",
 142             "sort_by": "PUBLISH_DATE",
 143             "any": "video_type_long_form:Full Episode",
 144             "page_number": str(page_number),
 145         })
 146
 147
 148 def fill_nodes(root_node):
 149     Ch9RootNode(root_node)
 150     Ch10RootNode(root_node)
 151