mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-06-27 17:08:32 +00:00
953 lines
37 KiB
Python
953 lines
37 KiB
Python
import datetime as dt
|
||
import functools
|
||
import itertools
|
||
import json
|
||
import re
|
||
import urllib.parse
|
||
|
||
from .common import InfoExtractor, SearchInfoExtractor
|
||
from ..networking import Request
|
||
from ..networking.exceptions import HTTPError
|
||
from ..utils import (
|
||
ExtractorError,
|
||
OnDemandPagedList,
|
||
clean_html,
|
||
determine_ext,
|
||
float_or_none,
|
||
int_or_none,
|
||
parse_bitrate,
|
||
parse_iso8601,
|
||
parse_qs,
|
||
parse_resolution,
|
||
qualities,
|
||
remove_start,
|
||
str_or_none,
|
||
time_seconds,
|
||
unescapeHTML,
|
||
unified_timestamp,
|
||
update_url_query,
|
||
url_basename,
|
||
urlencode_postdata,
|
||
urljoin,
|
||
)
|
||
from ..utils.traversal import find_element, traverse_obj
|
||
|
||
|
||
class NiconicoBaseIE(InfoExtractor):
|
||
_API_BASE = 'https://nvapi.nicovideo.jp'
|
||
_BASE_URL = 'https://www.nicovideo.jp'
|
||
_GEO_BYPASS = False
|
||
_GEO_COUNTRIES = ['JP']
|
||
_HEADERS = {
|
||
'X-Frontend-ID': '6',
|
||
'X-Frontend-Version': '0',
|
||
}
|
||
_LOGIN_BASE = 'https://account.nicovideo.jp'
|
||
_NETRC_MACHINE = 'niconico'
|
||
|
||
@property
|
||
def is_logged_in(self):
|
||
return bool(self._get_cookies('https://www.nicovideo.jp').get('user_session'))
|
||
|
||
def _raise_login_error(self, message, expected=True):
|
||
raise ExtractorError(f'Unable to login: {message}', expected=expected)
|
||
|
||
def _perform_login(self, username, password):
|
||
if self.is_logged_in:
|
||
return
|
||
|
||
self._request_webpage(
|
||
f'{self._LOGIN_BASE}/login', None, 'Requesting session cookies')
|
||
webpage = self._download_webpage(
|
||
f'{self._LOGIN_BASE}/login/redirector', None,
|
||
'Logging in', 'Unable to log in', headers={
|
||
'Content-Type': 'application/x-www-form-urlencoded',
|
||
'Referer': f'{self._LOGIN_BASE}/login',
|
||
}, data=urlencode_postdata({
|
||
'mail_tel': username,
|
||
'password': password,
|
||
}))
|
||
|
||
if self.is_logged_in:
|
||
return
|
||
elif err_msg := traverse_obj(webpage, (
|
||
{find_element(cls='notice error')}, {find_element(cls='notice__text')}, {clean_html},
|
||
)):
|
||
self._raise_login_error(err_msg or 'Invalid username or password')
|
||
elif 'oneTimePw' in webpage:
|
||
post_url = self._search_regex(
|
||
r'<form[^>]+action=(["\'])(?P<url>.+?)\1', webpage, 'post url', group='url')
|
||
mfa, urlh = self._download_webpage_handle(
|
||
urljoin(self._LOGIN_BASE, post_url), None,
|
||
'Performing MFA', 'Unable to complete MFA', headers={
|
||
'Content-Type': 'application/x-www-form-urlencoded',
|
||
}, data=urlencode_postdata({
|
||
'otp': self._get_tfa_info('6 digit number shown on app'),
|
||
}))
|
||
if self.is_logged_in:
|
||
return
|
||
elif 'error-code' in parse_qs(urlh.url):
|
||
err_msg = traverse_obj(mfa, ({find_element(cls='pageMainMsg')}, {clean_html}))
|
||
self._raise_login_error(err_msg or 'MFA session expired')
|
||
elif 'formError' in mfa:
|
||
err_msg = traverse_obj(mfa, (
|
||
{find_element(cls='formError')}, {find_element(tag='div')}, {clean_html}))
|
||
self._raise_login_error(err_msg or 'MFA challenge failed')
|
||
|
||
self._raise_login_error('Unexpected login error', expected=False)
|
||
|
||
|
||
class NiconicoIE(NiconicoBaseIE):
|
||
IE_NAME = 'niconico'
|
||
IE_DESC = 'ニコニコ動画'
|
||
|
||
_ERROR_MAP = {
|
||
'FORBIDDEN': {
|
||
'ADMINISTRATOR_DELETE_VIDEO': 'Video unavailable, possibly removed by admins',
|
||
'CHANNEL_MEMBER_ONLY': 'Channel members only',
|
||
'DELETED_CHANNEL_VIDEO': 'Video unavailable, channel was closed',
|
||
'DELETED_COMMUNITY_VIDEO': 'Video unavailable, community deleted or missing',
|
||
'DEFAULT': 'Page unavailable, check the URL',
|
||
'HARMFUL_VIDEO': 'Sensitive content, login required',
|
||
'HIDDEN_VIDEO': 'Video unavailable, set to private',
|
||
'NOT_ALLOWED': 'No parmission',
|
||
'PPV_VIDEO': 'PPV video, payment information required',
|
||
'PREMIUM_ONLY': 'Premium members only',
|
||
},
|
||
'INVALID_PARAMETER': {
|
||
'DEFAULT': 'Video unavailable, may not exist or was deleted',
|
||
},
|
||
'MAINTENANCE': {
|
||
'DEFAULT': 'Maintenance is in progress',
|
||
},
|
||
'NOT_FOUND': {
|
||
'DEFAULT': 'Video unavailable, may not exist or was deleted',
|
||
'RIGHT_HOLDER_DELETE_VIDEO': 'Removed by rights-holder request',
|
||
},
|
||
'UNAUTHORIZED': {
|
||
'DEFAULT': 'Invalid session, re-login required',
|
||
},
|
||
'UNKNOWN': {
|
||
'DEFAULT': 'Failed to fetch content',
|
||
},
|
||
}
|
||
_STATUS_MAP = {
|
||
'needs_auth': 'PPV video, payment information required',
|
||
'premium_only': 'Premium members only',
|
||
'subscriber_only': 'Channel members only',
|
||
}
|
||
_VALID_URL = r'https?://(?:(?:embed|sp|www)\.)?nicovideo\.jp/watch/(?P<id>(?:[a-z]{2})?\d+)'
|
||
_TESTS = [{
|
||
'url': 'http://www.nicovideo.jp/watch/sm22312215',
|
||
'info_dict': {
|
||
'id': 'sm22312215',
|
||
'ext': 'mp4',
|
||
'title': 'Big Buck Bunny',
|
||
'thumbnail': r're:https?://.*',
|
||
'uploader': 'takuya0301',
|
||
'uploader_id': '2698420',
|
||
'upload_date': '20131123',
|
||
'timestamp': int, # timestamp is unstable
|
||
'description': '(c) copyright 2008, Blender Foundation / www.bigbuckbunny.org',
|
||
'duration': 33,
|
||
'view_count': int,
|
||
'comment_count': int,
|
||
'genres': ['未設定'],
|
||
'tags': [],
|
||
},
|
||
'params': {'skip_download': 'm3u8'},
|
||
'skip': True,
|
||
}, {
|
||
# File downloaded with and without credentials are different, so omit
|
||
# the md5 field
|
||
'url': 'http://www.nicovideo.jp/watch/nm14296458',
|
||
'info_dict': {
|
||
'id': 'nm14296458',
|
||
'ext': 'mp4',
|
||
'title': '【Kagamine Rin】Dance on media【Original】take2!',
|
||
'description': 'md5:9368f2b1f4178de64f2602c2f3d6cbf5',
|
||
'thumbnail': r're:https?://.*',
|
||
'uploader': 'りょうた',
|
||
'uploader_id': '18822557',
|
||
'upload_date': '20110429',
|
||
'timestamp': 1304065916,
|
||
'duration': 208.0,
|
||
'comment_count': int,
|
||
'view_count': int,
|
||
'genres': ['音楽・サウンド'],
|
||
'tags': ['Translation_Request', 'Kagamine_Rin', 'Rin_Original'],
|
||
},
|
||
'params': {'skip_download': 'm3u8'},
|
||
'skip': True,
|
||
}, {
|
||
# 'video exists but is marked as "deleted"
|
||
# md5 is unstable
|
||
'url': 'http://www.nicovideo.jp/watch/sm10000',
|
||
'info_dict': {
|
||
'id': 'sm10000',
|
||
'ext': 'unknown_video',
|
||
'description': 'deleted',
|
||
'title': 'ドラえもんエターナル第3話「決戦第3新東京市」＜前編＞',
|
||
'thumbnail': r're:https?://.*',
|
||
'upload_date': '20071224',
|
||
'timestamp': int, # timestamp field has different value if logged in
|
||
'duration': 304,
|
||
'view_count': int,
|
||
},
|
||
'skip': 'Requires an account',
|
||
}, {
|
||
'url': 'http://www.nicovideo.jp/watch/so22543406',
|
||
'info_dict': {
|
||
'id': '1388129933',
|
||
'ext': 'mp4',
|
||
'title': '【第1回】RADIOアニメロミックス ラブライブ！～のぞえりRadio Garden～',
|
||
'description': 'md5:b27d224bb0ff53d3c8269e9f8b561cf1',
|
||
'thumbnail': r're:https?://.*',
|
||
'timestamp': 1388851200,
|
||
'upload_date': '20140104',
|
||
'uploader': 'アニメロチャンネル',
|
||
'uploader_id': '312',
|
||
},
|
||
'skip': 'The viewing period of the video you were searching for has expired.',
|
||
}, {
|
||
# video not available via `getflv`; "old" HTML5 video
|
||
'url': 'http://www.nicovideo.jp/watch/sm1151009',
|
||
'info_dict': {
|
||
'id': 'sm1151009',
|
||
'ext': 'mp4',
|
||
'title': 'マスターシステム本体内蔵のスペハリのメインテーマ（ＰＳＧ版）',
|
||
'description': 'md5:f95a3d259172667b293530cc2e41ebda',
|
||
'thumbnail': r're:https?://.*',
|
||
'duration': 184,
|
||
'timestamp': 1190835883,
|
||
'upload_date': '20070926',
|
||
'uploader': 'denden2',
|
||
'uploader_id': '1392194',
|
||
'view_count': int,
|
||
'comment_count': int,
|
||
'genres': ['ゲーム'],
|
||
'tags': [],
|
||
},
|
||
'params': {'skip_download': 'm3u8'},
|
||
'skip': True,
|
||
}, {
|
||
# "New" HTML5 video
|
||
'url': 'http://www.nicovideo.jp/watch/sm31464864',
|
||
'info_dict': {
|
||
'id': 'sm31464864',
|
||
'ext': 'mp4',
|
||
'title': '新作TVアニメ「戦姫絶唱シンフォギアAXZ」PV 最高画質',
|
||
'description': 'md5:e52974af9a96e739196b2c1ca72b5feb',
|
||
'timestamp': 1498481660,
|
||
'upload_date': '20170626',
|
||
'uploader': 'no-namamae',
|
||
'uploader_id': '40826363',
|
||
'thumbnail': r're:https?://.*',
|
||
'duration': 198,
|
||
'view_count': int,
|
||
'comment_count': int,
|
||
'genres': ['アニメ'],
|
||
'tags': [],
|
||
},
|
||
'params': {'skip_download': 'm3u8'},
|
||
'skip': True,
|
||
}, {
|
||
# Video without owner
|
||
'url': 'http://www.nicovideo.jp/watch/sm18238488',
|
||
'info_dict': {
|
||
'id': 'sm18238488',
|
||
'ext': 'mp4',
|
||
'title': '【実写版】ミュータントタートルズ',
|
||
'description': 'md5:15df8988e47a86f9e978af2064bf6d8e',
|
||
'timestamp': 1341128008,
|
||
'upload_date': '20120701',
|
||
'thumbnail': r're:https?://.*',
|
||
'duration': 5271,
|
||
'view_count': int,
|
||
'comment_count': int,
|
||
'genres': ['エンターテイメント'],
|
||
'tags': [],
|
||
},
|
||
'params': {'skip_download': 'm3u8'},
|
||
'skip': True,
|
||
}, {
|
||
'url': 'http://sp.nicovideo.jp/watch/sm28964488?ss_pos=1&cp_in=wt_tg',
|
||
'only_matching': True,
|
||
}, {
|
||
'note': 'a video that is only served as an ENCRYPTED HLS.',
|
||
'url': 'https://www.nicovideo.jp/watch/so38016254',
|
||
'only_matching': True,
|
||
}]
|
||
|
||
def _extract_formats(self, api_data, video_id):
|
||
fmt_filter = lambda _, v: v['isAvailable'] and v['id']
|
||
videos = traverse_obj(api_data, ('media', 'domand', 'videos', fmt_filter))
|
||
audios = traverse_obj(api_data, ('media', 'domand', 'audios', fmt_filter))
|
||
access_key = traverse_obj(api_data, ('media', 'domand', 'accessRightKey', {str}))
|
||
track_id = traverse_obj(api_data, ('client', 'watchTrackId', {str}))
|
||
if not all((videos, audios, access_key, track_id)):
|
||
return
|
||
|
||
m3u8_url = self._download_json(
|
||
f'{self._API_BASE}/v1/watch/{video_id}/access-rights/hls',
|
||
video_id, headers={
|
||
'Accept': 'application/json;charset=utf-8',
|
||
'Content-Type': 'application/json',
|
||
'X-Access-Right-Key': access_key,
|
||
'X-Request-With': self._BASE_URL,
|
||
**self._HEADERS,
|
||
}, query={
|
||
'actionTrackId': track_id,
|
||
}, data=json.dumps({
|
||
'outputs': list(itertools.product((v['id'] for v in videos), (a['id'] for a in audios))),
|
||
}).encode(),
|
||
)['data']['contentUrl']
|
||
raw_fmts = self._extract_m3u8_formats(m3u8_url, video_id, 'mp4')
|
||
|
||
formats = []
|
||
for a_fmt in traverse_obj(raw_fmts, lambda _, v: v['vcodec'] == 'none'):
|
||
formats.append({
|
||
**a_fmt,
|
||
**traverse_obj(audios, (lambda _, v: a_fmt['format_id'].startswith(v['id']), {
|
||
'abr': ('bitRate', {float_or_none(scale=1000)}),
|
||
'asr': ('samplingRate', {int_or_none}),
|
||
'format_id': ('id', {str}),
|
||
'quality': ('qualityLevel', {int_or_none}),
|
||
}, any)),
|
||
'acodec': 'aac',
|
||
})
|
||
|
||
# Sort first, keeping the lowest-tbr formats
|
||
v_fmts = sorted((fmt for fmt in raw_fmts if fmt['vcodec'] != 'none'), key=lambda f: f['tbr'])
|
||
self._remove_duplicate_formats(v_fmts)
|
||
# Calculate the true vbr/tbr by subtracting the lowest abr
|
||
min_abr = traverse_obj(audios, (
|
||
..., 'bitRate', {float_or_none(scale=1000)}, all, {min}), default=0)
|
||
for v_fmt in v_fmts:
|
||
v_fmt['format_id'] = url_basename(v_fmt['url']).rpartition('.')[0]
|
||
v_fmt['quality'] = traverse_obj(videos, (
|
||
lambda _, v: v['id'] == v_fmt['format_id'], 'qualityLevel', {int_or_none}, any), default=-1)
|
||
v_fmt['tbr'] -= min_abr
|
||
formats.extend(v_fmts)
|
||
|
||
return formats
|
||
|
||
def _real_extract(self, url):
|
||
video_id = self._match_id(url)
|
||
|
||
path = 'v3' if self.is_logged_in else 'v3_guest'
|
||
api_resp = self._download_json(
|
||
f'{self._BASE_URL}/api/watch/{path}/{video_id}', video_id,
|
||
'Downloading API JSON', 'Unable to fetch data', headers={
|
||
**self._HEADERS,
|
||
**self.geo_verification_headers(),
|
||
}, query={
|
||
'actionTrackId': f'AAAAAAAAAA_{round(time_seconds() * 1000)}',
|
||
}, expected_status=[400, 404])
|
||
|
||
api_data = api_resp['data']
|
||
release_timestamp = traverse_obj(api_data, ('publishScheduledAt', {parse_iso8601}))
|
||
|
||
meta = api_resp['meta']
|
||
if meta.get('status') != 200:
|
||
err_code = meta['errorCode']
|
||
reason_code = traverse_obj(api_data, ('reasonCode', {str_or_none}))
|
||
err_msg = 'Server busy, service temporarily unavailable'
|
||
|
||
if reason_code in ('DOMESTIC_VIDEO', 'HIGH_RISK_COUNTRY_VIDEO'):
|
||
self.raise_geo_restricted(countries=self._GEO_COUNTRIES)
|
||
elif reason_code == 'HARMFUL_VIDEO' and traverse_obj(api_data, (
|
||
'viewer', 'allowSensitiveContents', {bool},
|
||
)) is False:
|
||
err_msg = 'Sensitive content, adjust display settings to watch'
|
||
elif reason_code == 'HIDDEN_VIDEO' and release_timestamp:
|
||
err_msg = f'Scheduled release, please wait. Release time: {release_timestamp}'
|
||
elif msg := traverse_obj(self._ERROR_MAP, (
|
||
err_code.upper(), (reason_code, 'DEFAULT'), {str}, any,
|
||
)):
|
||
err_msg = msg
|
||
|
||
raise ExtractorError(err_msg, expected=True)
|
||
|
||
availability = self._availability(**{
|
||
**dict.fromkeys(('is_private', 'is_unlisted'), False),
|
||
**traverse_obj(api_data, ('payment', 'video', {
|
||
'needs_auth': (('isContinuationBenefit', 'isPpv'), {bool}, any),
|
||
'needs_premium': ('isPremium', {bool}),
|
||
'needs_subscription': ('isAdmission', {bool}),
|
||
})),
|
||
})
|
||
|
||
formats = self._extract_formats(api_data, video_id)
|
||
if not formats:
|
||
if err_msg := self._STATUS_MAP.get(availability):
|
||
self.raise_login_required(err_msg, metadata_available=True)
|
||
|
||
thumb_prefs = qualities(['url', 'middleUrl', 'largeUrl', 'player', 'ogp'])
|
||
|
||
return {
|
||
'availability': availability,
|
||
'display_id': video_id,
|
||
'formats': formats,
|
||
'genre': traverse_obj(api_data, ('genre', 'label', {str})),
|
||
'release_timestamp': release_timestamp,
|
||
'subtitles': self.extract_subtitles(video_id, api_data),
|
||
'tags': traverse_obj(api_data, ('tag', 'items', ..., 'name', {str})),
|
||
'thumbnails': [{
|
||
'ext': 'jpg',
|
||
'id': key,
|
||
'preference': thumb_prefs(key),
|
||
'url': url,
|
||
**parse_resolution(url, lenient=True),
|
||
} for key, url in traverse_obj(api_data, (
|
||
'video', 'thumbnail', {dict}), default={}).items()],
|
||
**traverse_obj(api_data, (('channel', 'owner'), any, {
|
||
'channel': (('name', 'nickname'), {str}, any),
|
||
'channel_id': ('id', {str_or_none}),
|
||
'uploader': (('name', 'nickname'), {str}, any),
|
||
'uploader_id': ('id', {str_or_none}),
|
||
})),
|
||
**traverse_obj(api_data, ('video', {
|
||
'id': ('id', {str_or_none}),
|
||
'title': ('title', {str}),
|
||
'description': ('description', {clean_html}),
|
||
'duration': ('duration', {int_or_none}),
|
||
'timestamp': ('registeredAt', {parse_iso8601}),
|
||
})),
|
||
**traverse_obj(api_data, ('video', 'count', {
|
||
'comment_count': ('comment', {int_or_none}),
|
||
'like_count': ('like', {int_or_none}),
|
||
'view_count': ('view', {int_or_none}),
|
||
})),
|
||
}
|
||
|
||
def _get_subtitles(self, video_id, api_data):
|
||
comments_info = traverse_obj(api_data, ('comment', 'nvComment', {dict}), default={})
|
||
if not comments_info.get('server'):
|
||
return
|
||
|
||
danmaku = traverse_obj(self._download_json(
|
||
f'{comments_info["server"]}/v1/threads', video_id,
|
||
'Downloading comments', 'Failed to download comments', headers={
|
||
'Content-Type': 'text/plain;charset=UTF-8',
|
||
'Origin': self._BASE_URL,
|
||
'Referer': f'{self._BASE_URL}/',
|
||
'X-Client-Os-Type': 'others',
|
||
**self._HEADERS,
|
||
}, data=json.dumps({
|
||
'additionals': {},
|
||
'params': comments_info.get('params'),
|
||
'threadKey': comments_info.get('threadKey'),
|
||
}).encode(), fatal=False,
|
||
), ('data', 'threads', ..., 'comments', ...))
|
||
|
||
return {
|
||
'comments': [{
|
||
'ext': 'json',
|
||
'data': json.dumps(danmaku),
|
||
}],
|
||
}
|
||
|
||
|
||
class NiconicoPlaylistBaseIE(InfoExtractor):
|
||
_PAGE_SIZE = 100
|
||
|
||
_API_HEADERS = {
|
||
'X-Frontend-ID': '6',
|
||
'X-Frontend-Version': '0',
|
||
'X-Niconico-Language': 'en-us',
|
||
}
|
||
|
||
def _call_api(self, list_id, resource, query):
|
||
raise NotImplementedError('Must be implemented in subclasses')
|
||
|
||
@staticmethod
|
||
def _parse_owner(item):
|
||
return {
|
||
'uploader': traverse_obj(item, ('owner', ('name', ('user', 'nickname')), {str}, any)),
|
||
'uploader_id': traverse_obj(item, ('owner', 'id', {str})),
|
||
}
|
||
|
||
def _fetch_page(self, list_id, page):
|
||
page += 1
|
||
resp = self._call_api(list_id, f'page {page}', {
|
||
'page': page,
|
||
'pageSize': self._PAGE_SIZE,
|
||
})
|
||
# this is needed to support both mylist and user
|
||
for video in traverse_obj(resp, ('items', ..., ('video', None))) or []:
|
||
video_id = video.get('id')
|
||
if not video_id:
|
||
# skip {"video": {"id": "blablabla", ...}}
|
||
continue
|
||
count = video.get('count') or {}
|
||
get_count = lambda x: int_or_none(count.get(x))
|
||
yield {
|
||
'_type': 'url',
|
||
'id': video_id,
|
||
'title': video.get('title'),
|
||
'url': f'https://www.nicovideo.jp/watch/{video_id}',
|
||
'description': video.get('shortDescription'),
|
||
'duration': int_or_none(video.get('duration')),
|
||
'view_count': get_count('view'),
|
||
'comment_count': get_count('comment'),
|
||
'thumbnail': traverse_obj(video, ('thumbnail', ('nHdUrl', 'largeUrl', 'listingUrl', 'url'))),
|
||
'ie_key': NiconicoIE.ie_key(),
|
||
**self._parse_owner(video),
|
||
}
|
||
|
||
def _entries(self, list_id):
|
||
return OnDemandPagedList(functools.partial(self._fetch_page, list_id), self._PAGE_SIZE)
|
||
|
||
|
||
class NiconicoPlaylistIE(NiconicoPlaylistBaseIE):
|
||
IE_NAME = 'niconico:playlist'
|
||
_VALID_URL = r'https?://(?:(?:www\.|sp\.)?nicovideo\.jp|nico\.ms)/(?:user/\d+/)?(?:my/)?mylist/(?:#/)?(?P<id>\d+)'
|
||
|
||
_TESTS = [{
|
||
'url': 'http://www.nicovideo.jp/mylist/27411728',
|
||
'info_dict': {
|
||
'id': '27411728',
|
||
'title': 'AKB48のオールナイトニッポン',
|
||
'description': 'md5:d89694c5ded4b6c693dea2db6e41aa08',
|
||
'uploader': 'のっく',
|
||
'uploader_id': '805442',
|
||
},
|
||
'playlist_mincount': 291,
|
||
}, {
|
||
'url': 'https://www.nicovideo.jp/user/805442/mylist/27411728',
|
||
'only_matching': True,
|
||
}, {
|
||
'url': 'https://www.nicovideo.jp/my/mylist/#/68048635',
|
||
'only_matching': True,
|
||
}]
|
||
|
||
def _call_api(self, list_id, resource, query):
|
||
return self._download_json(
|
||
f'https://nvapi.nicovideo.jp/v2/mylists/{list_id}', list_id,
|
||
f'Downloading {resource}', query=query,
|
||
headers=self._API_HEADERS)['data']['mylist']
|
||
|
||
def _real_extract(self, url):
|
||
list_id = self._match_id(url)
|
||
mylist = self._call_api(list_id, 'list', {
|
||
'pageSize': 1,
|
||
})
|
||
return self.playlist_result(
|
||
self._entries(list_id), list_id,
|
||
mylist.get('name'), mylist.get('description'), **self._parse_owner(mylist))
|
||
|
||
|
||
class NiconicoSeriesIE(NiconicoPlaylistBaseIE):
|
||
IE_NAME = 'niconico:series'
|
||
_VALID_URL = r'https?://(?:(?:www\.|sp\.)?nicovideo\.jp(?:/user/\d+)?|nico\.ms)/series/(?P<id>\d+)'
|
||
|
||
_TESTS = [{
|
||
'url': 'https://www.nicovideo.jp/user/44113208/series/110226',
|
||
'info_dict': {
|
||
'id': '110226',
|
||
'title': 'ご立派ァ！のシリーズ',
|
||
'description': '楽しそうな外人の吹き替えをさせたら終身名誉ホモガキの右に出る人はいませんね…',
|
||
'uploader': 'アルファるふぁ',
|
||
'uploader_id': '44113208',
|
||
},
|
||
'playlist_mincount': 10,
|
||
}, {
|
||
'url': 'https://www.nicovideo.jp/series/12312/',
|
||
'info_dict': {
|
||
'id': '12312',
|
||
'title': 'バトルスピリッツ お勧めカード紹介(調整中)',
|
||
'description': '',
|
||
'uploader': '野鳥',
|
||
'uploader_id': '2275360',
|
||
},
|
||
'playlist_mincount': 103,
|
||
}, {
|
||
'url': 'https://nico.ms/series/203559',
|
||
'only_matching': True,
|
||
}]
|
||
|
||
def _call_api(self, list_id, resource, query):
|
||
return self._download_json(
|
||
f'https://nvapi.nicovideo.jp/v2/series/{list_id}', list_id,
|
||
f'Downloading {resource}', query=query,
|
||
headers=self._API_HEADERS)['data']
|
||
|
||
def _real_extract(self, url):
|
||
list_id = self._match_id(url)
|
||
series = self._call_api(list_id, 'list', {
|
||
'pageSize': 1,
|
||
})['detail']
|
||
|
||
return self.playlist_result(
|
||
self._entries(list_id), list_id,
|
||
series.get('title'), series.get('description'), **self._parse_owner(series))
|
||
|
||
|
||
class NiconicoHistoryIE(NiconicoPlaylistBaseIE):
|
||
IE_NAME = 'niconico:history'
|
||
IE_DESC = 'NicoNico user history or likes. Requires cookies.'
|
||
_VALID_URL = r'https?://(?:www\.|sp\.)?nicovideo\.jp/my/(?P<id>history(?:/like)?)'
|
||
|
||
_TESTS = [{
|
||
'note': 'PC page, with /video',
|
||
'url': 'https://www.nicovideo.jp/my/history/video',
|
||
'only_matching': True,
|
||
}, {
|
||
'note': 'PC page, without /video',
|
||
'url': 'https://www.nicovideo.jp/my/history',
|
||
'only_matching': True,
|
||
}, {
|
||
'note': 'mobile page, with /video',
|
||
'url': 'https://sp.nicovideo.jp/my/history/video',
|
||
'only_matching': True,
|
||
}, {
|
||
'note': 'mobile page, without /video',
|
||
'url': 'https://sp.nicovideo.jp/my/history',
|
||
'only_matching': True,
|
||
}, {
|
||
'note': 'PC page',
|
||
'url': 'https://www.nicovideo.jp/my/history/like',
|
||
'only_matching': True,
|
||
}, {
|
||
'note': 'Mobile page',
|
||
'url': 'https://sp.nicovideo.jp/my/history/like',
|
||
'only_matching': True,
|
||
}]
|
||
|
||
def _call_api(self, list_id, resource, query):
|
||
path = 'likes' if list_id == 'history/like' else 'watch/history'
|
||
return self._download_json(
|
||
f'https://nvapi.nicovideo.jp/v1/users/me/{path}', list_id,
|
||
f'Downloading {resource}', query=query, headers=self._API_HEADERS)['data']
|
||
|
||
def _real_extract(self, url):
|
||
list_id = self._match_id(url)
|
||
try:
|
||
mylist = self._call_api(list_id, 'list', {'pageSize': 1})
|
||
except ExtractorError as e:
|
||
if isinstance(e.cause, HTTPError) and e.cause.status == 401:
|
||
self.raise_login_required('You have to be logged in to get your history')
|
||
raise
|
||
return self.playlist_result(self._entries(list_id), list_id, **self._parse_owner(mylist))
|
||
|
||
|
||
class NicovideoSearchBaseIE(InfoExtractor):
|
||
_SEARCH_TYPE = 'search'
|
||
|
||
def _entries(self, url, item_id, query=None, note='Downloading page %(page)s'):
|
||
query = query or {}
|
||
pages = [query['page']] if 'page' in query else itertools.count(1)
|
||
for page_num in pages:
|
||
query['page'] = str(page_num)
|
||
webpage = self._download_webpage(url, item_id, query=query, note=note % {'page': page_num})
|
||
results = re.findall(r'(?<=data-video-id=)["\']?(?P<videoid>.*?)(?=["\'])', webpage)
|
||
for item in results:
|
||
yield self.url_result(f'https://www.nicovideo.jp/watch/{item}', 'Niconico', item)
|
||
if not results:
|
||
break
|
||
|
||
def _search_results(self, query):
|
||
return self._entries(
|
||
self._proto_relative_url(f'//www.nicovideo.jp/{self._SEARCH_TYPE}/{query}'), query)
|
||
|
||
|
||
class NicovideoSearchIE(NicovideoSearchBaseIE, SearchInfoExtractor):
|
||
IE_DESC = 'Nico video search'
|
||
IE_NAME = 'nicovideo:search'
|
||
_SEARCH_KEY = 'nicosearch'
|
||
|
||
|
||
class NicovideoSearchURLIE(NicovideoSearchBaseIE):
|
||
IE_NAME = f'{NicovideoSearchIE.IE_NAME}_url'
|
||
IE_DESC = 'Nico video search URLs'
|
||
_VALID_URL = r'https?://(?:www\.)?nicovideo\.jp/search/(?P<id>[^?#&]+)?'
|
||
_TESTS = [{
|
||
'url': 'http://www.nicovideo.jp/search/sm9',
|
||
'info_dict': {
|
||
'id': 'sm9',
|
||
'title': 'sm9',
|
||
},
|
||
'playlist_mincount': 40,
|
||
}, {
|
||
'url': 'https://www.nicovideo.jp/search/sm9?sort=h&order=d&end=2020-12-31&start=2020-01-01',
|
||
'info_dict': {
|
||
'id': 'sm9',
|
||
'title': 'sm9',
|
||
},
|
||
'playlist_count': 31,
|
||
}]
|
||
|
||
def _real_extract(self, url):
|
||
query = self._match_id(url)
|
||
return self.playlist_result(self._entries(url, query), query, query)
|
||
|
||
|
||
class NicovideoSearchDateIE(NicovideoSearchBaseIE, SearchInfoExtractor):
|
||
IE_DESC = 'Nico video search, newest first'
|
||
IE_NAME = f'{NicovideoSearchIE.IE_NAME}:date'
|
||
_SEARCH_KEY = 'nicosearchdate'
|
||
_TESTS = [{
|
||
'url': 'nicosearchdateall:a',
|
||
'info_dict': {
|
||
'id': 'a',
|
||
'title': 'a',
|
||
},
|
||
'playlist_mincount': 1610,
|
||
}]
|
||
|
||
_START_DATE = dt.date(2007, 1, 1)
|
||
_RESULTS_PER_PAGE = 32
|
||
_MAX_PAGES = 50
|
||
|
||
def _entries(self, url, item_id, start_date=None, end_date=None):
|
||
start_date, end_date = start_date or self._START_DATE, end_date or dt.datetime.now().date()
|
||
|
||
# If the last page has a full page of videos, we need to break down the query interval further
|
||
last_page_len = len(list(self._get_entries_for_date(
|
||
url, item_id, start_date, end_date, self._MAX_PAGES,
|
||
note=f'Checking number of videos from {start_date} to {end_date}')))
|
||
if (last_page_len == self._RESULTS_PER_PAGE and start_date != end_date):
|
||
midpoint = start_date + ((end_date - start_date) // 2)
|
||
yield from self._entries(url, item_id, midpoint, end_date)
|
||
yield from self._entries(url, item_id, start_date, midpoint)
|
||
else:
|
||
self.to_screen(f'{item_id}: Downloading results from {start_date} to {end_date}')
|
||
yield from self._get_entries_for_date(
|
||
url, item_id, start_date, end_date, note=' Downloading page %(page)s')
|
||
|
||
def _get_entries_for_date(self, url, item_id, start_date, end_date=None, page_num=None, note=None):
|
||
query = {
|
||
'start': str(start_date),
|
||
'end': str(end_date or start_date),
|
||
'sort': 'f',
|
||
'order': 'd',
|
||
}
|
||
if page_num:
|
||
query['page'] = str(page_num)
|
||
|
||
yield from super()._entries(url, item_id, query=query, note=note)
|
||
|
||
|
||
class NicovideoTagURLIE(NicovideoSearchBaseIE):
|
||
IE_NAME = 'niconico:tag'
|
||
IE_DESC = 'NicoNico video tag URLs'
|
||
_SEARCH_TYPE = 'tag'
|
||
_VALID_URL = r'https?://(?:www\.)?nicovideo\.jp/tag/(?P<id>[^?#&]+)?'
|
||
_TESTS = [{
|
||
'url': 'https://www.nicovideo.jp/tag/ドキュメンタリー淫夢',
|
||
'info_dict': {
|
||
'id': 'ドキュメンタリー淫夢',
|
||
'title': 'ドキュメンタリー淫夢',
|
||
},
|
||
'playlist_mincount': 400,
|
||
}]
|
||
|
||
def _real_extract(self, url):
|
||
query = self._match_id(url)
|
||
return self.playlist_result(self._entries(url, query), query, query)
|
||
|
||
|
||
class NiconicoUserIE(InfoExtractor):
|
||
_VALID_URL = r'https?://(?:www\.)?nicovideo\.jp/user/(?P<id>\d+)(?:/video)?/?(?:$|[#?])'
|
||
_TEST = {
|
||
'url': 'https://www.nicovideo.jp/user/419948',
|
||
'info_dict': {
|
||
'id': '419948',
|
||
},
|
||
'playlist_mincount': 101,
|
||
}
|
||
_API_URL = 'https://nvapi.nicovideo.jp/v2/users/%s/videos?sortKey=registeredAt&sortOrder=desc&pageSize=%s&page=%s'
|
||
_PAGE_SIZE = 100
|
||
|
||
_API_HEADERS = {
|
||
'X-Frontend-ID': '6',
|
||
'X-Frontend-Version': '0',
|
||
}
|
||
|
||
def _entries(self, list_id):
|
||
total_count = 1
|
||
count = page_num = 0
|
||
while count < total_count:
|
||
json_parsed = self._download_json(
|
||
self._API_URL % (list_id, self._PAGE_SIZE, page_num + 1), list_id,
|
||
headers=self._API_HEADERS,
|
||
note='Downloading JSON metadata%s' % (f' page {page_num}' if page_num else ''))
|
||
if not page_num:
|
||
total_count = int_or_none(json_parsed['data'].get('totalCount'))
|
||
for entry in json_parsed['data']['items']:
|
||
count += 1
|
||
yield self.url_result(
|
||
f'https://www.nicovideo.jp/watch/{entry["essential"]["id"]}', ie=NiconicoIE)
|
||
page_num += 1
|
||
|
||
def _real_extract(self, url):
|
||
list_id = self._match_id(url)
|
||
return self.playlist_result(self._entries(list_id), list_id)
|
||
|
||
|
||
class NiconicoLiveIE(NiconicoBaseIE):
|
||
IE_NAME = 'niconico:live'
|
||
IE_DESC = 'ニコニコ生放送'
|
||
_VALID_URL = r'https?://(?:sp\.)?live2?\.nicovideo\.jp/(?:watch|gate)/(?P<id>lv\d+)'
|
||
_TESTS = [{
|
||
'note': 'this test case includes invisible characters for title, pasting them as-is',
|
||
'url': 'https://live.nicovideo.jp/watch/lv339533123',
|
||
'info_dict': {
|
||
'id': 'lv339533123',
|
||
'title': '激辛ペヤング食べます\u202a( ;ᯅ; )\u202c（歌枠オーディション参加中）',
|
||
'view_count': 1526,
|
||
'comment_count': 1772,
|
||
'description': '初めましてもかって言います❕\nのんびり自由に適当に暮らしてます',
|
||
'uploader': 'もか',
|
||
'channel': 'ゲストさんのコミュニティ',
|
||
'channel_id': 'co5776900',
|
||
'channel_url': 'https://com.nicovideo.jp/community/co5776900',
|
||
'timestamp': 1670677328,
|
||
'is_live': True,
|
||
},
|
||
'skip': 'livestream',
|
||
}, {
|
||
'url': 'https://live2.nicovideo.jp/watch/lv339533123',
|
||
'only_matching': True,
|
||
}, {
|
||
'url': 'https://sp.live.nicovideo.jp/watch/lv339533123',
|
||
'only_matching': True,
|
||
}, {
|
||
'url': 'https://sp.live2.nicovideo.jp/watch/lv339533123',
|
||
'only_matching': True,
|
||
}]
|
||
|
||
def _real_extract(self, url):
|
||
video_id = self._match_id(url)
|
||
webpage, urlh = self._download_webpage_handle(f'https://live.nicovideo.jp/watch/{video_id}', video_id)
|
||
|
||
embedded_data = self._parse_json(unescapeHTML(self._search_regex(
|
||
r'<script\s+id="embedded-data"\s*data-props="(.+?)"', webpage, 'embedded data')), video_id)
|
||
|
||
ws_url = traverse_obj(embedded_data, ('site', 'relive', 'webSocketUrl'))
|
||
if not ws_url:
|
||
raise ExtractorError('The live hasn\'t started yet or already ended.', expected=True)
|
||
ws_url = update_url_query(ws_url, {
|
||
'frontend_id': traverse_obj(embedded_data, ('site', 'frontendId')) or '9',
|
||
})
|
||
|
||
hostname = remove_start(urllib.parse.urlparse(urlh.url).hostname, 'sp.')
|
||
|
||
ws = self._request_webpage(
|
||
Request(ws_url, headers={'Origin': f'https://{hostname}'}),
|
||
video_id=video_id, note='Connecting to WebSocket server')
|
||
|
||
self.write_debug('Sending HLS server request')
|
||
ws.send(json.dumps({
|
||
'type': 'startWatching',
|
||
'data': {
|
||
'stream': {
|
||
'quality': 'abr',
|
||
'protocol': 'hls',
|
||
'latency': 'high',
|
||
'accessRightMethod': 'single_cookie',
|
||
'chasePlay': False,
|
||
},
|
||
'room': {
|
||
'protocol': 'webSocket',
|
||
'commentable': True,
|
||
},
|
||
'reconnect': False,
|
||
},
|
||
}))
|
||
|
||
while True:
|
||
recv = ws.recv()
|
||
if not recv:
|
||
continue
|
||
data = json.loads(recv)
|
||
if not isinstance(data, dict):
|
||
continue
|
||
if data.get('type') == 'stream':
|
||
m3u8_url = data['data']['uri']
|
||
qualities = data['data']['availableQualities']
|
||
cookies = data['data']['cookies']
|
||
break
|
||
elif data.get('type') == 'disconnect':
|
||
self.write_debug(recv)
|
||
raise ExtractorError('Disconnected at middle of extraction')
|
||
elif data.get('type') == 'error':
|
||
self.write_debug(recv)
|
||
message = traverse_obj(data, ('body', 'code')) or recv
|
||
raise ExtractorError(message)
|
||
elif self.get_param('verbose', False):
|
||
if len(recv) > 100:
|
||
recv = recv[:100] + '...'
|
||
self.write_debug(f'Server said: {recv}')
|
||
|
||
title = traverse_obj(embedded_data, ('program', 'title')) or self._html_search_meta(
|
||
('og:title', 'twitter:title'), webpage, 'live title', fatal=False)
|
||
|
||
raw_thumbs = traverse_obj(embedded_data, ('program', 'thumbnail')) or {}
|
||
thumbnails = []
|
||
for name, value in raw_thumbs.items():
|
||
if not isinstance(value, dict):
|
||
thumbnails.append({
|
||
'id': name,
|
||
'url': value,
|
||
**parse_resolution(value, lenient=True),
|
||
})
|
||
continue
|
||
|
||
for k, img_url in value.items():
|
||
res = parse_resolution(k, lenient=True) or parse_resolution(img_url, lenient=True)
|
||
width, height = res.get('width'), res.get('height')
|
||
|
||
thumbnails.append({
|
||
'id': f'{name}_{width}x{height}',
|
||
'url': img_url,
|
||
'ext': traverse_obj(parse_qs(img_url), ('image', 0, {determine_ext(default_ext='jpg')})),
|
||
**res,
|
||
})
|
||
|
||
for cookie in cookies:
|
||
self._set_cookie(
|
||
cookie['domain'], cookie['name'], cookie['value'],
|
||
expire_time=unified_timestamp(cookie.get('expires')), path=cookie['path'], secure=cookie['secure'])
|
||
|
||
fmt_common = {
|
||
'live_latency': 'high',
|
||
'origin': hostname,
|
||
'protocol': 'niconico_live',
|
||
'video_id': video_id,
|
||
'ws': ws,
|
||
}
|
||
q_iter = (q for q in qualities[1:] if not q.startswith('audio_')) # ignore initial 'abr'
|
||
a_map = {96: 'audio_low', 192: 'audio_high'}
|
||
|
||
formats = self._extract_m3u8_formats(m3u8_url, video_id, ext='mp4', live=True)
|
||
for fmt in formats:
|
||
if fmt.get('acodec') == 'none':
|
||
fmt['format_id'] = next(q_iter, fmt['format_id'])
|
||
elif fmt.get('vcodec') == 'none':
|
||
abr = parse_bitrate(fmt['url'].lower())
|
||
fmt.update({
|
||
'abr': abr,
|
||
'format_id': a_map.get(abr, fmt['format_id']),
|
||
})
|
||
fmt.update(fmt_common)
|
||
|
||
return {
|
||
'id': video_id,
|
||
'title': title,
|
||
**traverse_obj(embedded_data, {
|
||
'view_count': ('program', 'statistics', 'watchCount'),
|
||
'comment_count': ('program', 'statistics', 'commentCount'),
|
||
'uploader': ('program', 'supplier', 'name'),
|
||
'channel': ('socialGroup', 'name'),
|
||
'channel_id': ('socialGroup', 'id'),
|
||
'channel_url': ('socialGroup', 'socialGroupPageUrl'),
|
||
}),
|
||
'description': clean_html(traverse_obj(embedded_data, ('program', 'description'))),
|
||
'timestamp': int_or_none(traverse_obj(embedded_data, ('program', 'openTime'))),
|
||
'is_live': True,
|
||
'thumbnails': thumbnails,
|
||
'formats': formats,
|
||
}
|