This commit is contained in:
Han Hyeji 2024-04-14 17:27:04 -05:00 committed by GitHub
commit 15504c82ed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 198 additions and 0 deletions

View File

@ -750,6 +750,10 @@ from .nationalgeographic import (
NationalGeographicVideoIE,
NationalGeographicTVIE,
)
from .nate import (
NateIE,
NateProgramIE,
)
from .naver import NaverIE
from .nba import (
NBAWatchEmbedIE,

View File

@ -0,0 +1,194 @@
# coding: utf-8
from __future__ import unicode_literals
import itertools
from .common import InfoExtractor
from ..utils import (
ExtractorError,
int_or_none,
merge_dicts,
T,
traverse_obj,
txt_or_none,
unified_strdate,
url_or_none,
)
class NateBaseIE(InfoExtractor):
_API_BASE = 'https://tv.nate.com/api/v1/'
def _download_webpage_handle(self, url_or_request, video_id, *args, **kwargs):
fatal = kwargs.get('fatal', True)
kwargs['fatal'] = False
res = super(NateBaseIE, self)._download_webpage_handle(
url_or_request, video_id, *args, **kwargs)
if not res:
if fatal:
raise ExtractorError('Failed to download webpage')
return res
status = res[1].getcode()
if 200 <= status < 400:
new_url = res[1].geturl()
if url_or_request != new_url and '/Error.html' in new_url:
raise ExtractorError(
'Download redirected to Error.html: expired?',
expected=True)
else:
msg = 'Failed to download webpage: HTTP code %d' % status
if fatal:
raise ExtractorError(msg)
else:
self.report_warning(msg)
return res
class NateIE(NateBaseIE):
_VALID_URL = r'https?://(?:m\.)?tv\.nate\.com/clip/(?P<id>[0-9]+)'
_TESTS = [{
'url': 'https://tv.nate.com/clip/1848976',
'info_dict': {
'id': '1848976',
'ext': 'mp4',
'title': '[결승 오프닝 타이틀] 2018 LCK 서머 스플릿 결승전 kt Rolster VS Griffin',
'description': 'md5:e1b79a7dcf0d8d586443f11366f50e6f',
'thumbnail': r're:^http?://.*\.jpg$',
'upload_date': '20180908',
'age_limit': 15,
'duration': 73,
'uploader': '2018 LCK 서머 스플릿(롤챔스)',
'channel': '2018 LCK 서머 스플릿(롤챔스)',
'channel_id': '3606',
'uploader_id': '3606',
'tags': 'count:59',
},
'skip': 'Redirect to Error.html',
}, {
'url': 'https://tv.nate.com/clip/4300566',
# 'md5': '02D3CAB3907B60C58043761F8B5BF2B3',
'info_dict': {
'id': '4300566',
'ext': 'mp4',
'title': '[심쿵엔딩] 이준호x이세영, 서로를 기억하며 끌어안는 두 사람!💕, MBC 211204 방송',
'description': 'md5:edf489c54ea2682c7973154b2089aa0e',
'thumbnail': r're:^http?://.*\.jpg$',
'upload_date': '20211204',
'age_limit': 15,
'duration': 201,
'uploader': '옷소매 붉은 끝동',
'channel': '옷소매 붉은 끝동',
'channel_id': '27987',
'uploader_id': '27987',
'tags': 'count:20',
},
'params': {'skip_download': True},
}, {
'url': 'https://tv.nate.com/clip/4764792',
'info_dict': {
'id': '4764792',
'ext': 'mp4',
'title': '흥을 돋우는 가야금 연주와 트롯의 만남⬈ ‘열두줄’♪ TV CHOSUN 230625 방송',
'description': 'md5:85734d3f9daebe4aa4f20cc73bdcc90c',
'upload_date': '20230625',
'uploader_id': '29116',
'uploader': '쇼퀸',
'age_limit': 15,
'thumbnail': r're:^http?://.*\.jpg$',
'duration': 182,
'channel': '쇼퀸',
'channel_id': '29116',
'tags': 'count:25',
},
'params': {'skip_download': True},
}]
_QUALITY = {
'36': 2160,
'35': 1080,
'34': 720,
'33': 480,
'32': 360,
'31': 270,
}
def _real_extract(self, url):
video_id = self._match_id(url)
video_data, urlh = self._download_json_handle(
'{0}clip/{1}'.format(self._API_BASE, video_id), video_id,
fatal=False)
if not video_data:
raise ExtractorError('Empty programme JSON')
title = video_data['clipTitle']
formats = []
for f_url in traverse_obj(video_data, ('smcUriList', Ellipsis, T(url_or_none))):
fmt_id = f_url[-2:]
formats.append({
'format_id': fmt_id,
'url': f_url,
'height': self._QUALITY.get(fmt_id),
'quality': int_or_none(fmt_id),
})
self._sort_formats(formats)
info = traverse_obj(video_data, {
'uploader': ('programTitle', T(txt_or_none)),
'uploader_id': ('programSeq', T(txt_or_none)),
})
for up, ch in (('uploader', 'channel'), ('uploader_id', 'channel_id')):
info[ch] = info.get(up)
return merge_dicts({
'id': video_id,
'title': title,
'formats': formats,
}, info, traverse_obj(video_data, {
'description': ('synopsis', T(txt_or_none)),
'thumbnail': ('contentImg', T(url_or_none)),
'upload_date': (('broadDate', 'regDate'), T(unified_strdate)),
'age_limit': ('targetAge', T(int_or_none)),
'duration': ('playTime', T(int_or_none)),
'tags': ('hashTag', T(lambda s: s.split(',') or None)),
}, get_all=False))
class NateProgramIE(NateBaseIE):
_VALID_URL = r'https?://tv\.nate\.com/program/clips/(?P<id>[0-9]+)'
_TESTS = [{
'url': 'https://tv.nate.com/program/clips/27987',
'playlist_mincount': 191,
'info_dict': {
'id': '27987',
},
}, {
'url': 'https://tv.nate.com/program/clips/3606',
'playlist_mincount': 15,
'info_dict': {
'id': '3606',
},
'skip': 'Redirect to Error.html',
}]
def _entries(self, pl_id):
for page_num in itertools.count(1):
program_data, urlh = self._download_json_handle(
'{0}program/{1}/clip/ranking'.format(self._API_BASE, pl_id),
pl_id, query={'size': 20, 'page': page_num},
note='Downloading page {0}'.format(page_num), fatal=False)
empty = True
for clip_id in traverse_obj(program_data, ('content', Ellipsis, 'clipSeq', T(txt_or_none))):
yield self.url_result(
'https://tv.nate.com/clip/%s' % clip_id,
ie=NateIE.ie_key(), video_id=clip_id)
empty = False
if traverse_obj(program_data, 'last') or (program_data and empty):
break
def _real_extract(self, url):
pl_id = self._match_id(url)
return self.playlist_result(self._entries(pl_id), playlist_id=pl_id)