From d8a3486fac547c13f62ac527385b6b3f2caba353 Mon Sep 17 00:00:00 2001
From: Stefan Huber <shuber@sthu.org>
Date: Mon, 3 May 2021 08:14:14 +0200
Subject: [PATCH] Initial version

This is the version transferred from shutils.git commit ab59dc3e.
---
 oe1archive.py | 250 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 250 insertions(+)
 create mode 100755 oe1archive.py

diff --git a/oe1archive.py b/oe1archive.py
new file mode 100755
index 0000000..de4d412
--- /dev/null
+++ b/oe1archive.py
@@ -0,0 +1,250 @@
+#!/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)
-- 
2.30.2