+++ /dev/null
-#!/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)
projects / shutils.git / commitdiff