+#!/usr/bin/env python3
+
+"""A simple tool to query the Oe1 7 Tage archive."""
+
+__version__ = "2.0"
+__author__ = "Stefan Huber"
+
+
+import urllib.request
+import simplejson
+import dateutil.parser
+import sys
+import getopt
+import re
+import os
+
+
+class Archive:
+
+ def __init__(self):
+ self.json = read_json("http://audioapi.orf.at/oe1/json/2.0/broadcasts/")
+
+ def get_days(self):
+ return map(_json_to_day, self.json)
+
+ def get_broadcasts(self, day):
+ bjson = self.json[day]['broadcasts']
+ return map(_json_to_broadcast, bjson)
+
+ def get_broadcast(self, day, broadcast):
+ return _json_to_broadcast(self.json[day]['broadcasts'][broadcast])
+
+ def get_player_url(self, day, broadcast):
+ date = self.json[day]['day']
+ pk = self.json[day]['broadcasts'][broadcast]['programKey']
+ url = "http://oe1.orf.at/player/%d/%s"
+ return url % (date, pk)
+
+ def get_broadcast_subtitle(self, day, broadcast):
+ return self.json[day]['broadcasts'][broadcast]['subtitle']
+
+ def get_broadcast_pk(self, day, broadcast):
+ return self.json[day]['broadcasts'][broadcast]['programKey']
+
+ def get_broadcast_url(self, day, broadcast):
+ date = self.json[day]['day']
+ pk = self.json[day]['broadcasts'][broadcast]['programKey']
+
+ burl = 'https://audioapi.orf.at/oe1/api/json/current/broadcast/%s/%d'
+ bjson = read_json(burl % (pk, date))
+
+ sjson = bjson['streams']
+ if len(sjson) == 0:
+ return None
+
+ sid = sjson[0]['loopStreamId']
+ surl = 'http://loopstream01.apa.at/?channel=oe1&shoutcast=0&id=%s'
+ return surl % sid
+
+ def get_broadcast_description(self, day, broadcast):
+ date = self.json[day]['day']
+ pk = self.json[day]['broadcasts'][broadcast]['programKey']
+
+ burl = 'https://audioapi.orf.at/oe1/api/json/current/broadcast/%s/%d'
+ bjson = read_json(burl % (pk, date))
+
+ description = bjson['description']
+ akm = bjson['akm']
+ if description is None:
+ description = ""
+ if akm is None:
+ akm = ""
+ return description + "<br>" + akm;
+
+ def get_broadcasts_by_regex(self, key):
+ rex = re.compile(key, re.IGNORECASE)
+
+ res = []
+ for d, djson in enumerate(self.json):
+ for b, bjson in enumerate(djson['broadcasts']):
+ if rex.search(bjson['title']) is not None:
+ res.append((d, b))
+ elif bjson['subtitle'] is not None and rex.search(bjson['subtitle']) is not None:
+ res.append((d, b))
+ return res
+
+def _json_to_day(djson):
+ return dateutil.parser.parse(djson['dateISO'])
+
+def _json_to_broadcast(bjson):
+ dt = dateutil.parser.parse(bjson['startISO'])
+ return (dt, bjson['title'])
+
+
+def read_json(url):
+ with urllib.request.urlopen(url) as f:
+ dec = simplejson.JSONDecoder()
+ return dec.decode(f.read())
+
+def input_index(prompt, li):
+ while True:
+ try:
+ idx = int(input(prompt))
+ if idx < 0 or idx >= len(li):
+ print("Out out range!")
+ else:
+ return idx
+
+ except ValueError:
+ print("Unknown input.")
+ except EOFError:
+ sys.exit(1)
+
+def screen_help():
+ print("""Usage:
+ {0} -h, --help
+ {0} -c, --choose
+ {0} -s, --search TITLE""".format(sys.argv[0]))
+
+def screen_choose():
+ a = Archive()
+
+ print("Choose a date:")
+ days = list(a.get_days())
+ for i, date in enumerate(days):
+ print(" [%d] %s" % (i, date.strftime("%a %d. %b %Y")))
+ day = input_index("Date: ", days)
+ chosen_datetime = days[day]
+ print()
+
+ print("Choose a broadcast:")
+ broadcasts = list(a.get_broadcasts(day))
+ for i, b in enumerate(broadcasts):
+ date, title = b
+ print(" [%2d] %s %s" % (i, date.strftime("%H:%M:%S"), title))
+ broadcast = input_index("Broadcast: ", broadcasts)
+ print()
+
+ print_broadcast_info(a, day, broadcast)
+ print()
+
+ url = a.get_broadcast_url(day, broadcast)
+ if url is not None:
+ answer = input("Do you want to download the chosen broadcast? (y/N) ")
+ if answer in ["y", "Y", "j", "J"]:
+ name = input("Download directory (prefix): ")
+
+ try:
+ dirname = get_directory_name(name, chosen_datetime)
+ print("Downloading to %s..." % dirname)
+
+ make_directory(name, chosen_datetime)
+
+ description = a.get_broadcast_description(day, broadcast)
+ write_html_file(name, chosen_datetime, description)
+
+ write_mp3_file(name, chosen_datetime, url)
+
+ except OSError as e:
+ print("Error creating directory.")
+ print(e)
+
+ except requests.exceptions.RequestException as e:
+ print("Request getting mp3 failed.")
+
+ except Exception as e:
+ print("Error downloading mp3.")
+ print(e)
+
+def get_directory_name(name, datetime):
+ prefix = ""
+ if len(name) > 0:
+ prefix = name + "_"
+
+ return prefix + datetime.strftime("%Y-%m-%d")
+
+def make_directory(name, datetime):
+ """Creates the download subdirectory for the given name and datetime."""
+ dirname = get_directory_name(name, datetime)
+ if not os.path.exists(dirname):
+ os.makedirs(dirname)
+
+def write_html_file(name, datetime, description):
+ """Stores broadcast description into a html file."""
+
+ longname = get_directory_name(name, datetime)
+ filepath = os.path.join(longname, "description.html")
+ file = open(filepath, 'w+')
+ file.write("<!DOCTYPE html>\n")
+ file.write("<html>\n")
+ file.write("<head>\n")
+ file.write("<title>\n")
+ file.write("%s %s\n" % (name, datetime.strftime("%d.%m.%Y")))
+ file.write("</title>\n")
+ file.write("<meta charset = \"utf-8\">\n")
+ file.write("</head>\n")
+ file.write("<body>\n")
+ file.write("%s %s" % (name, datetime.strftime("%d.%m.%Y")))
+ file.write(description)
+ file.write("</body>\n")
+ file.write("</html>")
+ file.close()
+
+def write_mp3_file(name, datetime, url):
+ import requests
+
+ longname = get_directory_name(name, datetime)
+ filepath = os.path.join(longname, "stream.mp3")
+
+ print("Fetching mp3...")
+ r = requests.get(url, stream=True)
+ if r.status_code == 200:
+ with open(filepath, 'wb') as f:
+ f.write(r.content)
+ else:
+ print("Error downloading mp3. Status code: %d" % r.status_code)
+
+def screen_search(key):
+ a = Archive()
+ for d, b in a.get_broadcasts_by_regex(key):
+ print_broadcast_info(a, d, b)
+ print()
+
+def print_broadcast_info(archive, day, broadcast):
+ a, d, b = archive, day, broadcast
+ date, title = a.get_broadcast(d, b)
+
+ print("%s %s" % (date.strftime("%a %d.%m.%Y %H:%M:%S"), title))
+ print(" %s" % a.get_broadcast_subtitle(d, b))
+ print(" Broadcast: %s" % a.get_broadcast_url(d, b))
+ print(" Player: %s" % a.get_player_url(d, b))
+ print(" Program key: %s" % a.get_broadcast_pk(d, b))
+
+if __name__ == "__main__":
+
+ try:
+ opts, args = getopt.getopt(sys.argv[1:], "hcs:",
+ ["help", "choose", "search="])
+ except getopt.GetoptError as err:
+ print(err)
+ screen_help()
+ sys.exit(2)
+
+ for o, a in opts:
+ if o in ["-h", "--help"]:
+ screen_help()
+ if o in ["-c", "--choose"]:
+ screen_choose()
+ if o in ["-s", "--search"]:
+ screen_search(a)