mirror of
https://github.com/ytdl-org/youtube-dl.git
synced 2024-11-16 06:17:27 +00:00
306 lines
12 KiB
Python
306 lines
12 KiB
Python
|
# 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
|