[ardaudiothek] Add support for the ARD Audiothek

This commit is contained in:
TinyToweringTree 2019-09-03 22:28:30 +02:00
parent bff90fc518
commit cd2c7ab40e
2 changed files with 309 additions and 0 deletions

View File

@ -0,0 +1,305 @@
# coding: utf-8
from __future__ import unicode_literals
import re
from .common import InfoExtractor
from ..utils import (
compat_str,
ExtractorError,
int_or_none,
parse_duration,
str_or_none,
try_get,
unified_strdate,
unified_timestamp,
)
class ARDAudiothekBaseIE(InfoExtractor):
def _extract_episode_info(self, title):
"""Try to extract episode data from the title."""
res = {}
if not title:
return res
for pattern in [
r'.*(?P<ep_info> \(S(?P<season_number>\d+)/E(?P<episode_number>\d+)\)).*',
r'.*(?P<ep_info> \((?:Folge |Teil )?(?P<episode_number>\d+)(?:/\d+)?\)).*',
r'.*(?P<ep_info>Folge (?P<episode_number>\d+)(?:\:| -|) )\"(?P<episode>.+)\".*',
r'.*(?P<ep_info>Folge (?P<episode_number>\d+)(?:/\d+)?(?:\:| -|) ).*',
]:
m = re.match(pattern, title)
if m:
groupdict = m.groupdict()
for int_entry in ['season_number', 'episode_number']:
res[int_entry] = int_or_none(groupdict.get(int_entry))
for str_entry in ['episode']:
res[str_entry] = str_or_none(groupdict.get(str_entry))
# Build the episode title by removing numeric episode
# information.
if groupdict.get('ep_info') and not res['episode']:
res['episode'] = str_or_none(
title.replace(groupdict.get('ep_info'), ''))
if res['episode']:
res['episode'] = res['episode'].strip()
break
# As a fallback use the whole title as the episode name
if not res.get('episode'):
res['episode'] = title.strip()
return res
def _extract_id_title_desc(self, json_data):
res = {
'id': try_get(json_data, lambda x: x['id'], compat_str),
'display_id': try_get(json_data, lambda x: x['slug'], compat_str),
}
res['title'] = try_get(
json_data, lambda x: x['title'], compat_str)
res['description'] = try_get(
json_data, lambda x: x['summary'], compat_str)
return res
def _extract_episode(self, ep_data):
res = self._extract_id_title_desc(ep_data)
res['url'] = try_get(ep_data, [
lambda x: x['enclosure']['download_url'],
lambda x: x['enclosure']['playback_url'],
lambda x: x['guid'],
], compat_str)
if not res['url']:
raise ExtractorError(msg='Could not find any downloads',
expected=True)
res['format_note'] = try_get(
ep_data, lambda x: x['enclosure']['type'], compat_str)
res['duration'] = parse_duration(
try_get(ep_data, lambda x: x['duration'], compat_str))
res['release_date'] = unified_strdate(
try_get(ep_data, lambda x: x['publication_date'], compat_str))
res['timestamp'] = unified_timestamp(
try_get(ep_data, lambda x: x['publication_date'], compat_str))
res['channel'] = try_get(ep_data, [
lambda x: x['podcast']['station'],
lambda x: x['podcast']['organization_name'],
], compat_str)
# 'sharing_url' might be a redirecting URL. The generic extractor will
# handle the redirection just fine, so that this extractor here will
# be used.
res['webpage_url'] = try_get(
ep_data, lambda x: x['sharing_url'], compat_str)
res['categories'] = [
try_get(ep_data, lambda x: x['podcast']['category'], compat_str),
]
res['is_live'] = False
res['series'] = try_get(ep_data,
lambda x: x['podcast']['title'],
compat_str)
def make_thumbnail(url, id, preference):
# Note that the images don't necessarily have the advertised
# aspect ratio! So don't set the height based on the aspect
# ratio.
# Also note that the server will not return an image of any given
# width. Most multiples of 32 (or of 64 for higher numbers) seem to
# work. When requesting a width of 1080, the server returns an
# image with a width of 1024, for instance. Requesting 1400 gives
# us 1344, and so on. So a width of 1920 works best for both 1x1
# and 16x9 images.
thumb_width = 1920
return {
'id': id,
# Only set the width if we actually replace the {width}
# placeholder in the URL.
'width': thumb_width if '{width}' in url else None,
'url': url.replace('{width}', str(thumb_width)),
'preference': preference,
}
# We prefer 1x1 images and we prefer episode images. But still provide
# all available images so that the user can choose. We use the
# thumbnail's 'preference' entry to sort them (the higher the better).
# The preferred thumbnail order is:
# (0) podcast-16x9 < (1) episode-16x9
# < (2) podcast-1x1 < (3) episode-1x1
thumbnails = []
for ar_index, aspect_ratio in enumerate(['16x9', '1x1']):
image_key = 'image_%s' % aspect_ratio
image_sources = [
{'name': 'podcast',
'access': lambda x: x['podcast'][image_key]},
{'name': 'episode',
'access': lambda x: x[image_key]},
]
for src_index, src in enumerate(image_sources):
thumb_url = try_get(ep_data, src['access'], compat_str)
if thumb_url:
thumbnails.append(make_thumbnail(
thumb_url,
src['name'] + '-' + aspect_ratio,
ar_index * len(image_sources) + src_index))
res['thumbnails'] = thumbnails
res.update(self._extract_episode_info(res.get('title')))
return res
class ARDAudiothekIE(ARDAudiothekBaseIE):
_VALID_URL = r'https?://(?:www\.|beta\.)?ardaudiothek\.de/(?:[^/]+)/(?:[^/]+)/(?P<id>[0-9]+)(?:/.*)?'
_TESTS = [{
'url': 'https://www.ardaudiothek.de/hoerspiel-pool/virginia-woolf-zum-leuchtturm-1-3-die-tuer-aus-glas/53728640',
'md5': 'dc12a86bb46faadbdba7a8c9b5a24246',
'info_dict': {
'id': '53728640',
'ext': 'mp3',
'title': 'Virginia Woolf: Zum Leuchtturm (1/3) - Die Tür aus Glas',
'description': r're:^Am Anfang steht die Frage.*',
'thumbnail': compat_str,
'timestamp': 1478818860,
'upload_date': '20161110',
}
}, {
'url': 'https://www.ardaudiothek.de/eine-stunde-talk/soziologe-matthias-quent-nicht-neutral-gegenueber-rechtsradikalismus/65904422',
'md5': '326065e45e8172124165c3b0addd4553',
'info_dict': {
'id': '65904422',
'ext': 'mp3',
'title': 'Soziologe Matthias Quent - Nicht neutral gegenüber Rechtsradikalismus',
'description': r're:^Matthias Quent erforscht die Ziele.*',
'thumbnail': compat_str,
'timestamp': 1565809200,
'upload_date': '20190814',
}
}]
def _real_extract(self, url):
episode_id = self._match_id(url)
api_url = 'https://www.ardaudiothek.de/api/episodes/%s' % episode_id
result_data = self._download_json(api_url, episode_id, fatal=False)
ep_data = try_get(result_data, lambda x: x['result']['episode'], dict)
if not ep_data:
raise ExtractorError(msg="Could not find any episode data",
expected=True)
return self._extract_episode(ep_data)
class ARDAudiothekPlaylistIE(ARDAudiothekBaseIE):
_VALID_URL = r'https?://(?:www\.|beta\.)?ardaudiothek\.de/(?!kategorie)(?:[^/]+)/(?P<id>[0-9]+)(?:/.*)?'
_TESTS = [{
'url': 'https://www.ardaudiothek.de/wirtschaft/62037362',
'info_dict': {
'id': '62037362',
'title': 'Wirtschaft',
'description': compat_str,
},
'playlist_mincount': 5,
}, {
'url': 'https://www.ardaudiothek.de/redezeit/7852070',
'info_dict': {
'id': '7852070',
'title': 'Redezeit',
'description': compat_str,
},
'playlist_mincount': 5,
}, {
'url': 'https://www.ardaudiothek.de/nur-fuer-starke-nerven-krimis-und-thriller/51581650/alle',
'info_dict': {
'id': '51581650',
'title': r're:^Nur für starke Nerven',
'description': compat_str,
},
'playlist_mincount': 5,
}]
def _extract_episodes(self, podcast_id, n_entries):
# items_per_page works from 1 up to 2147483647 (2^31 - 1).
# The website calls the API with items_per_page set to 24. Setting it
# to 500 or 1000 would download the data of all episodes in one or two
# pages. Increasing this value might however trigger server errors in
# the future. So to avoid any problems we will keep using the default
# value and just download a few more pages.
items_per_page = 24
page = 1
api_url_template = 'https://www.ardaudiothek.de/api/podcasts/{}/episodes?items_per_page={}{}'
entries = []
while True:
# The API sometimes returns 404s for page=1. So only add that
# parameter if we actually have paginated content.
page_str = '&page=' + compat_str(page) if page > 1 else ''
api_url = api_url_template.format(podcast_id,
items_per_page,
page_str)
result_data = self._download_json(api_url, podcast_id, fatal=False)
episodes = try_get(result_data,
lambda x: x['result']['episodes'],
list)
if episodes is None:
break
for episode in episodes:
entries.append(self._extract_episode(episode))
# Check if we're done
if len(entries) >= n_entries:
break
# Sanity check, just in case
meta_total = try_get(result_data,
lambda x:
x['result']['meta']['episodes']['total'],
(int, float))
meta_pages = try_get(result_data,
lambda x:
x['result']['meta']['episodes']['pages'],
(int, float))
if not meta_total or not meta_pages:
break
page += 1
return entries
def _real_extract(self, url):
podcast_id = self._match_id(url)
api_url = 'https://www.ardaudiothek.de/api/podcasts/%s' % podcast_id
result_data = self._download_json(api_url, podcast_id, fatal=False)
pc_data = try_get(result_data, lambda x: x['result']['podcast'], dict)
if not pc_data:
raise ExtractorError(msg="Could not find any playlist data",
expected=True)
n_entries = try_get(pc_data,
lambda x: x['number_of_elements'],
(int, float))
res = self._extract_id_title_desc(pc_data)
res['_type'] = 'playlist'
res['entries'] = self._extract_episodes(podcast_id, n_entries)
if n_entries > len(res['entries']):
self.to_screen('Only received {} of {} reported episode IDs'
.format(len(res['entries']), n_entries))
return res

View File

@ -57,6 +57,10 @@ from .ard import (
ARDIE, ARDIE,
ARDMediathekIE, ARDMediathekIE,
) )
from .ardaudiothek import (
ARDAudiothekIE,
ARDAudiothekPlaylistIE,
)
from .arte import ( from .arte import (
ArteTVPlus7IE, ArteTVPlus7IE,
ArteTVEmbedIE, ArteTVEmbedIE,