MarcoBuster/youtube-dl
1
1
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl synced 2024-12-02 23:02:57 +01:00
Code Issues Releases Wiki Activity
5b21754197
youtube-dl/youtube_dl/extractor/switchtube.py
Bartosz Białas 5b21754197 [SwitchTube] Add new extractor
2022-07-31 20:43:16 +02:00

220 lines
10 KiB
Python
Raw Blame History

# coding: utf-8
from __future__ import unicode_literals
import itertools
import re
from .common import InfoExtractor
from ..utils import (
clean_html,
extract_attributes,
float_or_none,
get_element_by_attribute,
int_or_none,
parse_iso8601,
strip_or_none,
urljoin,
)
class SwitchTubeIE(InfoExtractor):
_VALID_URL = r'https?://tube\.switch\.ch/videos/(?P<id>[\da-zA-Z]+)'
IE_NAME = 'switchtube'
_TESTS = [{
'url': 'https://tube.switch.ch/videos/0T1XfaIFSX',
'info_dict': {
'id': '0T1XfaIFSX',
'title': '2016_ASE_sqC03-Entretien',
'channel': 'ASE Assistant-e socio-éducatif-ve CFC',
'channel_url': 'https://tube.switch.ch/channels/bsaer76yoL',
'channel_id': 'bsaer76yoL',
'ext': 'mp4',
'description': None,
'thumbnail': r're:^https?://tube.switch.ch/image/representations/[\w-]+$',
'license': 'All rights reserved',
'creator': 'Jean-Marc Pouly from Eidgenössische Hochschule für Berufsbildung',
'uploader': 'Jean-Marc Pouly from Eidgenössische Hochschule für Berufsbildung',
'uploader_url': 'https://tube.switch.ch/profiles/42481',
'uploader_id': '42481',
'upload_date': '20220309',
'timestamp': 1646839068,
}
}, {
'url': 'https://tube.switch.ch/videos/0cf3886d',
'info_dict': {
'id': '0cf3886d',
'ext': 'mp4',
'title': 'Introduction: Mini-Batches in On- and Off-Policy Deep Reinforcement Learning',
'license': 'All rights reserved',
'description': 'One of the challenges in Deep Reinforcement Learning is to decorrelate the data. How this is possible with replay buffers is explained here.',
'thumbnail': r're:^https?://tube.switch.ch/image/representations/[\w-]+$',
'channel': 'CS-456 Artificial Neural Networks',
'channel_url': 'https://tube.switch.ch/channels/1deb03e0',
'channel_id': '1deb03e0',
'timestamp': 1590733406,
'upload_date': '20200529',
'creator': 'Wulfram Gerstner from École polytechnique fédérale de Lausanne (EPFL)',
'uploader': 'Wulfram Gerstner from École polytechnique fédérale de Lausanne (EPFL)',
'uploader_url': 'https://tube.switch.ch/profiles/94248',
'uploader_id': '94248',
}
}]
def _real_extract(self, url):
video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id)
title = self._html_search_regex(r'(?s)<title\b[^>]*>(.+?)</title>', webpage, 'title')
info = {
'id': video_id,
'title': title,
'is_live': False,
}
info['view_count'] = int_or_none(self._html_search_regex(r'(?s)<p\b[^>]*>\s*(\d+)\s+views?\s*</p>', webpage,
'view_count', default=None))
info['license'] = self._html_search_regex(r'''(?s)<span\b[^>]*?\bproperty\s*=\s*["'](?:\b[cd]c:license\b\s*){2,}[^>]+>(.+)</span>''',
webpage, 'license', default=None)
info['description'] = strip_or_none(clean_html(
get_element_by_attribute('property', 'dc:description', webpage)))
info['duration'] = float_or_none(
self._search_regex(r'''\bdata-duration\s*=\s*["']([\d.]+)''',
webpage, 'duration', default=None))
channel_groups = self._search_regex(r'''(?s)<a\b[^>]+?\bhref\s*=\s*["']/channels/(.{4,}?)</a>''',
webpage, 'channel groups', default='')
channel_groups = re.split(r'''(?s)["'][^>]*>\s*''', channel_groups, 1)
if len(channel_groups) == 2:
for i, ch in enumerate(('channel_id', 'channel')):
info[ch] = strip_or_none(channel_groups[i])
if info['channel_id']:
info['channel_url'] = 'https://tube.switch.ch/channels/' + info['channel_id']
def outer_elements_by_attribute(attr, value, html, tag=None, escape_value=True, include_inner=False):
"""generate matching HTML element strings
if include_inner, tuples of (element, content)"""
pattern = r'''(?s)(?P<element><(%s)\b[^>]+?\b%s\s*=\s*("|')%s\b[^>]+>)(?P<inner>.*?)</\2>''' % \
(re.escape(tag) if tag is not None else r'\w+', attr, re.escape(value) if escape_value else value)
matches = re.finditer(pattern, html)
for m in matches:
yield m.group('element', 'inner') if include_inner else m.group('element')
for dt in outer_elements_by_attribute('property', 'dc:date', webpage, tag='span'):
dt = extract_attributes(dt)
if dt.get('class') == 'dt-published':
info['timestamp'] = parse_iso8601(dt.get('content'))
break
creator_groups = self._search_regex(
r'''(?s)<span\b[^>]+?\bclass\s*=\s*("|')(?:(?!\1).)*\bp-author\b(?:(?!\1).)*\1\s*property\s*=\s*["']dc:creator\b[^>]+>\s*(.*?<span\b[^>]+?\bclass\s*=\s*["']p-name\b.*</span>).*?</span>''',
webpage, 'creator groups', default='', group=2)
creator_groups = re.match(r'''(?s)<a\b[^>]+?\bhref\s*=\s*("|')/profiles/(?P<profile_id>.+?)\1[^>]*>\s*<span\b[^>]+>\s*(?P<creator_name>.+?)\s*</span>[,\s]*<span\b[^>]+\bclass\s*=\s*["']p-organization-name\b[^>]+>\s*(?P<organization_name>.+?)\s*</span>''', creator_groups)
if creator_groups:
creator_groups = creator_groups.groupdict()
info['uploader'] = info['creator'] = ' from '.join((creator_groups['creator_name'], creator_groups['organization_name']))
info['uploader_id'] = creator_groups['profile_id']
info['uploader_url'] = 'https://tube.switch.ch/profiles/' + info['uploader_id']
parsed_media_entries = self._parse_html5_media_entries(url, webpage, video_id)[0]
info['thumbnail'] = parsed_media_entries['thumbnail']
info['formats'] = parsed_media_entries['formats']
self._sort_formats(info['formats'])
return info
class SwitchTubeProfileIE(InfoExtractor):
_VALID_URL = r'https?://tube\.switch\.ch/profiles/(?P<id>[\da-zA-Z]+)'
IE_NAME = 'switchtube:profile'
_TESTS = [{
'url': 'https://tube.switch.ch/profiles/94248',
'info_dict': {
'id': '94248',
'title': 'Wulfram Gerstner',
'description': None,
},
'playlist_mincount': 94,
}]
@classmethod
def suitable(cls, url):
return False if SwitchTubeIE.suitable(url) else super(SwitchTubeProfileIE, cls).suitable(url)
def _real_extract(self, url):
channel_id = self._match_id(url)
webpage = self._download_webpage(url, channel_id)
channel_title = self._html_search_regex(r'(?s)<title\b[^>]*>(.+?)</title>', webpage, 'title', default=None)
description = self._html_search_regex(r'''(?s)<div\b[^>]+class\s*=\s*("|')p-summary formatted\1[^>]+property\s*=\s*("|')dc:description\2[^>]*>\s*(.+?)\s*</div>''',
webpage, 'description', default=None, group=3)
entries = []
next_page = None
for current_page_number in itertools.count(1):
if next_page:
webpage = self._download_webpage(next_page, channel_id, note='Downloading page %d' % (current_page_number, ))
for _, video_path, video_id, title in re.findall(
r'''(?s)<a\b[^>]+\bhref\s*=\s*("|')(/videos/((?:(?!\1).)+?))\1[^>]*>\s*<div\b[^>]+\bclass\s*=\s*["']title\b[^>]+>(.+?)</div>''',
webpage):
video_url = urljoin(url, video_path)
if video_url:
entries.append(self.url_result(video_url, ie=SwitchTubeIE.ie_key(), video_id=video_id))
next_page = self._search_regex(
r'''<a\b[^>]+?\bhref\s*=\s*("|')(?P<path>/profiles/%s\?(?:(?!\1).)+)\1[^>]*>\s*Next\s*</a>''' % (channel_id,),
webpage, 'next page', group='path', default=None)
if next_page:
next_page = urljoin(url, next_page)
if not next_page:
break
return self.playlist_result(entries, channel_id, channel_title,
description)
class SwitchTubeChannelIE(InfoExtractor):
_VALID_URL = r'https?://tube\.switch\.ch/channels/(?P<id>[\da-zA-Z]+)'
IE_NAME = 'switchtube:channel'
_TESTS = [{
'url': 'https://tube.switch.ch/channels/1deb03e0',
'info_dict': {
'id': '1deb03e0',
'title': 'CS-456 Artificial Neural Networks',
'description': 'Class on Artificial Neural Networks and Reinforcement Learning designed for EPFL master students in CS and related disciplines.'
},
'playlist_mincount': 94,
}]
@classmethod
def suitable(cls, url):
return False if SwitchTubeIE.suitable(url) else super(SwitchTubeChannelIE, cls).suitable(url)
def _real_extract(self, url):
channel_id = self._match_id(url)
entries = []
for current_page_number in itertools.count(0):
page_url = urljoin(url, '/channels/%s?order=episodes&page=%d' % (channel_id, current_page_number))
webpage = self._download_webpage(page_url, channel_id, note='Downloading page %d' % (current_page_number + 1, ))
if current_page_number == 0:
channel_title = self._html_search_regex(r'(?s)<title\b[^>]*>(.+?)</title>', webpage, 'title', default=None)
description = self._html_search_regex(r'''(?s)<div\b[^>]+class\s*=\s*("|')description formatted\1[^>]*>\s*<p>\s*(.+?)\s*</p>\s*</div>''',
webpage, 'description', default=None, group=2)
this_page_still_has_something = False
for _, video_path, video_id in re.findall(
r'''(?s)<a\b[^>]+\bhref\s*=\s*("|')(/videos/((?:(?!\1).)+?))\1[^>]*>''',
webpage):
video_url = urljoin(url, video_path)
if video_url:
this_page_still_has_something = True
entries.append(self.url_result(video_url, ie=SwitchTubeIE.ie_key(), video_id=video_id))
if not this_page_still_has_something:
break
return self.playlist_result(entries, channel_id, channel_title,
description)
Reference in New Issue View Git Blame Copy Permalink